def testInsertMerge2(self): res = Resort() res.name = "Test Insert Resort" res.id = 2 t1 = Trail() t1.id = 1 t2 = Trail() t2.id = 1 res.trails.append(t1) res.trails.append(t2) self.acc.insertData([res]) self.assertIsNone(self.acc.queryResort(2))
def testRelationships(self): res = Resort() res.name = "Test Resort" res.id = 123 trail = Trail() trail.name = "Test Trail" trail.id = 222 photo = Photo() photo.name = "Test Trail photo" photo.id = 222 photo.trail = trail trail.photos.append(photo) res.trails.append(trail) res.photos.append(photo) self.acc.insertData([res]) self.assertEqual(self.acc.queryResort(123).name, "Test Resort") self.assertEqual( self.acc.queryResort(123).trails[0].name, "Test Trail") self.assertEqual( self.acc.queryResort(123).photos[0].name, "Test Trail photo") self.assertEqual( self.acc.queryTrail(222).photos[0].name, "Test Trail photo")
def testQueryTrail(self): trail = Trail() trail.name = "Test Query Trail" trail.id = 50 self.acc.insertData([trail]) self.assertEqual(self.acc.queryTrail(50).name, "Test Query Trail")