from db import DBReader db_reader = DBReader() while True: username_input = input("Enter your username: "******"Enter your password: ") db_reader.get_by_username_password(username_input, password_input) if type(db_reader.get_by_username_password(username_input, password_input)) == list: print(db_reader.get_by_username_password(username_input, password_input)) else: print(db_reader.get_by_username_password(username_input, password_input)) continue while True: print(db_reader.input_direction()) break
def test_db_reader_get_info_from_correct_password_entry(self): file_input = ["bhigh,password,benjamin high,26,160"] db_reader = DBReader(file_input) self.assertEqual(db_reader.get_by_username_password("bhigh", "password"),["benjamin high","26","160"])
def test_db_reader_get_info_from_incorrect_password_entry(self): file_input = ["bhigh,password,benjamin high,26,160"] db_reader = DBReader(file_input) self.assertEqual(db_reader.get_by_username_password("asdf", "adfdsf"),("Either your username or password is incorrect."))
def test_db_reader_can_format_expected_file_structure(self): expected_input = ["bhigh,password,benjamin high,26,160", "mhigh,1234,marisa high,25,115"] expected_output = [["bhigh","password","benjamin high","26","160"], ["mhigh","1234","marisa high","25","115"]] self.assertEqual(DBReader.clean_file(expected_input), expected_output)