def test_get_by_id(self): e1 = TestEdgeModel.create(self.v1, self.v2, test_val=3) results = TestEdgeModel.get(e1.id) self.assertIsInstance(results, TestEdgeModel) self.assertEqual(results, e1) from mogwai.exceptions import MogwaiQueryError with self.assertRaises(TestEdgeModel.DoesNotExist): results = TestEdgeModel.get(None) with self.assertRaises(TestEdgeModel.DoesNotExist): results = TestEdgeModel.get('nonexistant') e2 = TestEdgeModel2.create(self.v1, self.v2, test_val=2) with self.assertRaises(TestEdgeModel.WrongElementType): results = TestEdgeModel.get(e2.id) e1.delete() e2.delete()
def test_reload(self): """ Tests that the reload method performs an inplace update of an instance's values """ e1 = TestEdgeModel.create(self.v1, self.v2, test_val=3) e2 = TestEdgeModel.get(e1.id) print_('\n{} {} {}: {} {}'.format(e1.id, e2.id, e1 == e2, e1.test_val, e2.test_val)) e2.test_val = 5 e2.save() print_("{} {} {}: {} {}".format(e1.id, e2.id, e1 == e2, e1.test_val, e2.test_val)) self.assertEqual(e1.test_val, 3) e1.reload() self.assertEqual(e1.test_val, 5) e1 = self.v1.outE()[0] self.assertEqual(e1.test_val, 5) e2.delete()