def setUp(self): self.database_file = 'test.db' with open('schema.sql') as schema: databaseservice.initialize_database(self.database_file, schema) self.database_connection = \ databaseservice.connect_database(self.database_file) self.database_cursor = self.database_connection.cursor() self.database = databaseservice.DatabaseService( \ self.database_connection, self.database_cursor) # Test data self.first_url = 'first_url' self.first_shortened = 'first_shortened' self.second_url = 'second_url' self.second_shortened = 'second_shortened' self.first_url_first_visit_time = 12345 self.first_url_second_visit_time = 2345 self.second_url_first_visit_time = 9876 self.second_url_second_visit_time = 9999 self.test_user = databaseservice.User('rahulraj', 'fake_hash', 'fake_salt') self.second_user = databaseservice.User('second', 'fake_hash2', 'fake_salt2')
def setUp(self): self.database_file = "test.db" with open("schema.sql") as schema: initialize_database(self.database_file, schema) self.database_connection = connect_database(self.database_file) self.database_cursor = self.database_connection.cursor() self.database = DatabaseService(self.database_connection, self.database_cursor) # Test data self.test_user = User("rahulraj", "fake_hash", "fake_salt") self.second_user = User("second", "fake_hash2", "fake_salt2") self.test_room = Room("A test room", "This room was created for testing") self.second_room = Room("A second room", "This room was created for more complex tests") self.third_room = Room("A third room", "This room was created for even more complex tests")
def connect_database(): """ Connect to app.database and return the connection object. """ return databaseservice.connect_database(app.database)