def test_add_device(self): p = Project(name="project1", path="path/project1") db.session.add(p) db.session.commit() d1 = Device(adb_sn="sn_1", constructor="constructeur1", model="model1", imei="imei1", status="Not Configured") d2 = Device(adb_sn="sn_2", constructor="constructeur2", model="model2", imei="imei2", status="Not Configured") db.session.add(d1) db.session.add(d2) db.session.commit() p.add_device([d1, d2]) db.session.commit() self.assertEqual(p.devices.count(), 2) p.delete_device(d2) db.session.commit() self.assertEqual(p.devices.count(), 1)
def test_get_project_from_device(self): p1 = Project(name="project1", path="path/project1") p2 = Project(name="project2", path="path/project2") db.session.add(p1) db.session.add(p2) db.session.commit() d = Device(adb_sn="sn_1", constructor="constructeur1", model="model1", imei="imei1", status="Not Configured") db.session.add(d) db.session.commit() p1.add_device(d) db.session.commit() self.assertTrue(d.project().name == "project1") self.assertFalse(d.project().name == "project2")