def test_insertRepairs(self):
with RepairDatabaseUtils(self.connection) as db:
count = self.countRepairs()
self.assertTrue(db.insertRepair(3, 3, 1))
self.assertTrue(count + 1 == self.countRepairs())
self.assertTrue(db.insertRepair(4, 4, 1))
self.assertTrue(count + 2 == self.countRepairs())
def test_setRepairStatus(self):
with RepairDatabaseUtils(self.connection) as db:
repairID = 1
self.assertTrue(self.repairExists(repairID))
self.assertFalse(self.repairIsOngoing(repairID))
db.setRepairStatus(repairID, 1)
self.assertTrue(self.repairIsOngoing(repairID))
db.setRepairStatus(repairID, 0)
self.assertFalse(self.repairIsOngoing(repairID))
def setRepairStatus(self, repairID, status):
with RepairDatabaseUtils() as db:
db.setRepairStatus(repairID, status)
def cancelRepair(self, repairID):
with RepairDatabaseUtils() as db:
db.setRepairStatus(repairID, 2)
def listPersonalOngoingRepairs(self, userID):
with RepairDatabaseUtils() as db:
return db.listPersonalOngoingRepairs(userID)
def listPersonalRepairsHistory(self, userID):
with RepairDatabaseUtils() as db:
return db.listPersonalRepairsHistory(userID)
def insertRepair(self, userID, CarID):
with RepairDatabaseUtils() as db:
db.insertRepair(userID, CarID, 1)
def createRepairTable(self):
with RepairDatabaseUtils() as db:
db.createRepairTable()
def test_listPersonalOngoingRepairs(self):
userID = 1
with RepairDatabaseUtils(self.connection) as db:
self.assertTrue(
self.countOngoingRepairsByID(userID) == len(
db.listPersonalOngoingRepairs(userID)))