def test_unicode_io(self):
"""
Tests that unicode is saved and retrieved properly
"""
tm1 = TestVertexModel.create(test_val=9, name=u'test2')
tm2 = TestVertexModel.get(tm1._id)
tm1.delete()
def test_model_deleting_works_properly(self):
"""
Tests that an instance's delete method deletes the instance
"""
tm = TestVertexModel.create(test_val=8, name='123456789')
vid = tm.id
tm.delete()
with self.assertRaises(TestVertexModel.DoesNotExist):
tm2 = TestVertexModel.get(vid)
def test_success_case(self):
""" Tests that the update method works as expected """
tm = TestVertexModel.create(test_val=8, name='123456789')
tm2 = tm.update(test_val=9)
tm3 = TestVertexModel.get(tm.id)
self.assertEqual(tm2.test_val, 9)
self.assertEqual(tm3.test_val, 9)
tm.delete()
def test_get_by_id(self):
v1 = TestVertexModel.create()
results = TestVertexModel.get(v1.id)
self.assertIsInstance(results, TestVertexModel)
self.assertEqual(results, v1)
with self.assertRaises(TestEdgeModel.DoesNotExist):
results = TestVertexModel.get(None)
with self.assertRaises(TestEdgeModel.DoesNotExist):
results = TestVertexModel.get('nonexistant')
v2 = TestVertexModel2.create(test_val=0)
with self.assertRaises(TestVertexModel.WrongElementType):
results = TestVertexModel.get(v2.id)
v2.delete()
v1.delete()
def test_reload(self):
""" Tests that and instance's reload method does an inplace update of the instance """
tm0 = TestVertexModel.create(test_val=8, name='123456789')
tm1 = TestVertexModel.get(tm0.id)
tm1.test_val = 7
tm1.save()
tm0.reload()
self.assertEqual(tm0.test_val, 7)
tm0.delete()
def test_model_updating_works_properly(self):
"""
Tests that subsequent saves after initial model creation work
"""
tm = TestVertexModel.create(test_val=8, name='123456789')
tm.test_val = 100
tm.save()
tm.test_val = 80
tm.save()
tm.test_val = 60
tm.save()
tm.test_val = 40
tm.save()
tm.test_val = 20
tm.save()
tm2 = TestVertexModel.get(tm.id)
self.assertEquals(tm.test_val, tm2.test_val)
tm.delete()
