+19

tests/test_cluster.py

reviewed

··· 191 191 assert "This insight should NOT appear" not in result 192 192 193 193 194 194 + def test_load_entries_from_toplevel_segment(tmp_path, monkeypatch): 195 195 + """_load_entries_from_segment resolves the day for top-level segment dirs.""" 196 196 + monkeypatch.setenv("_SOLSTONE_JOURNAL_OVERRIDE", str(tmp_path)) 197 197 + day_dir = day_path("20240101") 198 198 + segment = day_dir / "100000_300" 199 199 + segment.mkdir() 200 200 + 201 201 + mod = importlib.import_module("think.cluster") 202 202 + 203 203 + entries = mod._load_entries_from_segment( 204 204 + str(segment), 205 205 + transcripts=True, 206 206 + percepts=False, 207 207 + agents=False, 208 208 + ) 209 209 + 210 210 + assert entries == [] 211 211 + 212 212 + 194 213 def test_cluster_range_with_agents(tmp_path, monkeypatch): 195 214 """Test cluster_range with agents source loads all *.md files.""" 196 215 monkeypatch.setenv("_SOLSTONE_JOURNAL_OVERRIDE", str(tmp_path))

+56 -1

tests/test_think_utils.py

reviewed

··· 13 13 import pytest 14 14 15 15 from think.entities import load_entity_names 16 16 - from think.utils import day_from_path, segment_key, setup_cli 16 16 + from think.utils import ( 17 17 + DEFAULT_STREAM, 18 18 + day_from_path, 19 19 + iter_segments, 20 20 + segment_key, 21 21 + setup_cli, 22 22 + ) 17 23 18 24 19 25 class TestDayFromPath: ··· 811 817 # Now it should exist 812 818 assert health_dir.exists() 813 819 assert (health_dir / "new_service.port").read_text() == "9999" 820 820 + 821 821 + 822 822 + class TestIterSegments: 823 823 + def test_skips_health_directory(self, tmp_path): 824 824 + """iter_segments does not return segments from health/ dirs.""" 825 825 + day_dir = tmp_path / "20240101" 826 826 + day_dir.mkdir() 827 827 + health_seg = day_dir / "health" / "120000_300" 828 828 + health_seg.mkdir(parents=True) 829 829 + normal_seg = day_dir / "default" / "130000_300" 830 830 + normal_seg.mkdir(parents=True) 831 831 + 832 832 + results = iter_segments(day_dir) 833 833 + stream_names = [r[0] for r in results] 834 834 + assert "health" not in stream_names 835 835 + assert "default" in stream_names 836 836 + 837 837 + def test_toplevel_segments_as_default_stream(self, tmp_path): 838 838 + """Top-level segment dirs are returned with _default stream name.""" 839 839 + day_dir = tmp_path / "20240101" 840 840 + day_dir.mkdir() 841 841 + toplevel_seg = day_dir / "143022_300" 842 842 + toplevel_seg.mkdir() 843 843 + normal_seg = day_dir / "default" / "150000_300" 844 844 + normal_seg.mkdir(parents=True) 845 845 + 846 846 + results = iter_segments(day_dir) 847 847 + assert len(results) == 2 848 848 + default_results = [(s, k, p) for s, k, p in results if s == DEFAULT_STREAM] 849 849 + assert len(default_results) == 1 850 850 + assert default_results[0][1] == "143022_300" 851 851 + normal_results = [(s, k, p) for s, k, p in results if s == "default"] 852 852 + assert len(normal_results) == 1 853 853 + 854 854 + def test_normal_stream_discovery_unchanged(self, tmp_path): 855 855 + """Normal stream/segment discovery still works correctly.""" 856 856 + day_dir = tmp_path / "20240101" 857 857 + day_dir.mkdir() 858 858 + (day_dir / "default" / "100000_300").mkdir(parents=True) 859 859 + (day_dir / "default" / "110000_300").mkdir(parents=True) 860 860 + (day_dir / "import.apple" / "120000_600").mkdir(parents=True) 861 861 + 862 862 + results = iter_segments(day_dir) 863 863 + assert len(results) == 3 864 864 + assert results[0][1] == "100000_300" 865 865 + assert results[1][1] == "110000_300" 866 866 + assert results[2][1] == "120000_600" 867 867 + assert results[0][0] == "default" 868 868 + assert results[2][0] == "import.apple"

+5 -3

think/cluster.py

reviewed

··· 12 12 from observe.screen import format_screen_text 13 13 14 14 from .streams import read_segment_stream 15 15 - from .utils import day_path 15 15 + from .utils import day_from_path, day_path 16 16 17 17 18 18 def _date_str(day_dir: str) -> str: ··· 645 645 List of entry dicts with timestamp, prefix, content, etc. 646 646 """ 647 647 segment_path_obj = Path(segment_dir) 648 648 - # Parent is stream dir; grandparent is day dir 649 649 - date_str = _date_str(str(segment_path_obj.parent.parent)) 648 648 + day_str = day_from_path(segment_path_obj) 649 649 + if day_str is None: 650 650 + raise ValueError(f"Cannot determine day from segment path: {segment_dir}") 651 651 + date_str = day_str 650 652 entries = _process_segment( 651 653 segment_path_obj, date_str, transcripts, percepts, agents 652 654 )

+6

think/utils.py

reviewed

··· 27 27 from media import MIME_TYPES 28 28 29 29 DATE_RE = re.compile(r"\d{8}") 30 30 + DEFAULT_STREAM = "_default" 30 31 31 32 32 33 def now_ms() -> int: ··· 275 276 results = [] 276 277 for entry in day_dir.iterdir(): 277 278 if not entry.is_dir(): 279 279 + continue 280 280 + if segment_key(entry.name) is not None: 281 281 + results.append((DEFAULT_STREAM, entry.name, entry)) 282 282 + continue 283 283 + if entry.name == "health": 278 284 continue 279 285 stream_name = entry.name 280 286 for seg_entry in entry.iterdir():