init

backend-node/test/libraryDedup.test.js

@@ -0,0 +1,184 @@
+const test = require('node:test');
+const assert = require('node:assert/strict');
+const Database = require('better-sqlite3');
+
+const characterLibraryService = require('../src/services/characterLibraryService');
+const sceneLibraryService = require('../src/services/sceneLibraryService');
+const propLibraryService = require('../src/services/propLibraryService');
+
+const log = {
+  info() {},
+  warn() {},
+  error() {},
+};
+
+function createDb() {
+  const db = new Database(':memory:');
+  db.exec(`
+    CREATE TABLE dramas (
+      id INTEGER PRIMARY KEY AUTOINCREMENT,
+      title TEXT,
+      deleted_at TEXT
+    );
+
+    CREATE TABLE characters (
+      id INTEGER PRIMARY KEY AUTOINCREMENT,
+      drama_id INTEGER NOT NULL,
+      name TEXT NOT NULL DEFAULT '',
+      description TEXT,
+      appearance TEXT,
+      image_url TEXT,
+      local_path TEXT,
+      deleted_at TEXT
+    );
+
+    CREATE TABLE scenes (
+      id INTEGER PRIMARY KEY AUTOINCREMENT,
+      drama_id INTEGER NOT NULL,
+      location TEXT,
+      time TEXT,
+      prompt TEXT,
+      image_url TEXT,
+      local_path TEXT,
+      deleted_at TEXT
+    );
+
+    CREATE TABLE props (
+      id INTEGER PRIMARY KEY AUTOINCREMENT,
+      drama_id INTEGER NOT NULL,
+      name TEXT NOT NULL DEFAULT '',
+      type TEXT,
+      description TEXT,
+      prompt TEXT,
+      image_url TEXT,
+      local_path TEXT,
+      deleted_at TEXT
+    );
+
+    CREATE TABLE character_libraries (
+      id INTEGER PRIMARY KEY AUTOINCREMENT,
+      drama_id INTEGER,
+      name TEXT NOT NULL DEFAULT '',
+      category TEXT,
+      image_url TEXT,
+      local_path TEXT,
+      description TEXT,
+      tags TEXT,
+      source_type TEXT,
+      source_id TEXT,
+      created_at TEXT,
+      updated_at TEXT,
+      deleted_at TEXT
+    );
+
+    CREATE TABLE scene_libraries (
+      id INTEGER PRIMARY KEY AUTOINCREMENT,
+      drama_id INTEGER,
+      location TEXT NOT NULL DEFAULT '',
+      time TEXT,
+      prompt TEXT,
+      description TEXT,
+      image_url TEXT,
+      local_path TEXT,
+      category TEXT,
+      tags TEXT,
+      source_type TEXT,
+      source_id TEXT,
+      created_at TEXT,
+      updated_at TEXT,
+      deleted_at TEXT
+    );
+
+    CREATE TABLE prop_libraries (
+      id INTEGER PRIMARY KEY AUTOINCREMENT,
+      drama_id INTEGER,
+      name TEXT NOT NULL DEFAULT '',
+      description TEXT,
+      prompt TEXT,
+      image_url TEXT,
+      local_path TEXT,
+      category TEXT,
+      tags TEXT,
+      source_type TEXT,
+      source_id TEXT,
+      created_at TEXT,
+      updated_at TEXT,
+      deleted_at TEXT
+    );
+  `);
+  db.prepare('INSERT INTO dramas (id, title) VALUES (1, ?)').run('Test drama');
+  return db;
+}
+
+function countRows(db, table, where) {
+  return db.prepare(`SELECT COUNT(*) AS count FROM ${table} WHERE ${where}`).get().count;
+}
+
+test('adding the same character to drama and material libraries is idempotent', () => {
+  const db = createDb();
+  db.prepare(
+    'INSERT INTO characters (id, drama_id, name, description, image_url, local_path) VALUES (1, 1, ?, ?, ?, ?)'
+  ).run('Hero', 'original', '/static/projects/hero.png', 'projects/hero.png');
+
+  const firstDrama = characterLibraryService.addCharacterToLibrary(db, log, 1);
+  db.prepare('UPDATE characters SET name = ?, description = ? WHERE id = 1').run('Hero updated', 'updated');
+  const secondDrama = characterLibraryService.addCharacterToLibrary(db, log, 1);
+  const firstMaterial = characterLibraryService.addCharacterToMaterialLibrary(db, log, 1);
+  const secondMaterial = characterLibraryService.addCharacterToMaterialLibrary(db, log, 1);
+
+  assert.equal(firstDrama.item.id, secondDrama.item.id);
+  assert.equal(firstMaterial.item.id, secondMaterial.item.id);
+  assert.equal(countRows(db, 'character_libraries', 'drama_id = 1 AND deleted_at IS NULL'), 1);
+  assert.equal(countRows(db, 'character_libraries', 'drama_id IS NULL AND deleted_at IS NULL'), 1);
+  assert.equal(secondDrama.item.name, 'Hero updated');
+  assert.equal(
+    db.prepare('SELECT source_id FROM character_libraries WHERE id = ?').get(secondDrama.item.id).source_id,
+    '1'
+  );
+});
+
+test('adding the same scene to drama and material libraries is idempotent', () => {
+  const db = createDb();
+  db.prepare(
+    'INSERT INTO scenes (id, drama_id, location, time, prompt, image_url, local_path) VALUES (1, 1, ?, ?, ?, ?, ?)'
+  ).run('Village', 'day', 'quiet street', '/static/projects/village.png', 'projects/village.png');
+
+  const firstDrama = sceneLibraryService.addSceneToLibrary(db, log, 1);
+  db.prepare('UPDATE scenes SET location = ?, prompt = ? WHERE id = 1').run('Village updated', 'busy street');
+  const secondDrama = sceneLibraryService.addSceneToLibrary(db, log, 1);
+  const firstMaterial = sceneLibraryService.addSceneToMaterialLibrary(db, log, 1);
+  const secondMaterial = sceneLibraryService.addSceneToMaterialLibrary(db, log, 1);
+
+  assert.equal(firstDrama.item.id, secondDrama.item.id);
+  assert.equal(firstMaterial.item.id, secondMaterial.item.id);
+  assert.equal(countRows(db, 'scene_libraries', 'drama_id = 1 AND deleted_at IS NULL'), 1);
+  assert.equal(countRows(db, 'scene_libraries', 'drama_id IS NULL AND deleted_at IS NULL'), 1);
+  assert.equal(secondDrama.item.location, 'Village updated');
+  assert.equal(
+    db.prepare('SELECT source_id FROM scene_libraries WHERE id = ?').get(secondDrama.item.id).source_id,
+    '1'
+  );
+});
+
+test('adding the same prop to drama and material libraries is idempotent', () => {
+  const db = createDb();
+  db.prepare(
+    'INSERT INTO props (id, drama_id, name, description, prompt, image_url, local_path) VALUES (1, 1, ?, ?, ?, ?, ?)'
+  ).run('Sword', 'old blade', 'silver sword', '/static/projects/sword.png', 'projects/sword.png');
+
+  const firstDrama = propLibraryService.addPropToLibrary(db, log, 1);
+  db.prepare('UPDATE props SET name = ?, description = ? WHERE id = 1').run('Sword updated', 'polished blade');
+  const secondDrama = propLibraryService.addPropToLibrary(db, log, 1);
+  const firstMaterial = propLibraryService.addPropToMaterialLibrary(db, log, 1);
+  const secondMaterial = propLibraryService.addPropToMaterialLibrary(db, log, 1);
+
+  assert.equal(firstDrama.item.id, secondDrama.item.id);
+  assert.equal(firstMaterial.item.id, secondMaterial.item.id);
+  assert.equal(countRows(db, 'prop_libraries', 'drama_id = 1 AND deleted_at IS NULL'), 1);
+  assert.equal(countRows(db, 'prop_libraries', 'drama_id IS NULL AND deleted_at IS NULL'), 1);
+  assert.equal(secondDrama.item.name, 'Sword updated');
+  assert.equal(
+    db.prepare('SELECT source_id FROM prop_libraries WHERE id = ?').get(secondDrama.item.id).source_id,
+    '1'
+  );
+});