From 3021677d9e8a4982dd68366f04674ddfe85f6634 Mon Sep 17 00:00:00 2001
From: Kenneth Kehl <@kkehl@flexion.us>
Date: Fri, 11 Oct 2024 07:21:40 -0700
Subject: [PATCH] try to fix test code

---
 tests/app/dao/test_services_dao.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/app/dao/test_services_dao.py b/tests/app/dao/test_services_dao.py
index 2efba7472..613fc21df 100644
--- a/tests/app/dao/test_services_dao.py
+++ b/tests/app/dao/test_services_dao.py
@@ -338,7 +338,7 @@ def test_dao_add_user_to_service_raises_error_if_adding_folder_permissions_for_a
     other_service_folder = create_template_folder(other_service)
     folder_permissions = [str(other_service_folder.id)]
 
-    stmt = select(func.count(ServiceUser.id))
+    stmt = select(func.count(ServiceUser.service_id))
     assert db.session.execute(stmt).scalar() == 2
 
     with pytest.raises(IntegrityError) as e:
@@ -351,7 +351,7 @@ def test_dao_add_user_to_service_raises_error_if_adding_folder_permissions_for_a
         'insert or update on table "user_folder_permissions" violates foreign key constraint'
         in str(e.value)
     )
-    stmt = select(func.count(ServiceUser.id))
+    stmt = select(func.count(ServiceUser.service_id))
     assert db.session.execute(stmt).scalar() == 2
 
 
@@ -864,7 +864,7 @@ def test_delete_service_and_associated_objects(notify_db_session):
     create_notification(template=template, api_key=api_key)
     create_invited_user(service=service)
     user.organizations = [organization]
-    stmt = select(func.count(ServicePermission.id))
+    stmt = select(func.count(ServicePermission.service_id))
     assert db.session.execute(stmt).scalar() == len(
         (
             ServicePermissionType.SMS,