class TestTransaction(unittest.TestCase):
def setUp(self) -> None:
self.customer_data = {
"_id": "123",
"name": "Test Customer",
"address": "Cairo",
"phone": "0123456789",
"date": "13-2-2021",
}
self.vehicle_data = {
"_id": "123",
"make": "Test Vehicle",
"vin_number": "5555",
"owner_id": "0123456789",
}
self.customer = Customer(**self.customer_data)
self.vehicle = Vehicle(**self.vehicle_data)
def test_transaction_to_json(self):
expected = {
"date":
"13-2-2021",
"customer": {
"id": "123",
"name": "Test Customer",
"address": "Cairo",
"phone": "0123456789",
},
"vehicle": [{
"id": "123",
"make": "Test Vehicle",
"vin_number": "5555",
}]
}
result = {
"date": self.customer.date,
"customer": self.customer.to_json(),
"vehicle": [self.vehicle.to_json()]
}
self.assertDictEqual(result, expected)
class TestVehicle(unittest.TestCase):
def setUp(self) -> None:
self.data = {
"_id": "123",
"make": "Test Vehicle",
"vin_number": "5555",
"owner_id": "0123456789",
}
self.vehicle = Vehicle(**self.data)
def test_create_vehicle(self):
self.assertEqual(self.vehicle.id, "123")
self.assertEqual(self.vehicle.make, "Test Vehicle")
self.assertEqual(self.vehicle.vin_number, "5555")
self.assertEqual(self.vehicle.owner_id, "0123456789")
def test_vehicle_to_json(self):
expected = {
"id": "123",
"make": "Test Vehicle",
"vin_number": "5555",
}
result = self.vehicle.to_json()
self.assertDictEqual(result, expected)
