1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
|
import os
import tempfile
import unittest
from claude_logbook import memory, sessions
from .fixtures import (
memory_tree, simple_tree, write_index, write_memory,
)
class MemoryCase(unittest.TestCase):
"""Each test runs against a fake ~/.claude/projects."""
def setUp(self):
self._tmp = tempfile.TemporaryDirectory()
self.addCleanup(self._tmp.cleanup)
self.root = os.path.join(self._tmp.name, "projects")
os.makedirs(self.root)
def load_items(self):
"""Sessions + memories of the tree, as the CLI sees them."""
ss = sessions.load_sessions(root=self.root, use_cache=False)
return ss, memory.load_memories(ss, root=self.root)
class TestParsing(MemoryCase):
def test_reads_frontmatter_and_body(self):
path = write_memory(self.root, "-home-u-proj", "una",
body="el cuerpo", desc="qué es", kind="feedback",
origin="abc123")
m = memory.read_memory(path, "-home-u-proj")
self.assertEqual(m["name"], "una")
self.assertEqual(m["desc"], "qué es")
self.assertEqual(m["ty"], "feedback")
self.assertEqual(m["src"], "abc123")
self.assertEqual(m["body"], "el cuerpo")
def test_quoted_description_loses_the_escapes(self):
# Claude writes the description as a YAML string when it contains quotes.
path = write_memory(self.root, "-home-u-proj", "q",
desc=r'"la máquina \"legion\" y algo"')
self.assertEqual(memory.read_memory(path, "-home-u-proj")["desc"],
'la máquina "legion" y algo')
def test_no_frontmatter_falls_back_to_the_file_name(self):
path = write_memory(self.root, "-home-u-proj", "pelada",
body="solo texto", frontmatter=False)
m = memory.read_memory(path, "-home-u-proj")
self.assertEqual(m["name"], "pelada")
self.assertEqual(m["ty"], "—")
self.assertEqual(m["body"], "solo texto")
def test_collects_links_without_repeats(self):
path = write_memory(self.root, "-home-u-proj", "l",
body="[[uno]] y [[dos]] y otra vez [[uno]]")
self.assertEqual(memory.read_memory(path, "-home-u-proj")["ln"],
["dos", "uno"])
class TestLoading(MemoryCase):
def test_resolves_the_project_path_from_the_sessions(self):
simple_tree(self.root)
memory_tree(self.root)
_, mems = self.load_items()
deploy = next(m for m in mems if m["name"] == "deploy-docker")
self.assertEqual(deploy["p"], "/home/u/proj")
def test_no_sessions_keeps_the_encoded_name(self):
# It cannot be reversed: "/" and "." are both encoded as "-".
memory_tree(self.root)
_, mems = self.load_items()
self.assertEqual(
next(m for m in mems if m["name"] == "deploy-docker")["p"],
"-home-u-proj")
def test_ignores_empty_memory_directories(self):
memory_tree(self.root)
_, mems = self.load_items()
self.assertNotIn("-home-u-vacio", {m["project_dir"] for m in mems})
def test_marks_what_is_in_the_index(self):
memory_tree(self.root)
_, mems = self.load_items()
by_name = {m["name"]: m for m in mems}
self.assertTrue(by_name["deploy-docker"]["ix"])
self.assertFalse(by_name["suelta"]["ix"])
self.assertTrue(by_name["suelta"]["hix"])
self.assertFalse(by_name["sin-indice"]["hix"])
def test_sorts_by_date_descending(self):
memory_tree(self.root)
_, mems = self.load_items()
dates = [m["l"] for m in mems]
self.assertEqual(dates, sorted(dates, reverse=True))
def test_public_records_drops_internal_keys(self):
memory_tree(self.root)
_, mems = self.load_items()
for m in memory.public_records(mems):
self.assertNotIn("project_dir", m)
# The original is left untouched.
self.assertIn("project_dir", mems[0])
class TestFilters(MemoryCase):
def setUp(self):
super().setUp()
simple_tree(self.root)
memory_tree(self.root)
_, self.mems = self.load_items()
def test_by_type(self):
r = memory.apply_filters(self.mems, kind="reference")
self.assertEqual([m["name"] for m in r], ["roles-db"])
def test_by_project(self):
r = memory.apply_filters(self.mems, project="/home/u/proj")
self.assertNotIn("sin-indice", [m["name"] for m in r])
def test_query_reaches_the_body(self):
r = memory.apply_filters(self.mems, query="make up")
self.assertEqual([m["name"] for m in r], ["deploy-docker"])
def test_query_also_checks_the_description(self):
r = memory.apply_filters(self.mems, query="huérfana")
self.assertEqual([m["name"] for m in r], ["suelta"])
class TestPick(MemoryCase):
def setUp(self):
super().setUp()
memory_tree(self.root)
_, self.mems = self.load_items()
def test_by_index(self):
self.assertEqual(memory.pick(self.mems, "1"), self.mems[0])
def test_index_out_of_range(self):
with self.assertRaises(sessions.SessionError):
memory.pick(self.mems, "99")
def test_by_prefix(self):
self.assertEqual(memory.pick(self.mems, "deploy")["name"], "deploy-docker")
def test_falls_back_to_substring(self):
self.assertEqual(memory.pick(self.mems, "docker")["name"], "deploy-docker")
def test_sin_coincidencias(self):
with self.assertRaises(sessions.SessionError):
memory.pick(self.mems, "nada-que-ver")
def test_ambiguous(self):
write_memory(self.root, "-home-u-proj", "deploy-otro")
_, mems = self.load_items()
with self.assertRaises(sessions.SessionError) as ctx:
memory.pick(mems, "deploy")
self.assertIn("ambiguo", str(ctx.exception))
class TestAudit(MemoryCase):
def setUp(self):
super().setUp()
simple_tree(self.root)
memory_tree(self.root)
self.ss, self.mems = self.load_items()
self.report = memory.audit(self.mems, self.ss, root=self.root)
def test_project_without_index(self):
self.assertEqual([m["name"] for m in self.report["sin_indice"]],
["sin-indice"])
def test_memory_outside_the_index(self):
self.assertEqual([m["name"] for m in self.report["sin_listar"]],
["suelta"])
def test_index_entry_without_file(self):
self.assertEqual([n for _, n in self.report["indice_fantasma"]],
["borrada-hace-rato"])
def test_broken_link(self):
broken = [link for _, link in self.report["enlaces_rotos"]]
self.assertEqual(broken, ["no-existe"]) # [[roles-db]] does resolve
def test_lost_origin_session(self):
# deploy-docker points to a session that exists; sin-indice does not.
self.assertEqual([m["name"] for m in self.report["origen_perdido"]],
["sin-indice"])
def test_consistent_tree_reports_nothing(self):
is_clean = os.path.join(self._tmp.name, "limpio")
os.makedirs(is_clean)
write_memory(is_clean, "-p", "sola", body="sin enlaces")
write_index(is_clean, "-p", ["sola"])
mems = memory.load_memories([], root=is_clean)
self.assertEqual(memory.audit_total(memory.audit(mems, [], root=is_clean)), 0)
class TestDeletion(MemoryCase):
def setUp(self):
super().setUp()
memory_tree(self.root)
_, self.mems = self.load_items()
def by_name(self, name):
return next(m for m in self.mems if m["name"] == name)
def test_deletes_the_file_and_unindexes_it(self):
m = self.by_name("deploy-docker")
self.assertTrue(memory.delete(m, root=self.root))
self.assertFalse(os.path.exists(memory.memory_path(m, self.root)))
self.assertNotIn("deploy-docker",
memory.read_index("-home-u-proj", self.root))
def test_unindexing_leaves_other_lines_alone(self):
memory.unindex(self.by_name("deploy-docker"), root=self.root)
self.assertIn("roles-db", memory.read_index("-home-u-proj", self.root))
def test_delete_one_that_was_not_indexed(self):
m = self.by_name("suelta")
self.assertFalse(memory.delete(m, root=self.root))
self.assertFalse(os.path.exists(memory.memory_path(m, self.root)))
def test_delete_without_memory_md(self):
m = self.by_name("sin-indice")
self.assertFalse(memory.delete(m, root=self.root))
self.assertFalse(os.path.exists(memory.memory_path(m, self.root)))
if __name__ == "__main__":
unittest.main()
|
