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")
