build.tests.test_app.decorators.test_validation

Complexity

Total Complexity

18

Size/Duplication

Total Lines	88
Duplicated Lines	0 %

Importance

Changes

0

9 Methods

Rating	Name	Duplication	Size	Complexity
A	TestValidateWithPydantic.test_context_property_binding_with_payload_location_json()	0	5	2
A	TestValidateWithPydantic.test_context_property_binding_with_payload_location_args()	0	5	2
A	TestValidateWithPydantic.test_validation_pass_with_payload_location_args()	0	3	2
A	TestValidateWithPydantic.initialize_function_and_call()	0	19	1
A	TestValidateWithPydantic.setUp()	0	2	1
A	TestValidateWithPydantic.test_validation_pass_with_payload_location_json()	0	3	2
A	TestValidateWithPydantic.test_validation_error_with_payload_location_json()	0	4	3
A	TestValidateWithPydantic.test_validation_error_with_payload_location_args()	0	4	3
A	TestValidateWithPydantic.test_json_force_load()	0	13	2

import json
from typing import Type

from pydantic import BaseModel, ValidationError, constr

from app.context import context_property
from app.decorators.validation import validate_with_pydantic, PayloadLocation
from tests import BaseTestCase


class TestValidateWithPydantic(BaseTestCase):
    """
    validate_with_schematics로 데코레이팅된 함수를 만들고,
    요청 데이터와 함께 request context를 열어서 이를 호출하는 방식으로 테스트를 진행합니다.
    """

    class TestSchema(BaseModel):
        foo: constr(min_length=1)

    def setUp(self):
        super(TestValidateWithPydantic, self).setUp()

    def initialize_function_and_call(
        self,
        payload_location: PayloadLocation,
        schema: Type[BaseModel] = TestSchema,
        json_force_load: bool = False,
    ):
        """
        인자 정보를 통해 `validate_with_pydantic`으로 데코레이팅된 함수를 생성하고, 호출합니다.
        """

        @validate_with_pydantic(
            payload_location=payload_location,
            model=schema,
            json_force_load=json_force_load,
        )
        def handler():
            pass

        handler()

    def test_validation_pass_with_payload_location_args(self):
        with self.app.test_request_context(query_string={"foo": "bar"}):
            self.initialize_function_and_call(PayloadLocation.ARGS)

    def test_validation_pass_with_payload_location_json(self):
        with self.app.test_request_context(json={"foo": "bar"}):
            self.initialize_function_and_call(PayloadLocation.JSON)

    def test_validation_error_with_payload_location_args(self):
        with self.app.test_request_context(query_string={"foo": ""}):
            with self.assertRaises(ValidationError) as e:
                self.initialize_function_and_call(PayloadLocation.ARGS)

    def test_validation_error_with_payload_location_json(self):
        with self.app.test_request_context(json={"foo": ""}):
            with self.assertRaises(ValidationError):
                self.initialize_function_and_call(PayloadLocation.JSON)

    def test_context_property_binding_with_payload_location_args(self):
        with self.app.test_request_context(query_string={"foo": "bar"}):
            self.initialize_function_and_call(PayloadLocation.ARGS)
            self.assertEqual(
                self.TestSchema(foo="bar"), context_property.request_payload
            )

    def test_context_property_binding_with_payload_location_json(self):
        with self.app.test_request_context(json={"foo": "bar"}):
            self.initialize_function_and_call(PayloadLocation.JSON)
            self.assertEqual(
                self.TestSchema(foo="bar"), context_property.request_payload
            )

    def test_json_force_load(self):
        with self.app.test_request_context(data=json.dumps({"foo": "bar"})):
            # TODO
            # self.initialize_function_and_call(
            #     PayloadLocation.JSON, json_force_load=False
            # )
            # print(context_property.request_payload)

            self.initialize_function_and_call(
                PayloadLocation.JSON, json_force_load=True
            )
            self.assertEqual(
                self.TestSchema(foo="bar"), context_property.request_payload
            )