0.2 Vehicle

from vehicle.vehicle_code.vehicle import Vehicle
from unittest import TestCase, main


class VehicleTest(TestCase):
    def setUp(self) -> None:
        self.vehicle = Vehicle(100.10, 220.1)

    def test_init_with_fuel_and_horse_power(self):
        self.assertEqual(100.10, self.vehicle.fuel)
        self.assertEqual(100.10, self.vehicle.capacity)
        self.assertEqual(220.1, self.vehicle.horse_power)
        self.assertEqual(self.vehicle.fuel_consumption, self.vehicle.DEFAULT_FUEL_CONSUMPTION)

    def test_drive_if_fuel_needed_is_less_or_equal_to_the_fuel(self):
        self.vehicle.drive(10)
        self.assertEqual(87.6, self.vehicle.fuel)

    def test_drive_if_the_fuel_needed_is_more_than_the_fuel_raises(self):
        with self.assertRaises(Exception) as ex:
            self.vehicle.drive(100)

        self.assertEqual("Not enough fuel", str(ex.exception))

    def test_refuel_method_if_the_total_fuel_is_not_above_the_max_capacity(self):
        self.vehicle.drive(10)
        self.vehicle.refuel(5)
        self.assertEqual(92.6, self.vehicle.fuel)

    def test_refuel_method_if_the_total_fuel_is_more_than_the_given_capacity_raises(self):
        with self.assertRaises(Exception) as ex:
            self.vehicle.refuel(300)

        self.assertEqual("Too much fuel", str(ex.exception))

    def test__str__method_if_returns_the_correct_string(self):
        self.assertEqual(f"The vehicle has 220.1 horse power with 100.1 fuel left and 1.25 fuel consumption", str(self.vehicle))


if __name__ == '__main__':
    main()