From 06913064c3ca1d71c030c00b36b88bfd85538b46 Mon Sep 17 00:00:00 2001 From: brian Date: Tue, 15 Apr 2025 22:24:02 +0000 Subject: [PATCH] added unittests for repositories DTO base class --- .../unit/type_checking_base_model_test.py | 46 +++++++++++++++++++ 1 file changed, 46 insertions(+) create mode 100644 shared/repositories/tests/unit/type_checking_base_model_test.py diff --git a/shared/repositories/tests/unit/type_checking_base_model_test.py b/shared/repositories/tests/unit/type_checking_base_model_test.py new file mode 100644 index 0000000..fcf9f6c --- /dev/null +++ b/shared/repositories/tests/unit/type_checking_base_model_test.py @@ -0,0 +1,46 @@ +"""Definition of unittests for TypeCheckingBaseModel class.""" + +import unittest + +import pytest +from pydantic import ValidationError + +from shared.repositories.src.dto.type_checking_base_model import TypeCheckingBaseModel + + +class UUT(TypeCheckingBaseModel): + """Test class for TypeCheckingBaseModel.""" + + string_field: str + int_field: int + + +class TestTypeCheckingBaseModel(unittest.TestCase): + """Test class for TypeCheckingBaseModel.""" + + def setUp(self): + """Set up the test case.""" + self.uut = UUT + self.string_field = 'test' + self.int_field = 123 + + def test_type_checking_on_instantiation(self): + """Test type checking on instantiation.""" + with pytest.raises(ValidationError): + _ = self.uut( + string_field=self.int_field, + int_field=self.string_field, + ) + + def test_attributes_frozen(self): + """Test that attributes cannot be updated.""" + uut_instance = self.uut( + string_field=self.string_field, + int_field=self.int_field, + ) + with pytest.raises(ValidationError): + uut_instance.string_field = self.int_field + + +if __name__ == '__main__': + pytest.main(['-s', '-v', __file__])