Fix tests for invalid files

Author: dgarros

infrahub_sdk/yaml.py

@@ -60,6 +60,28 @@ def load_content(self) -> None:
     def validate_content(self) -> None:
         pass
 
+    @classmethod
+    def init(cls, location: Path, multiple_documents: bool, content: dict | None) -> Self:
+        if not content:
+            return cls._file_is_empty(path=location, has_multiple_document=multiple_documents)
+
+        return cls(location=location, multiple_documents=multiple_documents, content=content)
+
+    @classmethod
+    def _file_is_empty(cls, path: Path, has_multiple_document: bool) -> Self:
+        return cls(
+            location=path, multiple_documents=has_multiple_document, error_message="Invalid YAML/JSON file", valid=False
+        )
+
+    @classmethod
+    def _file_is_invalid(cls, path: Path, has_multiple_document: bool) -> Self:
+        return cls(
+            location=path,
+            multiple_documents=has_multiple_document,
+            error_message="Invalid YAML/JSON file",
+            valid=False,
+        )
+
     @classmethod
     def load_file_from_disk(cls, path: Path) -> list[Self]:
         yaml_files: list[Self] = []
@@ -71,24 +93,23 @@ def load_file_from_disk(cls, path: Path) -> list[Self]:
 
             if has_multiple_document:
                 for content in yaml.safe_load_all(file_content):
-                    yaml_files.append(cls(location=path, multiple_documents=has_multiple_document, content=content))
+                    yaml_files.append(
+                        cls.init(location=path, multiple_documents=has_multiple_document, content=content)
+                    )
+
             else:
                 yaml_files.append(
-                    cls(location=path, multiple_documents=has_multiple_document, content=yaml.safe_load(file_content))
+                    cls.init(
+                        location=path, multiple_documents=has_multiple_document, content=yaml.safe_load(file_content)
+                    )
                 )
+
         except FileNotValidError as exc:
             yaml_files.append(
                 cls(location=path, multiple_documents=has_multiple_document, error_message=exc.message, valid=False)
             )
         except (yaml.YAMLError, ParserError):
-            yaml_files.append(
-                cls(
-                    location=path,
-                    multiple_documents=has_multiple_document,
-                    error_message="Invalid YAML/JSON file",
-                    valid=False,
-                )
-            )
+            yaml_files.append(cls._file_is_invalid(path, has_multiple_document))
 
         if has_multiple_document:
             for idx, file in enumerate(yaml_files):

tests/unit/ctl/test_schema_app.py

@@ -14,7 +14,7 @@ def test_schema_load_empty(httpx_mock: HTTPXMock):
     result = runner.invoke(app=app, args=["load", str(fixture_file)])
 
     assert result.exit_code == 1
-    assert "Empty YAML/JSON file" in result.stdout
+    assert "Invalid YAML/JSON file" in result.stdout
 
 
 def test_schema_load_one_valid(httpx_mock: HTTPXMock):

tests/unit/ctl/test_validate_app.py

@@ -21,7 +21,7 @@ def test_validate_schema_empty():
 
     result = runner.invoke(app=app, args=["schema", str(fixture_file)])
     assert result.exit_code == 1
-    assert "Empty YAML/JSON file" in remove_ansi_color(result.stdout)
+    assert "Invalid YAML/JSON file" in remove_ansi_color(result.stdout)
 
 
 def test_validate_schema_non_valid():