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)