def test5_adding_an_entry_to_a_table_in_the_database_does_not_cause_exception(self): # Arrange exception_raised = False database = db._connect_to_database() create_test_table() # Act try: db._insert_row(f'INSERT INTO {test_table_name} ({test_table_field}) VALUES ("test")') except: exception_raised = True # Assert self.assertFalse(exception_raised) delete_test_table() database.close()
def test6_fetching_a_table_with_an_entry_in_the_database_returns_correct_value(self): # Arrange database = db._connect_to_database() create_test_table() test_value = 'hello' db._insert_row(f'INSERT INTO {test_table_name} ({test_table_field}) VALUES ("{test_value}")') # Act try: table = db._fetch_table(test_table_name) except: print('error fetching table') self.assertFalse(True) # Assert self.assertTrue(len(table) == 1 and table[0][0] == test_value) delete_test_table() database.close()
def _insert_preference_row(person_id: int, drink_name: str): values_string = f'({person_id},"{drink_name}")' query = "INSERT INTO preferences (person_id,drink_name) VALUES" + values_string db._insert_row(query)
def insert_round_row(owner_id: int): query = f"INSERT INTO rounds (owner_id,active) VALUES ({owner_id},1)" db._insert_row(query)
def insert_order_row(round_id: int, person_id: int, drink_name: str): query = f"INSERT INTO orders (round_id,person_id,drink_name) VALUES ({round_id},{person_id},\"{drink_name}\")" db._insert_row(query)
def _insert_person_row(person: Person): values_string = f"('{person.first_name}','{person.surname}','{person.slack_id}')" query = "INSERT INTO people (first_name, surname, slack_id) VALUES "+ values_string db._insert_row(query) print(f"{person.name.title()} added!")
def _insert_drink_row(drink: Drink): values_string = f"('{drink.name}','{drink.type}')" query = "INSERT INTO drinks (drink_name,drink_type) VALUES " + values_string db._insert_row(query)