system-pri/backend/tests/functional_tests/coordinator/test_students.py

import copy

from app.students.models import Student, YearGroup

from ...factory import StudentFactory, YearGroupFactory
from ...utils import (
    _test_case_client,
    _test_case_client_without_response,
    assert_model_changes,
    create_many_models,
    create_one_model,
)

new_data = {"first_name": "Martin", "last_name": "Green"}

data_to_create_student = {
    "first_name": "Albert",
    "last_name": "Marcus",
    "index": 123_456,
}


def test_list_students(test_app_with_context) -> None:
    with test_app_with_context.test_client() as client:
        yg = create_one_model(YearGroupFactory)
        create_many_models(45, StudentFactory, year_group_id=yg.id)
        data = _test_case_client_without_response(
            client,
            f"/api/coordinator/students/{yg.id}/?per_page=10",
            None,
            200,
            method="get",
        )
        assert data.get("max_pages") == 5
        assert len(data.get("students")) == 10


def test_detail_student(test_app_with_context) -> None:
    with test_app_with_context.test_client() as client:
        st = create_one_model(StudentFactory)
        data = _test_case_client_without_response(
            client,
            f"/api/coordinator/students/{st.id}/detail/",
            None,
            200,
            method="get",
        )
        assert_model_changes(st, data)


def test_detail_student_if_student_doesnt_exist(test_app_with_context) -> None:
    with test_app_with_context.test_client() as client:
        _test_case_client(
            client,
            "/api/coordinator/students/43/detail/",
            None,
            "Not found student!",
            404,
            method="get",
            key="error",
        )


def test_delete_student(test_app_with_context) -> None:
    with test_app_with_context.test_client() as client:
        st = create_one_model(StudentFactory)
        _test_case_client(
            client,
            f"/api/coordinator/students/{st.id}/",
            None,
            "Student was deleted!",
            202,
            method="delete",
        )
        assert len(Student.query.all()) == 0


def test_delete_student_if_student_doesnt_exist(test_app_with_context) -> None:
    with test_app_with_context.test_client() as client:
        _test_case_client(
            client,
            "/api/coordinator/students/43/",
            None,
            "Not found student!",
            404,
            method="delete",
            key="error",
        )


def test_edit_student(test_app_with_context) -> None:
    with test_app_with_context.test_client() as client:
        st = create_one_model(StudentFactory)
        _test_case_client(
            client,
            f"/api/coordinator/students/{st.id}/",
            new_data,
            "Student was updated!",
            200,
            method="put",
        )


def test_edit_student_with_invalid_data(test_app_with_context) -> None:
    data = copy.copy(new_data)
    data["index"] = 3_232_323
    with test_app_with_context.test_client() as client:
        st = create_one_model(StudentFactory)
        _test_case_client(
            client,
            f"/api/coordinator/students/{st.id}/",
            data,
            "Validation error",
            400,
            method="put",
        )


def test_edit_student_if_student_doesnt_exist(test_app_with_context) -> None:
    with test_app_with_context.test_client() as client:
        _test_case_client(
            client,
            "/api/coordinator/students/54/",
            new_data,
            "Not found student!",
            404,
            method="put",
            key="error",
        )


def test_create_student(test_app_with_context) -> None:
    data = copy.copy(data_to_create_student)
    with test_app_with_context.test_client() as client:
        yg = create_one_model(YearGroupFactory)
        data["year_group_id"] = yg.id
        _test_case_client(
            client,
            "/api/coordinator/students/",
            data,
            "Student was created!",
            200,
            method="post",
        )


def test_create_student_with_invalid_data(test_app_with_context) -> None:
    data = copy.copy(data_to_create_student)
    data["index"] = 1_123_432
    with test_app_with_context.test_client() as client:
        yg = create_one_model(YearGroupFactory)
        data["year_group_id"] = yg.id
        _test_case_client(
            client,
            "/api/coordinator/students/",
            data,
            "Validation error",
            400,
            method="post",
        )


def test_create_student_if_year_group_doesnt_exist(test_app_with_context) -> None:
    data = copy.copy(data_to_create_student)
    with test_app_with_context.test_client() as client:
        data["year_group_id"] = 34
        _test_case_client(
            client,
            "/api/coordinator/students/",
            data,
            "Not found year group!",
            404,
            method="post",
            key="error",
        )


def test_create_student_if_student_has_already_assigned_to_this_year_group(
    test_app_with_context,
) -> None:
    data = copy.copy(data_to_create_student)
    with test_app_with_context.test_client() as client:
        yg = create_one_model(YearGroupFactory)
        create_one_model(Student, data, year_group_id=yg.id, email="tmp@gmail.com")
        data["year_group_id"] = yg.id
        _test_case_client(
            client,
            "/api/coordinator/students/",
            data,
            "Student has already assigned to this year group!",
            400,
            method="post",
            key="error",
        )