doppler-cli 0.2.6__py3-none-any.whl

doppler_cli/tests/test_ports.py

@@ -0,0 +1,101 @@
+"""
+Tests for port auto-allocation.
+
+Covers:
+  - allocate() returns the requested number of ports
+  - Ports are in the valid range
+  - Ports are unique within a single allocation
+  - Pre-occupied ports (from fake state files) are skipped
+  - Requesting too many ports raises RuntimeError
+"""
+
+import json
+import pytest
+
+from doppler_cli import ports as ports_mod
+
+
+def _write_fake_state(chains_dir, chain_id: str, used_ports: list[int]):
+    chains_dir.mkdir(parents=True, exist_ok=True)
+    state = {
+        "id": chain_id,
+        "started": "2026-01-01T00:00:00+00:00",
+        "compose": "/tmp/fake.yml",
+        "blocks": [
+            {"name": "tone", "pid": 9999, "bind_port": p, "connect_port": None}
+            for p in used_ports
+        ],
+    }
+    (chains_dir / f"{chain_id}.json").write_text(json.dumps(state))
+
+
+# ---------------------------------------------------------------------------
+# Basic allocation
+# ---------------------------------------------------------------------------
+
+
+class TestAllocate:
+    def test_returns_correct_count(self, tmp_path, monkeypatch):
+        monkeypatch.setattr(ports_mod, "_CHAINS_DIR", tmp_path)
+        result = ports_mod.allocate(3)
+        assert len(result) == 3
+
+    def test_ports_unique(self, tmp_path, monkeypatch):
+        monkeypatch.setattr(ports_mod, "_CHAINS_DIR", tmp_path)
+        result = ports_mod.allocate(5)
+        assert len(set(result)) == 5
+
+    def test_ports_in_valid_range(self, tmp_path, monkeypatch):
+        monkeypatch.setattr(ports_mod, "_CHAINS_DIR", tmp_path)
+        result = ports_mod.allocate(4)
+        for p in result:
+            assert ports_mod._BASE_PORT <= p <= ports_mod._MAX_PORT
+
+    def test_single_port(self, tmp_path, monkeypatch):
+        monkeypatch.setattr(ports_mod, "_CHAINS_DIR", tmp_path)
+        result = ports_mod.allocate(1)
+        assert len(result) == 1
+        assert result[0] == ports_mod._BASE_PORT
+
+
+# ---------------------------------------------------------------------------
+# Skips occupied ports
+# ---------------------------------------------------------------------------
+
+
+class TestAllocateSkipsOccupied:
+    def test_skips_port_in_state_file(self, tmp_path, monkeypatch):
+        monkeypatch.setattr(ports_mod, "_CHAINS_DIR", tmp_path)
+        base = ports_mod._BASE_PORT
+        _write_fake_state(tmp_path, "aaa111", [base])
+        result = ports_mod.allocate(1)
+        assert base not in result
+
+    def test_skips_multiple_occupied(self, tmp_path, monkeypatch):
+        monkeypatch.setattr(ports_mod, "_CHAINS_DIR", tmp_path)
+        base = ports_mod._BASE_PORT
+        occupied = [base, base + 1, base + 2]
+        _write_fake_state(tmp_path, "bbb222", occupied)
+        result = ports_mod.allocate(2)
+        for p in occupied:
+            assert p not in result
+
+    def test_gracefully_skips_corrupt_state(self, tmp_path, monkeypatch):
+        monkeypatch.setattr(ports_mod, "_CHAINS_DIR", tmp_path)
+        (tmp_path / "corrupt.json").write_text("not json {{{")
+        # Should not raise — corrupt files are ignored
+        result = ports_mod.allocate(1)
+        assert len(result) == 1
+
+
+# ---------------------------------------------------------------------------
+# Exhaustion
+# ---------------------------------------------------------------------------
+
+
+class TestAllocateExhaustion:
+    def test_too_many_raises(self, tmp_path, monkeypatch):
+        monkeypatch.setattr(ports_mod, "_CHAINS_DIR", tmp_path)
+        capacity = ports_mod._MAX_PORT - ports_mod._BASE_PORT + 1
+        with pytest.raises(RuntimeError, match="No free ports"):
+            ports_mod.allocate(capacity + 1)

doppler_cli/tests/test_state.py

@@ -0,0 +1,213 @@
+"""
+Tests for chain state persistence.
+
+Covers:
+  - ChainState.save() writes a valid JSON file
+  - ChainState.load() round-trips correctly
+  - ChainState.delete() removes the file
+  - load() raises KeyError for missing chain
+  - list_chains() returns all saved chains
+  - list_chains() skips corrupt files silently
+  - pid_alive() returns False for dead PIDs
+  - stop_chain() calls the right signal and deletes state
+"""
+
+import json
+import os
+import signal
+
+import pytest
+
+from doppler_cli.state import (
+    BlockState,
+    ChainState,
+    list_chains,
+    pid_alive,
+    stop_chain,
+)
+
+
+def _make_chain(chain_id: str = "abc123") -> ChainState:
+    return ChainState(
+        id=chain_id,
+        started="2026-01-01T00:00:00+00:00",
+        compose="/tmp/fake.yml",
+        blocks=[
+            BlockState(name="tone", pid=11111, bind_port=5600),
+            BlockState(name="specan", pid=11112, connect_port=5600),
+        ],
+    )
+
+
+# ---------------------------------------------------------------------------
+# Save / load / delete
+# ---------------------------------------------------------------------------
+
+
+class TestPersistence:
+    def test_save_creates_file(self, tmp_path, monkeypatch):
+        from doppler_cli import state as state_mod
+
+        monkeypatch.setattr(state_mod, "_CHAINS_DIR", tmp_path)
+        chain = _make_chain()
+        chain.save()
+        assert (tmp_path / "abc123.json").exists()
+
+    def test_save_valid_json(self, tmp_path, monkeypatch):
+        from doppler_cli import state as state_mod
+
+        monkeypatch.setattr(state_mod, "_CHAINS_DIR", tmp_path)
+        _make_chain().save()
+        data = json.loads((tmp_path / "abc123.json").read_text())
+        assert data["id"] == "abc123"
+        assert len(data["blocks"]) == 2
+
+    def test_load_round_trips(self, tmp_path, monkeypatch):
+        from doppler_cli import state as state_mod
+
+        monkeypatch.setattr(state_mod, "_CHAINS_DIR", tmp_path)
+        original = _make_chain()
+        original.save()
+        loaded = ChainState.load("abc123")
+        assert loaded.id == original.id
+        assert loaded.started == original.started
+        assert loaded.blocks[0].name == "tone"
+        assert loaded.blocks[0].bind_port == 5600
+        assert loaded.blocks[1].connect_port == 5600
+
+    def test_load_missing_raises(self, tmp_path, monkeypatch):
+        from doppler_cli import state as state_mod
+
+        monkeypatch.setattr(state_mod, "_CHAINS_DIR", tmp_path)
+        with pytest.raises(KeyError):
+            ChainState.load("doesnotexist")
+
+    def test_delete_removes_file(self, tmp_path, monkeypatch):
+        from doppler_cli import state as state_mod
+
+        monkeypatch.setattr(state_mod, "_CHAINS_DIR", tmp_path)
+        chain = _make_chain()
+        chain.save()
+        chain.delete()
+        assert not (tmp_path / "abc123.json").exists()
+
+    def test_delete_missing_is_silent(self, tmp_path, monkeypatch):
+        from doppler_cli import state as state_mod
+
+        monkeypatch.setattr(state_mod, "_CHAINS_DIR", tmp_path)
+        chain = _make_chain()
+        chain.delete()  # never saved — should not raise
+
+
+# ---------------------------------------------------------------------------
+# list_chains
+# ---------------------------------------------------------------------------
+
+
+class TestListChains:
+    def test_empty_dir(self, tmp_path, monkeypatch):
+        from doppler_cli import state as state_mod
+
+        monkeypatch.setattr(state_mod, "_CHAINS_DIR", tmp_path)
+        assert list_chains() == []
+
+    def test_missing_dir(self, tmp_path, monkeypatch):
+        from doppler_cli import state as state_mod
+
+        monkeypatch.setattr(state_mod, "_CHAINS_DIR", tmp_path / "nonexistent")
+        assert list_chains() == []
+
+    def test_returns_all_chains(self, tmp_path, monkeypatch):
+        from doppler_cli import state as state_mod
+
+        monkeypatch.setattr(state_mod, "_CHAINS_DIR", tmp_path)
+        _make_chain("aaa111").save()
+        _make_chain("bbb222").save()
+        ids = {c.id for c in list_chains()}
+        assert ids == {"aaa111", "bbb222"}
+
+    def test_skips_corrupt_files(self, tmp_path, monkeypatch):
+        from doppler_cli import state as state_mod
+
+        monkeypatch.setattr(state_mod, "_CHAINS_DIR", tmp_path)
+        _make_chain("good000").save()
+        (tmp_path / "bad999.json").write_text("not json {{{")
+        chains = list_chains()
+        assert len(chains) == 1
+        assert chains[0].id == "good000"
+
+
+# ---------------------------------------------------------------------------
+# pid_alive
+# ---------------------------------------------------------------------------
+
+
+class TestPidAlive:
+    def test_own_pid_is_alive(self):
+        assert pid_alive(os.getpid()) is True
+
+    def test_dead_pid_returns_false(self):
+        # PID 1 exists but we won't have permission to signal it,
+        # which pid_alive treats as "not our process = alive" via
+        # PermissionError. Use a clearly invalid large PID instead.
+        assert pid_alive(999999999) is False
+
+
+# ---------------------------------------------------------------------------
+# stop_chain
+# ---------------------------------------------------------------------------
+
+
+class TestStopChain:
+    def test_sigterm_sent_and_state_deleted(self, tmp_path, monkeypatch):
+        from doppler_cli import state as state_mod
+
+        monkeypatch.setattr(state_mod, "_CHAINS_DIR", tmp_path)
+
+        signals_sent: list[tuple[int, int]] = []
+
+        def fake_kill(pid, sig):
+            signals_sent.append((pid, sig))
+
+        monkeypatch.setattr(os, "kill", fake_kill)
+        monkeypatch.setattr(state_mod, "pid_alive", lambda pid: True)
+
+        chain = _make_chain()
+        chain.save()
+        stop_chain(chain, kill=False)
+
+        assert (11111, signal.SIGTERM) in signals_sent
+        assert (11112, signal.SIGTERM) in signals_sent
+        assert not (tmp_path / "abc123.json").exists()
+
+    def test_sigkill_when_kill_true(self, tmp_path, monkeypatch):
+        from doppler_cli import state as state_mod
+
+        monkeypatch.setattr(state_mod, "_CHAINS_DIR", tmp_path)
+
+        signals_sent: list[tuple[int, int]] = []
+        monkeypatch.setattr(
+            os, "kill", lambda pid, sig: signals_sent.append((pid, sig))
+        )
+        monkeypatch.setattr(state_mod, "pid_alive", lambda pid: True)
+
+        chain = _make_chain()
+        chain.save()
+        stop_chain(chain, kill=True)
+
+        assert all(sig == signal.SIGKILL for _, sig in signals_sent)
+
+    def test_dead_pids_skipped(self, tmp_path, monkeypatch):
+        from doppler_cli import state as state_mod
+
+        monkeypatch.setattr(state_mod, "_CHAINS_DIR", tmp_path)
+        monkeypatch.setattr(state_mod, "pid_alive", lambda pid: False)
+
+        signals_sent: list = []
+        monkeypatch.setattr(os, "kill", lambda pid, sig: signals_sent.append(pid))
+
+        chain = _make_chain()
+        chain.save()
+        stop_chain(chain)
+
+        assert signals_sent == []

doppler_cli-0.2.6.dist-info/METADATA

@@ -0,0 +1,12 @@
+Metadata-Version: 2.3
+Name: doppler-cli
+Version: 0.2.6
+Summary: CLI for doppler-dsp signal processing pipelines
+Requires-Dist: doppler-dsp>=0.2.0
+Requires-Dist: doppler-specan>=0.2.0
+Requires-Dist: pydantic>=2.0
+Requires-Dist: pyyaml>=6.0
+Requires-Dist: rich>=13.0
+Requires-Python: >=3.12
+Project-URL: Homepage, https://doppler-dsp.github.io/doppler/
+Project-URL: Repository, https://github.com/doppler-dsp/doppler

doppler_cli-0.2.6.dist-info/RECORD

@@ -0,0 +1,22 @@
+doppler_cli/__init__.py,sha256=CiK-5hGYXfHw0j9uQotglcEdzkpsLXxRacguklaozmQ,87
+doppler_cli/__main__.py,sha256=xNdB_KhIWLgHNCXtLc5mMopkIHkPOp4OxTax7JyE2Vs,5150
+doppler_cli/blocks/__init__.py,sha256=yUKTxbmDXy92jnJ3DpTNZOMj6Y1giRWPgqLZNuSi4Gs,1924
+doppler_cli/blocks/fir.py,sha256=HBgWqisAaF3R3yg9m0ncua3HHhJjpDQVGaZ4mi8w6ZQ,755
+doppler_cli/blocks/specan.py,sha256=wsB1V2gecDaW-wBw6yWCn5aeg6xpLz2JCbJ3A2rumFQ,1432
+doppler_cli/blocks/tone.py,sha256=Beqfvs15VHHiFCwKPL_lU_a1f00NJvQG1DcP2TeCepI,1079
+doppler_cli/compose.py,sha256=Q59L9ksj3JUtXaSD5jFvuRXVWiiFnHDi3-oW0l5mdow,6399
+doppler_cli/dopplerfile.py,sha256=EuL9l9TG8wdaZ_bQj3tA5KDQ6YgvlaLMO26av_UKmVE,5690
+doppler_cli/ports.py,sha256=b_H5gVEzkPksayyNaUGrUsmD6T_UhfQUxvPQWYoamkE,1363
+doppler_cli/ps.py,sha256=Qs_YC-PYP68Cb05snNr1mnFeuMWSk6SIXgabzNOtFhg,3492
+doppler_cli/source.py,sha256=JTp4L21HM8y9uiK-DnptAIDPL4FvN58k8jZDywHfUdw,3062
+doppler_cli/state.py,sha256=PKCQWp1strejTDRJKkGtdYNAP6s-I9XXEQoXAqkmHT0,3006
+doppler_cli/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+doppler_cli/tests/test_blocks.py,sha256=HTwv4UP8Th3aJ46ue__SXJIEwOudnZ4Zv_0voLKrXRs,5622
+doppler_cli/tests/test_compose.py,sha256=Bim4hnQvJ4PkutE_W9GzHwweSp7llwuzB9m3Y3yZ-JQ,6049
+doppler_cli/tests/test_dopplerfile.py,sha256=abc8bhNfBfJPValnkc3l0BaiUp8WWWs10mjrh5sX4BU,10450
+doppler_cli/tests/test_ports.py,sha256=62WEO7F9w8Pu32-4OIU4Ui5xq66xesZoVWr9_DNLHTA,3638
+doppler_cli/tests/test_state.py,sha256=KDu9Z5VePJHuFTPBC6Dt5evFcrbB6MoslN3z6rUfwx0,6985
+doppler_cli-0.2.6.dist-info/WHEEL,sha256=bEhYrD-rjlF0iRRHiAnfJ0mEjMsRwm29hhDD7yRgWCY,80
+doppler_cli-0.2.6.dist-info/entry_points.txt,sha256=VZVDiNdRCzdToUoaQJfkcU6KgreQwiZV8VCY0m1CK2I,96
+doppler_cli-0.2.6.dist-info/METADATA,sha256=W0Jicw5SekZ01yi5gT0-Y73YRvHMZaCI2wwopp1c2Qk,415
+doppler_cli-0.2.6.dist-info/RECORD,,

doppler_cli-0.2.6.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: uv 0.11.3
+Root-Is-Purelib: true
+Tag: py3-none-any

doppler_cli-0.2.6.dist-info/entry_points.txt

@@ -0,0 +1,4 @@
+[console_scripts]
+doppler = doppler_cli.__main__:main
+doppler-source = doppler_cli.source:main
+