def initialize():
if config['TESTING']:
for k in debug_config.keys():
config[k] = debug_config[k]
# initialize Data Storage
userstore = UserStore()
userstore.set_database_path(config.get('DATABASE_PATH'))
# initialize loggers
logger = logging.getLogger('auth')
handler = logging.FileHandler(config.get('LOG_FILE'))
formatter = logging.Formatter(config['LOG']['FORMAT'])
handler.setFormatter(formatter)
logger.addHandler(handler)
logger.setLevel(logging.DEBUG)
logger = logging.getLogger('service')
handler = logging.FileHandler(config.get('LOG_FILE'))
formatter = logging.Formatter(config['LOG']['FORMAT'])
handler.setFormatter(formatter)
logger.addHandler(handler)
logger.setLevel(logging.DEBUG)
class DataStoreTestCase(unittest.TestCase):
database_path = "temp.db"
provider = None
def setUp(self):
# userprovider.initialize(self.database_path)
self.provider = UserStore()
self.provider.set_database_path(self.database_path)
pass
def tearDown(self):
#os.remove(self.database_path)
self.provider.close()
self.provider = None
os.unlink(self.database_path)
pass
def testInitialize(self):
# Check DB was created
self.assertTrue(os.path.isfile(self.database_path), 'Temporary db should\'ve been created')
# Check user_table was created.
query = "SELECT * FROM sqlite_master WHERE type='table' AND name='user_table'"
cursor = self.provider.__get_cursor__()
cursor.execute(query)
list = cursor.fetchall()
self.assertTrue(len(list) > 0, "User_table should've been created")
def test_add_user(self):
data = {
'username': '******',
'password': '******'
}
# Create user
user = User(data['username'], data['password'])
self.provider.save_user(user)
# Get user back and check if the same
saved_user = self.provider.get_user(data['username'])
self.assertEqual(user, saved_user, "Users should've been same")
self.assertTrue(self.provider.has_user(user), "User should already exists")
self.assertTrue(self.provider.is_username_taken(user.username), "Username should already exists")
