def test_all_method_invalid_length(self):
v1 = TestVertexModel.create()
v2 = TestVertexModel.create()
from mogwai.exceptions import MogwaiQueryError
with self.assertRaises(MogwaiQueryError):
TestVertexModel.all([v1.id, v2.id, 'invalid'])
v1.delete()
v2.delete()
def test_model_save_and_load(self):
"""
Tests that models can be saved and retrieved
"""
tm0 = TestVertexModel.create(test_val=8, name='123456789')
tm1 = TestVertexModel.create(test_val=9, name='456789')
tms = TestVertexModel.all([tm0.id, tm1.id])
self.assertEqual(len(tms), 2)
for pname in tm0._properties.keys():
self.assertEquals(getattr(tm0, pname), getattr(tms[0], pname))
tms = TestVertexModel.all([tm1.id, tm0.id])
self.assertEqual(tms[0].id, tm1.id)
self.assertEqual(tms[1].id, tm0.id)
tm0.delete()
tm1.delete()
def test_all_method(self):
with self.assertRaises(MogwaiQueryError):
TestVertexModel.all(1)
