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 initialize_database():
""" Initialize the database. """
with app.open_resource('schema.sql') as schema:
databaseservice.initialize_database(app.database, schema)
