unit test

Author: benpankow

python_modules/libraries/dagster-components/dagster_components_tests/code_locations/test_code_locations.py

@@ -0,0 +1,19 @@
+import importlib
+import sys
+import warnings
+from pathlib import Path
+
+
+def test_load_component_defs_with_extraneous_files_in_components_dir():
+    # import the module
+    # ensure module is added to path
+    # capture warnings
+    with warnings.catch_warnings(record=True) as w:
+        sys.path.append(str(Path(__file__).parent / "extraneous_files_in_components_dir"))
+        module = importlib.import_module("extraneous_files_in_components_dir.definitions")
+        assert module is not None
+        assert len(w) == 1
+        assert (
+            str(w[0].message)
+            == "Found 1 Python files in the components root. Components expect all Python files to be nested in component subdirectories."
+        )

python_modules/libraries/dagster-dg/dagster_dg_tests/cli_tests/test_component_type_commands.py

@@ -112,6 +112,20 @@ def test_component_type_docs_success():
         assert_runner_result(result)
 
 
+def test_component_type_docs_success_output_console():
+    with ProxyRunner.test() as runner, isolated_components_venv(runner):
+        result = runner.invoke(
+            "component-type",
+            "docs",
+            "complex_schema_asset@dagster_components.test",
+            "--output",
+            "cli",
+        )
+        assert_runner_result(result)
+        assert "<html" in result.output
+        assert "An asset that has a complex params schema." in result.output
+
+
 def test_component_type_docs_with_no_dagster_components_fails() -> None:
     with (
         ProxyRunner.test() as runner,