def test_add_user():
#create DB
my_backend = Backend(cfg, init=True)
#make sure user does not exist
assert my_backend.get_users(user_id=user_id) == []
#create user
assert my_backend.add_user(user_id)
db = sqlite3.connect(DB_path)
cursor = db.cursor()
sql = ''' SELECT * FROM users WHERE id=? '''
cursor.execute(sql, (user_id, ))
resp = cursor.fetchone()
assert resp[USER_ID] == user_id and \
resp[USER_ACTIVE] == 1
#adding the same user should fail
assert not my_backend.add_user(user_id)
def test_get_user():
#create DB
my_backend = Backend(cfg, init=True)
my_backend.add_user(user_id)
#check that user was correctly created
db = sqlite3.connect(DB_path)
cursor = db.cursor()
sql = ''' SELECT * FROM users WHERE id=? '''
cursor.execute(sql, (user_id, ))
resp = cursor.fetchone()
assert resp[USER_ID] == user_id and \
resp[USER_ACTIVE] == 1
#load user
user = my_backend.get_users(user_id=user_id)[0]
assert user[USER_ID] == user_id and \
resp[USER_ACTIVE] == 1
def test_toggle_user():
my_backend = Backend(cfg, init=True)
#create user
my_backend.add_user(user_id)
#check it's active
user = my_backend.get_users(user_id=user_id)[0]
assert user[USER_ACTIVE] == 1
#disable user
my_backend.toggle_user(user_id, active=False)
#check it's inactive
user = my_backend.get_users(user_id=user_id)[0]
assert user[USER_ACTIVE] == 0
#enable user
my_backend.toggle_user(user_id, active=True)
#check it's active
user = my_backend.get_users(user_id=user_id)[0]
assert user[USER_ACTIVE] == 1