create-walle 0.1.0

package/template/wall-e/tests/api-walle.test.js

@@ -0,0 +1,222 @@
+const { describe, it, before, after } = require('node:test');
+const assert = require('node:assert/strict');
+const path = require('path');
+const fs = require('fs');
+const os = require('os');
+
+const brain = require('../brain.js');
+const api = require('../api-walle.js');
+
+let testDir;
+
+function setupTestDb() {
+  testDir = fs.mkdtempSync(path.join(os.tmpdir(), 'wall-e-api-test-'));
+  const testDbPath = path.join(testDir, 'test-brain.db');
+  brain.initDb(testDbPath);
+
+  // Point the API read DB at the same test database
+  const Database = require('better-sqlite3');
+  const readDb = new Database(testDbPath, { readonly: true });
+  readDb.pragma('journal_mode = WAL');
+  api._setReadDb(readDb);
+  api._setBrain(brain);
+}
+
+function teardownTestDb() {
+  try { brain.closeDb(); } catch (_) {}
+  if (testDir && fs.existsSync(testDir)) {
+    fs.rmSync(testDir, { recursive: true, force: true });
+  }
+}
+
+function makeParams(obj = {}) {
+  return new URLSearchParams(obj);
+}
+
+// Seed test data
+function seedData() {
+  brain.setOwner('name', 'Test User');
+
+  brain.insertMemory({
+    source: 'slack', source_id: 'msg-1', memory_type: 'message',
+    content: 'Hello world', timestamp: '2025-01-15T10:00:00Z'
+  });
+  brain.insertMemory({
+    source: 'email', source_id: 'email-1', memory_type: 'email',
+    content: 'Project update', timestamp: '2025-01-15T11:00:00Z'
+  });
+  brain.insertMemory({
+    source: 'slack', source_id: 'msg-2', memory_type: 'message',
+    content: 'Older message', timestamp: '2025-01-14T09:00:00Z'
+  });
+
+  brain.insertKnowledge({
+    category: 'preference', subject: 'user', predicate: 'prefers', object: 'dark mode'
+  });
+  brain.insertKnowledge({
+    category: 'fact', subject: 'project-x', predicate: 'uses', object: 'Node.js'
+  });
+
+  brain.insertPerson({ name: 'Alice', relationship: 'colleague' });
+  brain.insertPerson({ name: 'Bob', relationship: 'friend' });
+
+  brain.insertQuestion({
+    question_type: 'clarification', question: 'What is the deploy schedule?'
+  });
+  brain.insertQuestion({
+    question_type: 'preference', question: 'Which editor do you prefer?'
+  });
+}
+
+describe('WALL-E API handlers', () => {
+  before(() => {
+    setupTestDb();
+    seedData();
+  });
+  after(() => teardownTestDb());
+
+  describe('getStatus', () => {
+    it('returns object with owner and stats fields', () => {
+      const result = api.getStatus();
+      assert.equal(result.owner, 'Test User');
+      assert.ok(result.stats);
+      assert.equal(typeof result.stats.memory_count, 'number');
+      assert.equal(typeof result.stats.knowledge_count, 'number');
+      assert.equal(typeof result.stats.people_count, 'number');
+      assert.equal(typeof result.stats.pattern_count, 'number');
+      assert.equal(typeof result.stats.pending_question_count, 'number');
+      assert.equal(result.stats.memory_count, 3);
+      assert.equal(result.stats.knowledge_count, 2);
+      assert.equal(result.stats.people_count, 2);
+      assert.equal(result.stats.pending_question_count, 2);
+    });
+  });
+
+  describe('getKnowledgeList', () => {
+    it('returns all knowledge with no filters', () => {
+      const result = api.getKnowledgeList(makeParams());
+      assert.ok(Array.isArray(result));
+      assert.equal(result.length, 2);
+    });
+
+    it('filters by category', () => {
+      const result = api.getKnowledgeList(makeParams({ category: 'preference' }));
+      assert.equal(result.length, 1);
+      assert.equal(result[0].subject, 'user');
+    });
+
+    it('filters by subject', () => {
+      const result = api.getKnowledgeList(makeParams({ subject: 'project-x' }));
+      assert.equal(result.length, 1);
+      assert.equal(result[0].object, 'Node.js');
+    });
+
+    it('filters by status', () => {
+      const result = api.getKnowledgeList(makeParams({ status: 'active' }));
+      assert.equal(result.length, 2);
+    });
+
+    it('respects limit and offset', () => {
+      const result = api.getKnowledgeList(makeParams({ limit: '1', offset: '0' }));
+      assert.equal(result.length, 1);
+      const result2 = api.getKnowledgeList(makeParams({ limit: '1', offset: '1' }));
+      assert.equal(result2.length, 1);
+      assert.notEqual(result[0].id, result2[0].id);
+    });
+  });
+
+  describe('getMemoriesList', () => {
+    it('returns all memories with no filters', () => {
+      const result = api.getMemoriesList(makeParams());
+      assert.ok(Array.isArray(result));
+      assert.equal(result.length, 3);
+    });
+
+    it('filters by source', () => {
+      const result = api.getMemoriesList(makeParams({ source: 'slack' }));
+      assert.equal(result.length, 2);
+    });
+
+    it('filters by since', () => {
+      const result = api.getMemoriesList(makeParams({ since: '2025-01-15T00:00:00Z' }));
+      assert.equal(result.length, 2);
+    });
+
+    it('respects limit', () => {
+      const result = api.getMemoriesList(makeParams({ limit: '1' }));
+      assert.equal(result.length, 1);
+    });
+  });
+
+  describe('getPeopleList', () => {
+    it('returns all people', () => {
+      const result = api.getPeopleList();
+      assert.ok(Array.isArray(result));
+      assert.equal(result.length, 2);
+      // Ordered by name
+      assert.equal(result[0].name, 'Alice');
+      assert.equal(result[1].name, 'Bob');
+    });
+  });
+
+  describe('getTimeline', () => {
+    it('returns memories ordered by timestamp DESC', () => {
+      const result = api.getTimeline(makeParams());
+      assert.ok(Array.isArray(result));
+      assert.equal(result.length, 3);
+      // Most recent first
+      assert.ok(result[0].timestamp >= result[1].timestamp);
+      assert.ok(result[1].timestamp >= result[2].timestamp);
+    });
+
+    it('respects limit and offset', () => {
+      const result = api.getTimeline(makeParams({ limit: '2' }));
+      assert.equal(result.length, 2);
+      const result2 = api.getTimeline(makeParams({ limit: '2', offset: '2' }));
+      assert.equal(result2.length, 1);
+    });
+  });
+
+  describe('getQuestionsList', () => {
+    it('returns pending questions by default', () => {
+      const result = api.getQuestionsList(makeParams());
+      assert.ok(Array.isArray(result));
+      assert.equal(result.length, 2);
+      for (const q of result) {
+        assert.equal(q.status, 'pending');
+      }
+    });
+
+    it('filters by status', () => {
+      const result = api.getQuestionsList(makeParams({ status: 'answered' }));
+      assert.equal(result.length, 0);
+    });
+  });
+
+  describe('getStats', () => {
+    it('returns brain stats', () => {
+      const result = api.getStats();
+      assert.equal(typeof result.memory_count, 'number');
+      assert.equal(typeof result.knowledge_count, 'number');
+      assert.equal(typeof result.people_count, 'number');
+      assert.equal(typeof result.pattern_count, 'number');
+      assert.equal(typeof result.pending_question_count, 'number');
+      assert.equal(result.memory_count, 3);
+    });
+  });
+
+  describe('getBrief', () => {
+    it('returns null when no summary exists', () => {
+      const result = api.getBrief(makeParams({ date: '2025-01-15' }));
+      assert.equal(result, null);
+    });
+
+    it('returns summary when it exists', () => {
+      brain.insertDailySummary('2025-01-15', 'A productive day', { tasks: 5 });
+      const result = api.getBrief(makeParams({ date: '2025-01-15' }));
+      assert.ok(result);
+      assert.equal(result.date, '2025-01-15');
+      assert.equal(result.summary, 'A productive day');
+    });
+  });
+});