start adding tests

Author: ghickman

.gitignore

@@ -1,3 +1,4 @@
+/.coverage
 /.env
 /output/
 dist

justfile

@@ -1,5 +1,5 @@
 black *args="":
-    uv run black --check src {{ args }}
+    uv run black --check src tests {{ args }}
 
 ruff *args="":
     uv run ruff check {{ args }}
@@ -14,8 +14,8 @@ check: black ruff type-check
     {{ just_executable() }} toml-sort --check
 
 fix:
-    uv run black src
-    uv run ruff check --fix src
+    uv run black src tests
+    uv run ruff check --fix src tests
     {{ just_executable() }} toml-sort --in-place
 
 release:
@@ -30,3 +30,8 @@ run *args="":
 
 @html:
     just run django.views.generic.FormView --renderer html --output output
+
+test *args="":
+    uv run -m coverage run --module pytest tests {{ args }}
+    -uv run -m coverage report
+    uv run -m coverage html

pyproject.toml

@@ -38,22 +38,49 @@ Changelog = "https://github.com/ghickman/classify/blob/main/CHANGELOG.md"
 [dependency-groups]
 dev = [
   "black>=25.9.0,<26",
+  "coverage>=7.11.0",
   "django>=5.2.7",
+  "httpx>=0.28.1",
   "ipdb>=0.13.13",
   "nox>=2025.10.16",
   "nox-uv>=0.6.3",
+  "pytest>=8.4.2",
+  "pytest-env>=1.2.0",
   "ruff>=0.14.2",
   "toml-sort>=0.24.3",
   "ty>=0.0.1a24",
 ]
 
+[tool.coverage.html]
+show_contexts = true
+
+[tool.coverage.report]
+# fail_under = 100
+show_missing = true
+skip_covered = true
+
+[tool.coverage.run]
+branch = true
+relative_files = true
+dynamic_context = "test_function"
+omit = [
+  "src/classify/contrib/*",
+]
+source = ["src", "tests"]
+
+[tool.pytest.ini_options]
+addopts = "--tb=native"
+env = [
+  "TEST_MODE=1",
+]
+
 [tool.ruff]
 line-length = 88
 extend-exclude = [
   "src/classify/contrib/django/",
   "src/classify/templates",
 ]
-src = ["src"]
+src = ["src", "tests"]
 
 [tool.ruff.lint]
 select = ["ALL"]

src/classify/renderers/html.py

@@ -1,5 +1,6 @@
 import contextlib
 import functools
+import os
 import socketserver
 import tempfile
 import webbrowser
@@ -36,8 +37,10 @@ def _(empty: None) -> Generator:  # noqa: ARG001
 def serve_output(port: int) -> None:
     httpd = socketserver.TCPServer(("", port), Handler)
 
-    print(f"Serving on port: {port}")
-    webbrowser.open_new_tab(f"http://localhost:{port}/")
+    if not os.environ.get("TEST_MODE", None):
+        print(f"Serving on port: {port}")
+        webbrowser.open_new_tab(f"http://localhost:{port}/")
+
     httpd.serve_forever()
 
 

tests/__init__.py

@@ -0,0 +1,40 @@
+import pytest
+
+from classify.library import Attribute, Class, Line, Method
+
+
+def attribute(name, **kwargs):
+    return Attribute(
+        name=name,
+        object=kwargs.get("object", ""),
+        defining_class=kwargs.get("defining_class", ""),
+        value=kwargs.get("value", ""),
+    )
+
+
+def method(name, **kwargs):
+    return Method(
+        name=name,
+        docstring=kwargs.get("docstring", ""),
+        defining_class=kwargs.get("defining_class", ""),
+        arguments=kwargs.get("arguments", ""),
+        code=kwargs.get("code", ""),
+        lines=Line(start=42, total=7),
+    )
+
+
+@pytest.fixture
+def dummy_class():
+    return Class(
+        name="MyClass",
+        methods={
+            "one": [
+                method("one", defining_class="ParentClass"),
+                method("one", defining_class="MyClass"),
+            ]
+        },
+        docstring="",
+        ancestors=[],
+        parents=["ParentClass"],
+        attributes={},
+    )

tests/conftest.py

@@ -0,0 +1,78 @@
+import subprocess
+import tempfile
+import time
+from pathlib import Path
+
+import httpx
+import pytest
+
+from classify.renderers.html import resolve_path, to_html
+
+
+class DummyClass:
+    some_attr = 7
+
+    def one(self):
+        pass
+
+
+@pytest.fixture(scope="session")
+def classify_server():
+    ds_proc = subprocess.Popen(
+        [
+            "classify",
+            "tests.renderers.test_html.DummyClass",
+            "--renderer",
+            "html",
+            "--serve",
+            "--port",
+            "8008",
+        ],
+        stdout=subprocess.PIPE,
+        stderr=subprocess.STDOUT,
+    )
+
+    # Give the server time to start
+    time.sleep(1)
+
+    # Check it started successfully
+    assert not ds_proc.poll(), ds_proc.stdout.read().decode("utf-8")
+
+    yield ds_proc
+
+    # Shut it down at the end of the pytest session
+    ds_proc.terminate()
+
+
+def test_to_html(dummy_class):
+    with tempfile.TemporaryDirectory() as path:
+        path = Path(path)  # noqa: PLW2901
+        to_html(dummy_class, output_path=path, serve=False, port=8000)
+
+        output = path / "classify.html"
+        assert output.exists()
+
+        content = output.read_text()
+        assert dummy_class.name in content
+
+
+def test_to_html_and_serve(classify_server):  # noqa: ARG001
+    response = httpx.get("http://127.0.0.1:8008/")
+    assert response.status_code == 200  # noqa: PLR2004
+
+
+def test_resolve_path_with_path():
+    with tempfile.TemporaryDirectory() as parent:
+        path = Path(parent) / "testing"
+        assert not path.exists()
+
+        with resolve_path(path) as made_path:
+            assert made_path == path
+            assert made_path.exists()
+
+
+def test_resolve_path_without_path():
+    with resolve_path(None) as path:
+        assert path.exists()
+
+    assert not path.exists()

tests/renderers/__init__.py

@@ -0,0 +1,42 @@
+import pytest
+
+from classify.library import get_members
+
+
+class DummyParent:
+    def one(self):
+        print("one parent")
+
+    def three(self):
+        pass
+
+
+class DummyClass(DummyParent):
+    def __init__(self):
+        super().__init__()
+
+    def _internal(self):
+        pass
+
+    def one(self):
+        super().one()
+        print("one child")
+
+    def two(self):
+        pass
+
+
+@pytest.mark.parametrize(
+    ("cls", "expected"),
+    [
+        (DummyParent, ["__dict__", "__weakref__", "one", "three"]),
+        (DummyClass, ["__init__", "one", "two"]),
+    ],
+    ids=["parent", "child"],
+)
+def test_get_members(cls, expected):
+    members = get_members(cls)
+
+    names = [m[0] for m in members]
+
+    assert names == expected, names