Fix bugs regarding service test data insertion

This commit is contained in:
pmoharana-cmd 2024-04-22 21:13:30 -04:00
parent a107337737
commit f4f816b94c
5 changed files with 94 additions and 87 deletions

View File

@ -8,7 +8,11 @@ from backend.models.service_model import Service
from backend.models.user_model import User
from backend.entities.service_entity import ServiceEntity
from backend.models.enum_for_models import ProgramTypeEnum, UserTypeEnum
from backend.services.exceptions import ServiceNotFoundException, ProgramNotAssignedException
from backend.services.exceptions import (
ServiceNotFoundException,
ProgramNotAssignedException,
)
class ServiceService:
@ -39,14 +43,15 @@ class ServiceService:
for program in programs:
query = select(ServiceEntity).filter(ServiceEntity.program == program)
entities = self._session.scalars(query)
all.append(entities)
services.append(entities)
return [service.to_model() for service in services]
def get_all(self, subject: User) -> list[Service]:
"""Service method retrieving all of the services in the table."""
if subject.role != UserTypeEnum.ADMIN:
raise ProgramNotAssignedException(f"User is not {UserTypeEnum.ADMIN}, cannot get all")
raise ProgramNotAssignedException(
f"User is not {UserTypeEnum.ADMIN}, cannot get all"
)
query = select(ServiceEntity)
entities = self._session.scalars(query).all()
@ -56,7 +61,9 @@ class ServiceService:
def create(self, subject: User, service: Service) -> Service:
"""Creates/adds a service to the table."""
if subject.role != UserTypeEnum.ADMIN:
raise ProgramNotAssignedException(f"User is not {UserTypeEnum.ADMIN}, cannot create service")
raise ProgramNotAssignedException(
f"User is not {UserTypeEnum.ADMIN}, cannot create service"
)
service_entity = ServiceEntity.from_model(service)
self._session.add(service_entity)
@ -66,12 +73,16 @@ class ServiceService:
def update(self, subject: User, service: Service) -> Service:
"""Updates a service if in the table."""
if subject.role != UserTypeEnum.ADMIN:
raise ProgramNotAssignedException(f"User is not {UserTypeEnum.ADMIN}, cannot update service")
raise ProgramNotAssignedException(
f"User is not {UserTypeEnum.ADMIN}, cannot update service"
)
service_entity = self._session.get(ServiceEntity, service.id)
if service_entity is None:
raise ServiceNotFoundException("The service you are searching for does not exist.")
raise ServiceNotFoundException(
"The service you are searching for does not exist."
)
service_entity.name = service.name
service_entity.status = service.status
@ -90,7 +101,9 @@ class ServiceService:
service_entity = self._session.get(ServiceEntity, service.id)
if service_entity is None:
raise ServiceNotFoundException("The service you are searching for does not exist.")
raise ServiceNotFoundException(
"The service you are searching for does not exist."
)
self._session.delete(service_entity)
self._session.commit()

View File

@ -4,7 +4,7 @@ import pytest
from sqlalchemy import Engine, create_engine, text
from sqlalchemy.orm import Session
from sqlalchemy.exc import OperationalError
from .services import user_test_data, tag_test_data
from .services import user_test_data, tag_test_data, service_test_data
from ..database import _engine_str
from ..env import getenv
@ -56,5 +56,6 @@ def session(test_engine: Engine):
def setup_insert_data_fixture(session: Session):
user_test_data.insert_fake_data(session)
tag_test_data.insert_fake_data(session)
service_test_data.insert_fake_data(session)
session.commit()
yield

View File

@ -2,12 +2,14 @@ import pytest
from ...services import ServiceService
from .fixtures import service_svc
from ...models.service_model import Service
from ...models.enum_for_models import ProgramTypeEnum
from . import user_test_data
from . import service_test_data
from ...services.exceptions import ServiceNotFoundException, ProgramNotAssignedException
def test_get_all(service_svc: ServiceService):
service = service_svc.get_all(user_test_data.admin)
assert len(service) == len(service_test_data.services)
@ -47,7 +49,9 @@ def test_update(service_svc: ServiceService):
def test_update_not_found(service_svc: ServiceService):
with pytest.raises(ServiceNotFoundException):
service = service_svc.update(user_test_data.admin, service_test_data.new_service)
service = service_svc.update(
user_test_data.admin, service_test_data.new_service
)
pytest.fail()

View File

@ -1,8 +1,9 @@
import pytest
from sqlalchemy.orm import Session
from backend.entities.service_entity import ServiceEntity
from backend.models.service_model import Service
from backend.models.enum_for_models import ProgramTypeEnum
from ...entities import ServiceEntity
from ...models.enum_for_models import ProgramTypeEnum
from ...models.service_model import Service
service_1 = Service(
id=1,
@ -10,7 +11,7 @@ service_1 = Service(
status="open",
summary="presentation educating community on domestic violence",
requirements=[""],
program= ProgramTypeEnum.COMMUNITY
program=ProgramTypeEnum.COMMUNITY,
)
service_2 = Service(
@ -19,7 +20,7 @@ service_2 = Service(
status="closed",
summary="service finding safe places to stay",
requirements=[""],
program= ProgramTypeEnum.DOMESTIC
program=ProgramTypeEnum.DOMESTIC,
)
service_3 = Service(
@ -28,7 +29,7 @@ service_3 = Service(
status="open",
summary="",
requirements=[""],
program= ProgramTypeEnum.DOMESTIC
program=ProgramTypeEnum.DOMESTIC,
)
service_4 = Service(
@ -37,7 +38,7 @@ service_4 = Service(
status="waitlist",
summary="community event",
requirements=[""],
program= ProgramTypeEnum.COMMUNITY
program=ProgramTypeEnum.COMMUNITY,
)
service_5 = Service(
@ -46,7 +47,7 @@ service_5 = Service(
status="open",
summary="talk circle for victims of domestic violence",
requirements=["18+"],
program= ProgramTypeEnum.COMMUNITY
program=ProgramTypeEnum.COMMUNITY,
)
service_6 = Service(
@ -55,7 +56,7 @@ service_6 = Service(
status="waitlist",
summary="program offering economic assistance",
requirements=[""],
program= ProgramTypeEnum.ECONOMIC
program=ProgramTypeEnum.ECONOMIC,
)
service_6_edit = Service(
@ -64,7 +65,7 @@ service_6_edit = Service(
status="open",
summary="program offering economic assistance",
requirements=["18+"],
program= ProgramTypeEnum.ECONOMIC
program=ProgramTypeEnum.ECONOMIC,
)
service_7 = Service(
@ -73,7 +74,7 @@ service_7 = Service(
status="waitlist",
summary="insert generic description",
requirements=[""],
program= ProgramTypeEnum.ECONOMIC
program=ProgramTypeEnum.ECONOMIC,
)
new_service = Service(
@ -82,7 +83,7 @@ new_service = Service(
status="open",
summary="insert other generic description",
requirements=[""],
program= ProgramTypeEnum.DOMESTIC
program=ProgramTypeEnum.DOMESTIC,
)
services = [service_1, service_2, service_3, service_4, service_5, service_6]
@ -115,7 +116,6 @@ def reset_table_id_seq(
def insert_fake_data(session: Session):
"""Inserts fake organization data into the test session."""
global services
# Create entities for test organization data
@ -130,14 +130,3 @@ def insert_fake_data(session: Session):
# Commit all changes
session.commit()
@pytest.fixture(autouse=True)
def fake_data_fixture(session: Session):
"""Insert fake data the session automatically when test is run.
Note:
This function runs automatically due to the fixture property `autouse=True`.
"""
insert_fake_data(session)
session.commit()
yield

View File

@ -71,7 +71,7 @@ toDelete = User(
group="none",
program=[programs.COMMUNITY],
created_at=datetime.now(),
role=roles.VOLUNTEER
role=roles.VOLUNTEER,
)
users = [volunteer, employee, admin, toDelete]