Add tests

Author: rmk135

tests/unit/providers/configuration/conftest.py

@@ -3,7 +3,7 @@
 import os
 
 from dependency_injector import providers
-from pytest import fixture, mark
+from pytest import fixture
 
 
 @fixture

tests/unit/providers/configuration/test_pydantic_settings_in_init_py36.py

@@ -0,0 +1,121 @@
+"""Configuration.from_pydantic() tests."""
+
+import pydantic
+from dependency_injector import providers
+from pytest import fixture, mark, raises
+
+
+class Section11(pydantic.BaseModel):
+    value1 = 1
+
+
+class Section12(pydantic.BaseModel):
+    value2 = 2
+
+
+class Settings1(pydantic.BaseSettings):
+    section1 = Section11()
+    section2 = Section12()
+
+
+class Section21(pydantic.BaseModel):
+    value1 = 11
+    value11 = 11
+
+
+class Section3(pydantic.BaseModel):
+    value3 = 3
+
+
+class Settings2(pydantic.BaseSettings):
+    section1 = Section21()
+    section3 = Section3()
+
+
+@fixture
+def config(config_type, pydantic_settings_1, pydantic_settings_2):
+    if config_type == "strict":
+        return providers.Configuration(strict=True)
+    elif config_type == "default":
+        return providers.Configuration(pydantic_settings=[pydantic_settings_1, pydantic_settings_2])
+    else:
+        raise ValueError("Undefined config type \"{0}\"".format(config_type))
+
+
+@fixture
+def pydantic_settings_1():
+    return Settings1()
+
+
+@fixture
+def pydantic_settings_2():
+    return Settings2()
+
+
+def test_load(config):
+    config.load()
+
+    assert config() == {
+        "section1": {
+            "value1": 11,
+            "value11": 11,
+        },
+        "section2": {
+            "value2": 2,
+        },
+        "section3": {
+            "value3": 3,
+        },
+    }
+    assert config.section1() == {"value1": 11, "value11": 11}
+    assert config.section1.value1() == 11
+    assert config.section1.value11() == 11
+    assert config.section2() == {"value2": 2}
+    assert config.section2.value2() == 2
+    assert config.section3() == {"value3": 3}
+    assert config.section3.value3() == 3
+
+
+def test_get_pydantic_settings(config, pydantic_settings_1, pydantic_settings_2):
+    assert config.get_pydantic_settings() == [pydantic_settings_1, pydantic_settings_2]
+
+
+def test_set_pydantic_settings(config):
+    class Settings3(pydantic.BaseSettings):
+        ...
+
+    class Settings4(pydantic.BaseSettings):
+        ...
+
+    settings_3 = Settings3()
+    settings_4 = Settings4()
+
+    config.set_pydantic_settings([settings_3, settings_4])
+    assert config.get_pydantic_settings() == [settings_3, settings_4]
+
+
+def test_file_does_not_exist(config):
+    config.set_pydantic_settings([pydantic.BaseSettings()])
+    config.load()
+    assert config() == {}
+
+
+@mark.parametrize("config_type", ["strict"])
+def test_file_does_not_exist_strict_mode(config):
+    config.set_pydantic_settings([pydantic.BaseSettings()])
+    with raises(ValueError):
+        config.load()
+    assert config() == {}
+
+
+def test_required_file_does_not_exist(config):
+    config.set_pydantic_settings([pydantic.BaseSettings()])
+    with raises(ValueError):
+        config.load(required=True)
+
+
+@mark.parametrize("config_type", ["strict"])
+def test_not_required_file_does_not_exist_strict_mode(config):
+    config.set_pydantic_settings([pydantic.BaseSettings()])
+    config.load(required=False)
+    assert config() == {}