def test_all(self, skip, limit, expected_length):
r = GraphRelationshipRepository('test', {})
r._g = self.g
actual = r.all(skip, limit)
self.assertTrue(self.g.match.called)
self.assertEqual(len(actual), expected_length)
self.assertIsInstance(actual[0], dict)
def test_match(self, origin, target):
r = GraphRelationshipRepository('test', {})
r._g = self.g
actual = r.match(origin, target)
self.assertTrue(self.g.match.called)
self.assertTrue('TEST', self.g.match.call_args[0][1])
if origin is None:
self.assertIsNone(self.g.match.call_args[0][0])
else:
self.assertIsNotNone(self.g.match.call_args[0][0])
if target is None:
self.assertIsNone(self.g.match.call_args[0][2])
else:
self.assertIsNotNone(self.g.match.call_args[0][2])