finished controller unittests

This commit is contained in:
Lucas Jensen
2024-05-03 18:19:20 -07:00
parent 3812ca272f
commit e2b3978fec
6 changed files with 300 additions and 18 deletions

View File

@@ -0,0 +1,59 @@
from unittest.mock import Mock
import pytest
from fastapi import HTTPException, status
from icecream import ic
from app.controllers.group import GroupController
from app.models.group import Group
mock_queries = Mock()
gc = GroupController(group_queries=mock_queries)
valid_group_data = {
"id": 1,
"name": "Test Group",
"bio": "Test Bio",
}
invalid_group_data = {
"id": 1,
"name": "Test Group",
}
def test_type():
"""Tests the type of the controller object."""
assert isinstance(gc, GroupController)
def test_get_group():
"""Tests the retrieval of a group from the database with valid data."""
mock_queries.select_one_by_id.return_value = valid_group_data
group = gc.get_group()
assert isinstance(group, Group)
assert group.id == 1
assert group.name == "Test Group"
assert group.bio == "Test Bio"
def test_get_group_failure():
"""Tests a failure during the retrieval process or if the group is not found."""
mock_queries.select_one_by_id.return_value = invalid_group_data
with pytest.raises(HTTPException) as e:
gc.get_group()
assert isinstance(e.value, HTTPException)
assert e.value.status_code == status.HTTP_500_INTERNAL_SERVER_ERROR
def test_update_group_bio():
"""This test does not test updating of the bio, but rather tests that the corresponding
method in the queries module is called with the correct arguments.
"""
new_bio = "New Bio"
mock_queries.update_group_bio = Mock()
mock_queries.select_one_by_id.return_value = valid_group_data
group = gc.update_group_bio(new_bio)
Mock.assert_called_once_with(mock_queries.update_group_bio, new_bio)
assert isinstance(group, Group)