def test_add_new_vehicle(self):
mileage.add_miles('Blue Car', 100)
expected = {'BLUE CAR': 100, 'WHITE': 111, 'GINGER': 222, 'PINK': 333}
self.compare_db_to_expected(expected)
mileage.add_miles('Green Car', 50)
expected['GREEN CAR'] = 50
self.compare_db_to_expected(expected)
def test_increase_miles_for_vehicle(self):
mileage.add_miles('Red Car', 100)
expected = {'Red Car': 100}
self.compare_db_to_expected(expected)
mileage.add_miles('Red Car', 50)
expected['Red Car'] = 100 + 50
self.compare_db_to_expected(expected)
def test_increase_miles_for_vehicle(self):
mileage.add_miles('Red Car', 100)
expected = {'RED CAR': 100, 'WHITE': 111, 'GINGER': 222, 'PINK': 333}
self.compare_db_to_expected(expected)
mileage.add_miles('Red Car', 50)
expected['RED CAR'] = 100 + 50
self.compare_db_to_expected(expected)
def test_add_new_vehicle_upper_case(self):
mileage.add_miles('YeLlOw car', 100)
expected = {'YELLOW CAR': 100}
self.compare_db_to_expected(expected)
mileage.add_miles('yellow car', 50)
expected['YELLOW CAR'] = 100 + 50
self.compare_db_to_expected(expected)
def test_add_new_vehicle(self):
mileage.add_miles('Blue Car', 100) #adds 100 miles to blue car
expected = {'BLUE CAR': 100} #should be uppercase + 100
self. compare_db_to_expected(expected) #compares
mileage.add_miles('Green Car', 50) #adds 50 miles to green car
expected ['GREEN CAR'] = 50 #green car is equal to 50
self.compare_db_to_expected(expected) #compares
def test_add_new_vehicle(self):
mileage.add_miles('Blue Car', 100)
expected = {'Blue Car': 100}
self.compare_db_to_expected(expected)
mileage.add_miles('Green Car', 50)
expected['Green Car'] = 50
self.compare_db_to_expected(expected)
def test_add_new_vehicle(self):
mileage.add_miles('Blue Car', 100)
expected = { 'BLUE CAR': 100 }
self.compare_db_to_expected(expected)
mileage.add_miles('gReEn CaR', 50)
expected['GREEN CAR'] = 50
self.compare_db_to_expected(expected)
def test_increase_miles_for_vehicle(self):
mileage.add_miles('Red Car', 100) #add miles to red car
expected = {'RED CAR' : 100} #expected input
self.compare_db_to_expected(expected) #compare
mileage.add_miles('Red Car', 50) #add more miles to red car
expected['RED CAR'] = 100+50 #expected to add miles together not overwrite
self.compare_db_to_expected(expected) #compare
def setUp(self):
# Overwrite the mileage
mileage.db_url = self.test_db_url
# drop everything from the DB to always start with an empty database
conn = sqlite3.connect(self.test_db_url)
conn.execute('DELETE FROM miles')
conn.commit()
conn.close()
mileage.add_miles('purple car', 50)
mileage.add_miles('green car', 25)
def test_to_upper_case(self):
v1 = mileage.to_upper('yellow car')
mileage.add_miles(v1, 200)
expected = 'YELLOW CAR'
conn = sqlite3.connect(self.test_db_url)
cursor = conn.cursor()
all_data = cursor.execute('SELECT * FROM MILES').fetchall()
for row in all_data:
# Vehicle exists, and mileage is correct
self.assertEqual(row[0], expected)
def test_search_vehicle_mileage(self):
mileage.add_miles('Blue Car', 100)
miles = mileage.search_vehicle_mileage('blue car')
self.assertEqual(100, miles)
mileage.add_miles('red car', 100)
miles = mileage.search_vehicle_mileage('RED CAR')
self.assertEqual(100, miles)
miles = mileage.search_vehicle_mileage('purple car')
self.assertEqual(None, miles)
def test_add_new_vehicle_mixed_case(self):
mileage.add_miles('OrAnGe car', 100)
expected = {
'ORANGE CAR': 100,
'WHITE': 111,
'GINGER': 222,
'PINK': 333
}
self.compare_db_to_expected(expected)
mileage.add_miles('orange car', 50)
expected['ORANGE CAR'] = 100 + 50
self.compare_db_to_expected(expected)
def setUp(self):
# Overwrite the mileage
mileage.db_url = self.test_db_url
mileage.create_database()
# drop everything from the DB to always start with an empty database
conn = sqlite3.connect(self.test_db_url)
conn.execute('DELETE FROM miles')
conn.commit()
conn.close()
mileage.add_miles("blue car", 20)
mileage.add_miles("red car", 30)
def test_add_new_vehicle_invalid_new_miles(self):
with self.assertRaises(MileageError):
mileage.add_miles('Car', -100)
with self.assertRaises(MileageError):
mileage.add_miles('Car', 'abc')
with self.assertRaises(MileageError):
mileage.add_miles('Car', '12.def')
def test_increase_miles_with_cap_variation(self):
mileage.add_miles('Purple Car', 100)
expected = {'PURPLE CAR': 100}
self.compare_db_to_expected(expected)
mileage.add_miles('pUrPlE cAr', 50)
expected['PURPLE CAR'] = 100 + 50
self.compare_db_to_expected(expected)
mileage.add_miles('PurPLe CAr', 50)
expected['PURPLE CAR'] = 100 + 50 + 50
self.compare_db_to_expected(expected)
mileage.add_miles('PURPLE CAR', 50)
expected['PURPLE CAR'] = 100 + 50 + 50 + 50
self.compare_db_to_expected(expected)
def test_search_for_vehicle(self, expected):
conn = sqlite3.connect(self.test_db_url)
mileage.add_miles('Red Car', 100) #adds miles to red car
mileage.add_miles('Blue Car', 50) #adds miles to blue car
mileage.add_miles('Green Car', 23) #adds miles to green car
carResult = mileage.search_vehicle('gReEn CaR') #searches for green car
expected[carResult] = 23 #expected
self.compare_db_to_expected(expected) #compare
def test_add_new_vehicle(self):
mileage.add_miles('Blue Car', 100)
expected = { 'BLUE CAR': 100 }
self.compare_db_to_expected(expected)
mileage.add_miles('Green Car', 50)
expected['GREEN CAR'] = 50
self.compare_db_to_expected(expected)
with self.assertRaises(MileageError):
mileage.add_miles('blue CaR', 'banana')
def setUp(self):
# Overwrite the mileage
mileage.db_url = self.test_db_url
# drop everything from the DB to always start with an empty database
conn = sqlite3.connect(self.test_db_url)
conn.execute('DELETE FROM miles')
conn.commit()
cursor = conn.cursor()
mileage.add_miles('Blue Car', 100)
mileage.add_miles('Green Car', 200)
mileage.add_miles('Red Car', 300)
conn.commit()
conn.close()
def test_add_new_vehicle_no_vehicle(self):
with self.assertRaises(Exception):
mileage.add_miles(None, 100)
def test_search(self):
v1 = mileage.to_upper('white car')
mileage.add_miles(v1, 200)
search_vehicle = mileage.search_vehicle(v1)
self.assertIsNotNone(search_vehicle)
def test_vehicle_name(self):
mileage.add_miles('Blue Car', 100) #adds 100 to blue car
expected = {'BLUE CAR': 100}
self. compare_db_to_expected(expected) #makes sure blue car is BLUE CAR
def test_vehicle_upper(self):
mileage.add_miles('yes', 34)
expected = {'YES': 34}
self.compare_db_to_expected(expected)
