mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-02-05 10:53:28 -05:00
108536234: Added a dao method to get all users.
This commit is contained in:
@@ -9,3 +9,7 @@ def insert_user(user):
|
||||
|
||||
def get_user_by_id(id):
|
||||
return Users.query.filter_by(id=id).first()
|
||||
|
||||
|
||||
def get_all_users():
|
||||
return Users.query.all()
|
||||
|
||||
32
tests/app/main/dao/test_get_all_users.py
Normal file
32
tests/app/main/dao/test_get_all_users.py
Normal file
@@ -0,0 +1,32 @@
|
||||
from datetime import datetime
|
||||
|
||||
from app.main.dao import users_dao
|
||||
from app.models import Users
|
||||
|
||||
|
||||
def test_get_all_users_returns_all_users(notifications_admin, notifications_admin_db):
|
||||
user1 = Users(name='test one',
|
||||
password='somepassword',
|
||||
email_address='test1@get_all.gov.uk',
|
||||
mobile_number='+441234123412',
|
||||
created_at=datetime.now(),
|
||||
role_id=1)
|
||||
user2 = Users(name='test two',
|
||||
password='some2ndpassword',
|
||||
email_address='test2@get_all.gov.uk',
|
||||
mobile_number='+441234123412',
|
||||
created_at=datetime.now(),
|
||||
role_id=1)
|
||||
user3 = Users(name='test two',
|
||||
password='some2ndpassword',
|
||||
email_address='test2@get_all.gov.uk',
|
||||
mobile_number='+441234123412',
|
||||
created_at=datetime.now(),
|
||||
role_id=1)
|
||||
|
||||
users_dao.insert_user(user1)
|
||||
users_dao.insert_user(user2)
|
||||
users_dao.insert_user(user3)
|
||||
users = users_dao.get_all_users()
|
||||
assert len(users) == 3
|
||||
assert users == [user1, user2, user3]
|
||||
Reference in New Issue
Block a user