From af9e2cf0fa1ded6de246d1ac5571c1150f714f5f Mon Sep 17 00:00:00 2001 From: Rebecca Law Date: Thu, 26 Nov 2015 09:59:23 +0000 Subject: [PATCH] 108536234: Added a dao method to get all users. --- app/main/dao/users_dao.py | 4 +++ tests/app/main/dao/test_get_all_users.py | 32 ++++++++++++++++++++++++ 2 files changed, 36 insertions(+) create mode 100644 tests/app/main/dao/test_get_all_users.py diff --git a/app/main/dao/users_dao.py b/app/main/dao/users_dao.py index 8a03de264..97bdfb9f3 100644 --- a/app/main/dao/users_dao.py +++ b/app/main/dao/users_dao.py @@ -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() diff --git a/tests/app/main/dao/test_get_all_users.py b/tests/app/main/dao/test_get_all_users.py new file mode 100644 index 000000000..3cc846ca4 --- /dev/null +++ b/tests/app/main/dao/test_get_all_users.py @@ -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]