more tests

tests/app/dao/test_permissions_dao.py

@@ -1,3 +1,4 @@
+from app.dao import DAOClass
 from app.dao.permissions_dao import permission_dao
 from tests.app.db import create_service
 
@@ -22,3 +23,17 @@ def test_get_permissions_by_user_id_returns_only_active_service(sample_user):
     assert len(permissions) == 7
     assert active_service in [i.service for i in permissions]
     assert inactive_service not in [i.service for i in permissions]
+
+
+def test_dao_class(sample_user):
+    create_service(user=sample_user, service_name="Active service")
+    create_service(user=sample_user, service_name="Inactive service", active=False)
+
+    permissions_orig = permission_dao.get_permissions_by_user_id(user_id=sample_user.id)
+    assert len(permissions_orig) == 7
+    dao = DAOClass()
+
+    for permission in permissions_orig:
+        dao.delete_instance(permission, True)
+    permissions = permission_dao.get_permissions_by_user_id(user_id=sample_user.id)
+    assert len(permissions) == 0

tests/app/dao/test_users_dao.py

@@ -12,6 +12,7 @@ from app.dao.service_user_dao import (
     dao_update_service_user,
 )
 from app.dao.users_dao import (
+    _remove_values_for_keys_if_present,
     count_user_verify_codes,
     create_secret_code,
     dao_archive_user,
@@ -294,3 +295,16 @@ def test_user_cannot_be_archived_if_the_other_service_members_do_not_have_the_ma
 
     assert len(sample_service.users) == 3
     assert not user_can_be_archived(active_user)
+
+
+def test_remove_values_for_keys_if_present():
+    keys = {'a', 'b', 'c'}
+    my_dict = {
+        'a': 1,
+        'b': 2,
+        'c': 3,
+        'd': 4,
+    }
+    _remove_values_for_keys_if_present(my_dict, keys)
+
+    assert my_dict == {'d': 4}