def test_all_method_for_known_bad_input(self):
from mogwai.exceptions import MogwaiQueryError
with self.assertRaises(TypeError):
results = TestEdgeModel.all()
with self.assertRaises(MogwaiQueryError):
results = TestEdgeModel.all(None)
with self.assertRaises(MogwaiQueryError):
results = TestEdgeModel.all('test')
def test_all_method_invalid_length(self):
e1 = TestEdgeModel.create(self.v1, self.v2, test_val=3)
e2 = TestEdgeModel.create(self.v1, self.v2, test_val=4)
from mogwai.exceptions import MogwaiQueryError
with self.assertRaises(MogwaiQueryError):
TestEdgeModel.all([e1.id, e2.id, 'invalid'])
e1.delete()
e2.delete()
def test_all_method_for_known_bad_input(self):
from mogwai.exceptions import MogwaiQueryError
with self.assertRaises(TypeError):
results = TestEdgeModel.all()
with self.assertRaises(MogwaiQueryError):
results = TestEdgeModel.all(None)
with self.assertRaises(MogwaiQueryError):
results = TestEdgeModel.all('test')
def test_all_method_invalid_length(self):
e1 = TestEdgeModel.create(self.v1, self.v2, test_val=3)
e2 = TestEdgeModel.create(self.v1, self.v2, test_val=4)
from mogwai.exceptions import MogwaiQueryError
with self.assertRaises(MogwaiQueryError):
TestEdgeModel.all([e1.id, e2.id, 'invalid'])
e1.delete()
e2.delete()
def test_all_method_for_known_ids(self):
e1 = TestEdgeModel.create(self.v1, self.v2, test_val=3)
e2 = TestEdgeModel.create(self.v1, self.v2, test_val=4)
results = TestEdgeModel.all([e1.id, e2.id])
self.assertEqual(len(results), 2)
for result in results:
self.assertIsInstance(result, TestEdgeModel)
self.assertIn(result, [e1, e2])
e1.delete()
e2.delete()
def test_all_method_for_known_ids(self):
e1 = TestEdgeModel.create(self.v1, self.v2, test_val=3)
e2 = TestEdgeModel.create(self.v1, self.v2, test_val=4)
results = TestEdgeModel.all([e1.id, e2.id])
self.assertEqual(len(results), 2)
for result in results:
self.assertIsInstance(result, TestEdgeModel)
self.assertIn(result, [e1, e2])
e1.delete()
e2.delete()
