workstream-cli 0.0.1__py3-none-any.whl

workstream/test/test_config.py

@@ -0,0 +1,54 @@
+"""Tests for workstream.config — auto_commit_notes field."""
+
+from pathlib import Path
+
+import yaml
+
+from workstream.config import Config, _parse_config_data, save_config
+
+
+def test_auto_commit_notes_default_true():
+    """When absent from YAML, auto_commit_notes defaults to True."""
+    data = {'repos': [], 'notes_dirs': [], 'repo_dirs': [], 'llm': 'auto'}
+    config = _parse_config_data(data, '/tmp/ws')
+    assert config.auto_commit_notes is True
+
+
+def test_auto_commit_notes_explicit_false():
+    """YAML `auto_commit_notes: false` parses to False."""
+    data = {
+        'repos': [], 'notes_dirs': [], 'repo_dirs': [],
+        'llm': 'auto', 'auto_commit_notes': False,
+    }
+    config = _parse_config_data(data, '/tmp/ws')
+    assert config.auto_commit_notes is False
+
+
+def test_auto_commit_notes_explicit_true():
+    """YAML `auto_commit_notes: true` parses to True."""
+    data = {
+        'repos': [], 'notes_dirs': [], 'repo_dirs': [],
+        'llm': 'auto', 'auto_commit_notes': True,
+    }
+    config = _parse_config_data(data, '/tmp/ws')
+    assert config.auto_commit_notes is True
+
+
+def test_save_config_includes_auto_commit_notes(tmp_path: Path):
+    """save_config persists auto_commit_notes to the project config file."""
+    ws_dir = tmp_path / '_Workstreams'
+    ws_dir.mkdir()
+    config = Config(workstreams_dir=str(ws_dir), auto_commit_notes=False)
+    save_config(config)
+    written = yaml.safe_load((ws_dir / 'config.yaml').read_text())
+    assert written['auto_commit_notes'] is False
+
+
+def test_save_config_auto_commit_notes_true(tmp_path: Path):
+    """save_config persists auto_commit_notes=True."""
+    ws_dir = tmp_path / '_Workstreams'
+    ws_dir.mkdir()
+    config = Config(workstreams_dir=str(ws_dir), auto_commit_notes=True)
+    save_config(config)
+    written = yaml.safe_load((ws_dir / 'config.yaml').read_text())
+    assert written['auto_commit_notes'] is True

workstream/test/test_focus_artifact.py

@@ -0,0 +1,211 @@
+"""Tests for workstream.focus_artifact — focus doc I/O and lookup."""
+
+from __future__ import annotations
+
+from datetime import date, timedelta
+from pathlib import Path
+
+import pytest
+
+from workstream.focus_artifact import (
+    FocusArtifact,
+    focus_dir,
+    load_current_week_focus,
+    load_focus,
+    load_latest_focus,
+    save_focus,
+)
+
+
+# ── FocusArtifact construction ────────────────────────────────────────
+
+def test_focus_artifact_defaults():
+    a = FocusArtifact()
+    assert a.date == date.today().isoformat()
+    assert a.type == 'focus'
+    assert a.expires == ''
+    assert a.supersedes == ''
+    assert a.priorities == []
+    assert a.blocked == []
+    assert a.agent_actionable == []
+    assert a.context == ''
+    assert a.body == ''
+    assert a.source_path is None
+
+
+def test_focus_artifact_full():
+    a = FocusArtifact(
+        focus_date='2026-04-01',
+        expires='2026-04-07',
+        supersedes='2026-03-25',
+        priorities=[{'id': 'ws-1', 'reason': 'Demo Wednesday'}],
+        blocked=[{'id': 'ws-2', 'on': 'API access'}],
+        agent_actionable=[{'id': 'ws-3', 'action': 'Needs /plan'}],
+        context='Demo day is Wednesday.',
+        body='## Focus\n\nPolish the demo.',
+    )
+    assert a.date == '2026-04-01'
+    assert a.expires == '2026-04-07'
+    assert len(a.priorities) == 1
+    assert a.priorities[0]['reason'] == 'Demo Wednesday'
+
+
+# ── frontmatter_dict ─────────────────────────────────────────────────
+
+def test_frontmatter_dict_minimal():
+    a = FocusArtifact(focus_date='2026-04-01')
+    d = a.frontmatter_dict()
+    assert d == {'date': '2026-04-01', 'type': 'focus'}
+
+
+def test_frontmatter_dict_full():
+    a = FocusArtifact(
+        focus_date='2026-04-01',
+        expires='2026-04-07',
+        priorities=[{'id': 'ws-1', 'reason': 'urgent'}],
+        context='Big demo.',
+    )
+    d = a.frontmatter_dict()
+    assert d['expires'] == '2026-04-07'
+    assert d['priorities'] == [{'id': 'ws-1', 'reason': 'urgent'}]
+    assert d['context'] == 'Big demo.'
+    assert 'blocked' not in d  # empty list omitted
+    assert 'supersedes' not in d  # empty string omitted
+
+
+# ── Round-trip: save + load ──────────────────────────────────────────
+
+def test_save_and_load_roundtrip(tmp_path: Path):
+    ws_dir = tmp_path / 'ws'
+    ws_dir.mkdir()
+
+    original = FocusArtifact(
+        focus_date='2026-04-01',
+        expires='2026-04-07',
+        supersedes='2026-03-25',
+        priorities=[{'id': 'ws-1', 'reason': 'Demo'}],
+        blocked=[{'id': 'ws-2', 'on': 'Waiting for key'}],
+        agent_actionable=[{'id': 'ws-3', 'action': 'Plan session'}],
+        context='Demo day.',
+        body='## Focus\n\nShip it.',
+    )
+    path = save_focus(original, ws_dir)
+    assert path.exists()
+    assert path.name == '2026-04-01.md'
+
+    loaded = load_focus(path)
+    assert loaded.date == '2026-04-01'
+    assert loaded.type == 'focus'
+    assert loaded.expires == '2026-04-07'
+    assert loaded.supersedes == '2026-03-25'
+    assert loaded.priorities == [{'id': 'ws-1', 'reason': 'Demo'}]
+    assert loaded.blocked == [{'id': 'ws-2', 'on': 'Waiting for key'}]
+    assert loaded.agent_actionable == [{'id': 'ws-3', 'action': 'Plan session'}]
+    assert loaded.context == 'Demo day.'
+    assert '## Focus' in loaded.body
+
+
+# ── focus_dir ────────────────────────────────────────────────────────
+
+def test_focus_dir_creates(tmp_path: Path):
+    ws_dir = tmp_path / 'ws'
+    ws_dir.mkdir()
+    fdir = focus_dir(ws_dir)
+    assert fdir.is_dir()
+    assert fdir == ws_dir / 'focus'
+
+
+# ── load_latest_focus ────────────────────────────────────────────────
+
+def test_load_latest_focus_empty(tmp_path: Path):
+    ws_dir = tmp_path / 'ws'
+    ws_dir.mkdir()
+    assert load_latest_focus(ws_dir) is None
+
+
+def test_load_latest_focus_picks_newest(tmp_path: Path):
+    ws_dir = tmp_path / 'ws'
+    ws_dir.mkdir()
+
+    save_focus(FocusArtifact(focus_date='2026-03-25', body='old'), ws_dir)
+    save_focus(FocusArtifact(focus_date='2026-04-01', body='new'), ws_dir)
+
+    latest = load_latest_focus(ws_dir)
+    assert latest is not None
+    assert latest.date == '2026-04-01'
+    assert latest.body == 'new'
+
+
+# ── load_current_week_focus ──────────────────────────────────────────
+
+def test_current_week_focus_empty(tmp_path: Path):
+    ws_dir = tmp_path / 'ws'
+    ws_dir.mkdir()
+    assert load_current_week_focus(ws_dir) is None
+
+
+def test_current_week_focus_finds_this_week(tmp_path: Path):
+    ws_dir = tmp_path / 'ws'
+    ws_dir.mkdir()
+
+    today = date.today()
+    today_str = today.isoformat()
+    # Doc created today, expires end of week
+    monday = today - timedelta(days=today.weekday())
+    expires = (monday + timedelta(days=6)).isoformat()
+
+    save_focus(FocusArtifact(
+        focus_date=today_str,
+        expires=expires,
+        body='this week',
+    ), ws_dir)
+
+    found = load_current_week_focus(ws_dir)
+    assert found is not None
+    assert found.date == today_str
+
+
+def test_current_week_focus_skips_expired(tmp_path: Path):
+    ws_dir = tmp_path / 'ws'
+    ws_dir.mkdir()
+
+    today = date.today()
+    # Doc from today but expired yesterday
+    yesterday = (today - timedelta(days=1)).isoformat()
+    save_focus(FocusArtifact(
+        focus_date=today.isoformat(),
+        expires=yesterday,
+        body='expired',
+    ), ws_dir)
+
+    assert load_current_week_focus(ws_dir) is None
+
+
+def test_current_week_focus_skips_last_week(tmp_path: Path):
+    ws_dir = tmp_path / 'ws'
+    ws_dir.mkdir()
+
+    today = date.today()
+    last_week = (today - timedelta(days=8)).isoformat()
+    save_focus(FocusArtifact(
+        focus_date=last_week,
+        body='old week',
+    ), ws_dir)
+
+    assert load_current_week_focus(ws_dir) is None
+
+
+def test_current_week_focus_no_expiry_is_valid(tmp_path: Path):
+    """A focus doc with no expires field never expires."""
+    ws_dir = tmp_path / 'ws'
+    ws_dir.mkdir()
+
+    today = date.today().isoformat()
+    save_focus(FocusArtifact(
+        focus_date=today,
+        body='no expiry',
+    ), ws_dir)
+
+    found = load_current_week_focus(ws_dir)
+    assert found is not None
+    assert found.body == 'no expiry'

workstream/test/test_git.py

@@ -0,0 +1,88 @@
+"""Tests for workstream.git helpers."""
+
+import subprocess
+from pathlib import Path
+
+from workstream.git import git_add, git_commit, modified_files
+
+
+def test_modified_files_empty_repo(tmp_path: Path):
+    """Clean repo returns no modified files."""
+    subprocess.run(['git', 'init'], cwd=tmp_path, capture_output=True)
+    subprocess.run(['git', 'config', 'user.email', 'test@test.com'], cwd=tmp_path, capture_output=True)
+    subprocess.run(['git', 'config', 'user.name', 'Test'], cwd=tmp_path, capture_output=True)
+    # Create and commit a file so the repo isn't empty
+    (tmp_path / 'readme.md').write_text('hello')
+    subprocess.run(['git', 'add', '.'], cwd=tmp_path, capture_output=True)
+    subprocess.run(['git', 'commit', '-m', 'init'], cwd=tmp_path, capture_output=True)
+    assert modified_files(tmp_path) == []
+
+
+def test_modified_files_with_changes(tmp_path: Path):
+    """Modified and untracked files are returned."""
+    subprocess.run(['git', 'init'], cwd=tmp_path, capture_output=True)
+    subprocess.run(['git', 'config', 'user.email', 'test@test.com'], cwd=tmp_path, capture_output=True)
+    subprocess.run(['git', 'config', 'user.name', 'Test'], cwd=tmp_path, capture_output=True)
+    (tmp_path / 'tracked.md').write_text('original')
+    subprocess.run(['git', 'add', '.'], cwd=tmp_path, capture_output=True)
+    subprocess.run(['git', 'commit', '-m', 'init'], cwd=tmp_path, capture_output=True)
+
+    # Modify tracked file and create untracked file
+    (tmp_path / 'tracked.md').write_text('modified')
+    (tmp_path / 'new.md').write_text('untracked')
+
+    result = modified_files(tmp_path)
+    assert 'tracked.md' in result
+    assert 'new.md' in result
+
+
+def test_modified_files_non_repo(tmp_path: Path):
+    """Non-git directory returns empty list."""
+    assert modified_files(tmp_path) == []
+
+
+
+def _init_repo(path: Path) -> None:
+    """Initialize a git repo with user config and an initial commit."""
+    subprocess.run(['git', 'init'], cwd=path, capture_output=True)
+    subprocess.run(['git', 'config', 'user.email', 'test@test.com'], cwd=path, capture_output=True)
+    subprocess.run(['git', 'config', 'user.name', 'Test'], cwd=path, capture_output=True)
+    (path / 'readme.md').write_text('init')
+    subprocess.run(['git', 'add', '.'], cwd=path, capture_output=True)
+    subprocess.run(['git', 'commit', '-m', 'init'], cwd=path, capture_output=True)
+
+
+def test_git_add_stages_files(tmp_path: Path):
+    """git_add stages specified files."""
+    _init_repo(tmp_path)
+    (tmp_path / 'a.md').write_text('content a')
+    (tmp_path / 'b.md').write_text('content b')
+    assert git_add(tmp_path, [tmp_path / 'a.md', tmp_path / 'b.md'])
+    # Verify staged
+    result = subprocess.run(
+        ['git', 'diff', '--cached', '--name-only'],
+        cwd=tmp_path, capture_output=True, text=True,
+    )
+    staged = result.stdout.strip().splitlines()
+    assert 'a.md' in staged
+    assert 'b.md' in staged
+
+
+def test_git_commit_creates_commit(tmp_path: Path):
+    """git_commit creates a commit from staged changes."""
+    _init_repo(tmp_path)
+    (tmp_path / 'note.md').write_text('content')
+    git_add(tmp_path, [tmp_path / 'note.md'])
+    assert git_commit(tmp_path, 'notes: 2026-03-31')
+    # Verify commit message
+    result = subprocess.run(
+        ['git', 'log', '-1', '--format=%s'],
+        cwd=tmp_path, capture_output=True, text=True,
+    )
+    assert result.stdout.strip() == 'notes: 2026-03-31'
+
+
+def test_git_commit_nothing_staged(tmp_path: Path):
+    """git_commit with nothing staged returns False."""
+    _init_repo(tmp_path)
+    assert git_commit(tmp_path, 'empty') is False

workstream/test/test_heuristics.py

@@ -0,0 +1,136 @@
+"""Tests for heuristic helpers: branch_is_merged, PlanRef.signal, find_duplicate_repo_names, resolve_repo_path."""
+
+import subprocess
+from pathlib import Path
+
+from workstream.config import Config
+from workstream.markdown import _render_plan_line, parse_plans, set_section
+from workstream.models import PlanRef
+from workstream.repo_discovery import find_duplicate_repo_names
+
+
+# ---------------------------------------------------------------------------
+# PlanRef.signal
+# ---------------------------------------------------------------------------
+
+
+def test_plan_ref_signal_default():
+    ref = PlanRef(repo='myapp', path='plan.md', status='active', title='Test', date='2026-03-30')
+    assert ref.signal == ''
+
+
+def test_plan_ref_signal_set():
+    ref = PlanRef(repo='myapp', path='plan.md', status='active', title='Test', date='2026-03-30', signal='likely-implemented')
+    assert ref.signal == 'likely-implemented'
+
+
+def test_plan_render_with_signal():
+    ref = PlanRef(repo='myapp', path='plan.md', status='active', title='Test', date='2026-03-30', signal='stale')
+    line = _render_plan_line(ref)
+    assert '[stale]' in line
+    assert line.startswith('- myapp')
+
+
+def test_plan_render_without_signal():
+    ref = PlanRef(repo='myapp', path='plan.md', status='active', title='Test', date='2026-03-30')
+    line = _render_plan_line(ref)
+    assert '[' not in line
+
+
+def test_plan_parse_roundtrip_with_signal():
+    ref = PlanRef(repo='myapp', path='plan.md', status='active', title='Test', date='2026-03-30', signal='stale')
+    line = _render_plan_line(ref)
+    body = set_section('', 'Plans', '<!-- Updated by ws sweep -->\n' + line)
+    parsed = parse_plans(body)
+    assert len(parsed) == 1
+    assert parsed[0].signal == 'stale'
+
+
+# ---------------------------------------------------------------------------
+# find_duplicate_repo_names
+# ---------------------------------------------------------------------------
+
+
+def test_find_duplicate_repo_names_none(tmp_path: Path):
+    parent = tmp_path / 'repos'
+    (parent / 'repo1' / '.git').mkdir(parents=True)
+    (parent / 'repo2' / '.git').mkdir(parents=True)
+    result = find_duplicate_repo_names([str(parent)])
+    assert result == {}
+
+
+def test_find_duplicate_repo_names_found(tmp_path: Path):
+    parent1 = tmp_path / 'dir1'
+    parent2 = tmp_path / 'dir2'
+    (parent1 / 'myrepo' / '.git').mkdir(parents=True)
+    (parent2 / 'myrepo' / '.git').mkdir(parents=True)
+    result = find_duplicate_repo_names([str(parent1), str(parent2)])
+    assert 'myrepo' in result
+    assert len(result['myrepo']) == 2
+
+
+# ---------------------------------------------------------------------------
+# resolve_repo_path duplicate warning
+# ---------------------------------------------------------------------------
+
+
+def test_resolve_repo_path_warns_on_duplicate(tmp_path: Path, capsys):
+    parent1 = tmp_path / 'dir1'
+    parent2 = tmp_path / 'dir2'
+    (parent1 / 'myrepo' / '.git').mkdir(parents=True)
+    (parent2 / 'myrepo' / '.git').mkdir(parents=True)
+    config = Config(repo_dirs=[str(parent1), str(parent2)])
+    result = config.resolve_repo_path('myrepo')
+    assert result is not None
+    captured = capsys.readouterr()
+    assert 'warning' in captured.err
+    assert 'myrepo' in captured.err
+
+
+# ---------------------------------------------------------------------------
+# branch_is_merged (real temporary git repos)
+# ---------------------------------------------------------------------------
+
+
+def _init_git_repo(path: Path) -> None:
+    """Create a git repo at *path* with one initial commit on main."""
+    path.mkdir(exist_ok=True)
+    subprocess.run(['git', 'init', '-b', 'main'], cwd=path, capture_output=True, check=True)
+    subprocess.run(['git', 'config', 'user.email', 'test@test.com'], cwd=path, capture_output=True, check=True)
+    subprocess.run(['git', 'config', 'user.name', 'Test'], cwd=path, capture_output=True, check=True)
+    (path / 'file.txt').write_text('initial')
+    subprocess.run(['git', 'add', '.'], cwd=path, capture_output=True, check=True)
+    subprocess.run(['git', 'commit', '-m', 'init'], cwd=path, capture_output=True, check=True)
+
+
+def test_branch_is_merged_in_real_repo(tmp_path: Path):
+    from workstream.git import branch_is_merged
+
+    repo = tmp_path / 'repo'
+    _init_git_repo(repo)
+
+    # Create and merge a feature branch
+    subprocess.run(['git', 'checkout', '-b', 'feature'], cwd=repo, capture_output=True, check=True)
+    (repo / 'feature.txt').write_text('feature')
+    subprocess.run(['git', 'add', '.'], cwd=repo, capture_output=True, check=True)
+    subprocess.run(['git', 'commit', '-m', 'feature'], cwd=repo, capture_output=True, check=True)
+    subprocess.run(['git', 'checkout', 'main'], cwd=repo, capture_output=True, check=True)
+    subprocess.run(['git', 'merge', 'feature'], cwd=repo, capture_output=True, check=True)
+
+    assert branch_is_merged(repo, 'feature', 'main') is True
+
+
+def test_branch_is_merged_not_merged(tmp_path: Path):
+    from workstream.git import branch_is_merged
+
+    repo = tmp_path / 'repo'
+    _init_git_repo(repo)
+
+    # Feature branch with commit not merged into main
+    subprocess.run(['git', 'checkout', '-b', 'feature'], cwd=repo, capture_output=True, check=True)
+    (repo / 'feature.txt').write_text('feature')
+    subprocess.run(['git', 'add', '.'], cwd=repo, capture_output=True, check=True)
+    subprocess.run(['git', 'commit', '-m', 'feature'], cwd=repo, capture_output=True, check=True)
+    subprocess.run(['git', 'checkout', 'main'], cwd=repo, capture_output=True, check=True)
+
+    assert branch_is_merged(repo, 'feature', 'main') is False