File size: 496 Bytes
5aee586
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
import pytest
     import sqlite3
     from app import init_db, load_users_from_db
     
     def test_init_db():
         init_db()
         conn = sqlite3.connect('private/users.db')
         cursor = conn.cursor()
         cursor.execute("SELECT username FROM users WHERE username='admin'")
         assert cursor.fetchone() is not None
         conn.close()
     
     def test_load_users():
         config = load_users_from_db()
         assert 'admin' in config['credentials']['usernames']