Anonymous avatar Anonymous committed ca618d1

fix tests

Comments (0)

Files changed (1)

dynamodb_mapper/tests/test_model.py

             "_aws_secret_access_key": None,
             "_region": None,
             "_connection": None,
+            "_tables_cache": {},
         }
 
     def tearDown(self):
             "_aws_secret_access_key": None,
             "_region": None,
             "_connection": None,
+            "_tables_cache": {},
         }
 
     def test_borgness(self):
         self.assertEqual(DATA[1][u'id'], res[1].id)
         self.assertEqual(DATA[1][u'name'], res[1].name)
 
-    @mock.patch("dynamodb_mapper.model.boto")
-    def test_get_batch_hash_range_key(self, m_boto):
+    @mock.patch("dynamodb_mapper.model.ConnectionBorg")
+    def test_get_batch_hash_range_key(self, m_borg):
         KEYS = [(1, u"The level with cute cats"),
                 (2, u"The level with horrible monsters")]
         DATA = [{
                     u"world": u"Hell",
                 }]
 
-        m_table = m_boto.connect_dynamodb.return_value.get_table.return_value
+        m_table = m_borg.return_value.get_table.return_value
         m_table.batch_get_item.return_value = DATA
 
         res = DoomMap.get_batch(KEYS)
Tip: Filter by directory path e.g. /media app.js to search for public/media/app.js.
Tip: Use camelCasing e.g. ProjME to search for ProjectModifiedEvent.java.
Tip: Filter by extension type e.g. /repo .js to search for all .js files in the /repo directory.
Tip: Separate your search with spaces e.g. /ssh pom.xml to search for src/ssh/pom.xml.
Tip: Use ↑ and ↓ arrow keys to navigate and return to view the file.
Tip: You can also navigate files with Ctrl+j (next) and Ctrl+k (previous) and view the file with Ctrl+o.
Tip: You can also navigate files with Alt+j (next) and Alt+k (previous) and view the file with Alt+o.