def test_initialization(self): storage = PlacesStorage() self.assertEqual(storage._data, {}) self.assertEqual(storage._version, None)
def setUp(self): super(PlacesStorageTest, self).setUp() self.p1, self.p2, self.p3 = create_test_map() self.storage = PlacesStorage() self.storage.sync()
class PlacesStorageTest(testcase.TestCase): def setUp(self): super(PlacesStorageTest, self).setUp() self.p1, self.p2, self.p3 = create_test_map() self.storage = PlacesStorage() self.storage.sync() def test_initialization(self): storage = PlacesStorage() self.assertEqual(storage._data, {}) self.assertEqual(storage._version, None) def test_sync(self): self.assertEqual(len(self.storage._data), 3) self.assertTrue(self.storage._version > 0) self.assertNotEqual(self.p1.size, 7) place = Place.objects.get(id=self.p1.id) place.size = 7 place.save() self.storage.sync() self.assertNotEqual(self.storage[self.p1.id].size, 7) self.storage.sync(force=True) self.assertEqual(self.storage[self.p1.id].size, 7) def test_sync_after_settings_update(self): self.assertEqual(len(self.storage._data), 3) self.assertTrue(self.storage._version > 0) self.assertNotEqual(self.p1.size, 7) place = Place.objects.get(id=self.p1.id) place.size = 7 place.save() self.storage.sync() self.assertNotEqual(self.storage[self.p1.id].size, 7) settings[self.storage.SETTINGS_KEY] = uuid.uuid4().hex self.storage.sync() self.assertEqual(self.storage[self.p1.id].size, 7) def test_getitem_wrong_id(self): self.assertRaises(exceptions.PlacesStorageError, self.storage.__getitem__, 666) def test_getitem(self): self.assertEqual(self.storage[self.p1.id].id, self.p1.id) def test_get(self): self.assertEqual(self.storage.get(666, self.p2).id, self.p2.id) self.assertEqual(self.storage.get(self.p3.id, self.p1).id, self.p3.id) def test_all(self): self.assertEqual(len(self.storage.all()), 3) def test_random_place(self): places = set([self.p1.id, self.p2.id, self.p3.id]) for i in xrange(100): places.discard(self.storage.random_place().id) self.assertFalse(places) def test_contains(self): self.assertTrue(self.p1.id in self.storage) self.assertFalse(666 in self.storage)