Allow Refs to be hashable using their data (#1053)

cloudtools · May 26, 2018 · 4eceaa0 · 4eceaa0
1 parent 8e2b9b4
commit 4eceaa0
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/tests/test_basic.py b/tests/test_basic.py
@@ -349,6 +349,22 @@ def test_ref_eq(self):
         self.assertNotEqual(r, Ref)
         self.assertNotEqual(wch.Ref(), "NonexistantResource")
 
+    def test_ref_hash(self):
+        s = hash("AWS::NoValue")
+        r = hash(Ref(s))
+
+        wch = cloudformation.WaitConditionHandle("TestResource")
+
+        self.assertEqual(s, r)
+        self.assertEqual(s, hash(NoValue))
+        self.assertEqual(r, hash(NoValue))
+        self.assertEqual(hash(wch.Ref()), hash("TestResource"))
+
+        self.assertNotEqual(r, hash("AWS::Region"))
+        self.assertNotEqual(r, hash(Region))
+        self.assertNotEqual(r, hash(Ref))
+        self.assertNotEqual(hash(wch.Ref()), hash("NonexistantResource"))
+
 
 class TestName(unittest.TestCase):
 

diff --git a/troposphere/__init__.py b/troposphere/__init__.py
@@ -493,6 +493,9 @@ def __eq__(self, other):
             return self.data == other.data
         return self.data.values()[0] == other
 
+    def __hash__(self):
+        return hash(self.data.values()[0])
+
 
 # Pseudo Parameter Ref's
 AccountId = Ref(AWS_ACCOUNT_ID)