Adjust unit tests to take account of BaseConfig new abstract method

Elektrobit · Mar 18, 2024 · 454cdbd · 454cdbd
1 parent d8ea7e5
commit 454cdbd
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/embdgen-core/tests/cli/test_cli.py b/embdgen-core/tests/cli/test_cli.py
@@ -15,6 +15,9 @@ class TestConfig(BaseConfig):
     def load(self, filename: Path):
         return MBR()
 
+    def load_str(self, data: str):
+        return MBR()
+
     @classmethod
     def probe(cls, _filename: Path) -> bool:
         return True
@@ -25,6 +28,10 @@ def load(self, filename: Path):
         self.__class__.was_called = True
         return MBR()
 
+    def load_str(self, data: str):
+        self.__class__.was_called = True
+        return MBR()
+
     @classmethod
     def probe(cls, _filename: Path) -> bool:
         return False

diff --git a/embdgen-core/tests/config/test_BaseConfig.py b/embdgen-core/tests/config/test_BaseConfig.py
@@ -8,6 +8,9 @@ class ConfigFoo(BaseConfig):
     def load(self, filename: Path) -> BaseLabel:
         return None
 
+    def load_str(self, data: str) -> BaseLabel:
+        return None
+
 def test_BaseConfig():
     c = ConfigFoo()
     assert c.probe(Path("")) is False