@malloy-publisher/server 0.0.235 → 0.0.236

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (301) hide show
  1. package/dist/app/api-doc.yaml +47 -17
  2. package/dist/package_load_worker.mjs +8 -1
  3. package/dist/server.mjs +471 -54
  4. package/package.json +5 -2
  5. package/.eslintrc.json +0 -44
  6. package/.prettierignore +0 -1
  7. package/.prettierrc +0 -4
  8. package/build.ts +0 -97
  9. package/eslint.config.mjs +0 -76
  10. package/k6-tests/README.md +0 -144
  11. package/k6-tests/bun.lock +0 -797
  12. package/k6-tests/load-test/load-test-crud-connections.ts +0 -263
  13. package/k6-tests/load-test/load-test-crud-packages.ts +0 -461
  14. package/k6-tests/load-test/load-test-crud-projects.ts +0 -160
  15. package/k6-tests/load-test/load-test-crud.ts +0 -142
  16. package/k6-tests/load-test/load-test.ts +0 -784
  17. package/k6-tests/package.json +0 -27
  18. package/k6-tests/scripts/run-with-otel.sh +0 -50
  19. package/k6-tests/smoke-test/smoke-test.ts +0 -158
  20. package/k6-tests/tsconfig.json +0 -29
  21. package/k6-tests/utils/client_factory.ts +0 -128
  22. package/k6-tests/utils/common.ts +0 -489
  23. package/k6-tests/utils/logger.ts +0 -62
  24. package/k6-tests/utils/types.d.ts +0 -35
  25. package/malloy_mcp.dxt +0 -0
  26. package/publisher.config.example.bigquery.json +0 -33
  27. package/publisher.config.example.duckdb.json +0 -23
  28. package/publisher.config.json +0 -23
  29. package/scripts/bake-duckdb-extensions.js +0 -110
  30. package/src/config.spec.ts +0 -1535
  31. package/src/config.theme.spec.ts +0 -178
  32. package/src/config.ts +0 -1142
  33. package/src/constants.ts +0 -105
  34. package/src/controller/compile.controller.ts +0 -40
  35. package/src/controller/connection.controller.spec.ts +0 -849
  36. package/src/controller/connection.controller.ts +0 -901
  37. package/src/controller/database.controller.ts +0 -24
  38. package/src/controller/materialization.controller.spec.ts +0 -293
  39. package/src/controller/materialization.controller.ts +0 -319
  40. package/src/controller/model.controller.ts +0 -164
  41. package/src/controller/package.controller.spec.ts +0 -273
  42. package/src/controller/package.controller.ts +0 -306
  43. package/src/controller/query.controller.ts +0 -152
  44. package/src/controller/theme.controller.ts +0 -83
  45. package/src/controller/watch-mode.controller.ts +0 -243
  46. package/src/data_styles.ts +0 -71
  47. package/src/default-publisher.config.json +0 -23
  48. package/src/ducklake_version.spec.ts +0 -163
  49. package/src/ducklake_version.ts +0 -153
  50. package/src/errors.spec.ts +0 -119
  51. package/src/errors.ts +0 -237
  52. package/src/filter_deprecation.spec.ts +0 -64
  53. package/src/filter_deprecation.ts +0 -42
  54. package/src/health.spec.ts +0 -123
  55. package/src/health.ts +0 -233
  56. package/src/heap_check.spec.ts +0 -144
  57. package/src/heap_check.ts +0 -144
  58. package/src/instrumentation.ts +0 -214
  59. package/src/json_utils.spec.ts +0 -51
  60. package/src/json_utils.ts +0 -33
  61. package/src/logger.spec.ts +0 -210
  62. package/src/logger.ts +0 -251
  63. package/src/malloy_pin_prereqs.spec.ts +0 -25
  64. package/src/materialization_metrics.spec.ts +0 -169
  65. package/src/materialization_metrics.ts +0 -346
  66. package/src/mcp/error_messages.spec.ts +0 -35
  67. package/src/mcp/error_messages.ts +0 -183
  68. package/src/mcp/handler_utils.spec.ts +0 -108
  69. package/src/mcp/handler_utils.ts +0 -251
  70. package/src/mcp/mcp_constants.ts +0 -6
  71. package/src/mcp/query_envelope.spec.ts +0 -229
  72. package/src/mcp/query_envelope.ts +0 -240
  73. package/src/mcp/server.protocol.spec.ts +0 -250
  74. package/src/mcp/server.ts +0 -87
  75. package/src/mcp/skills/build_skills_bundle.spec.ts +0 -51
  76. package/src/mcp/skills/build_skills_bundle.ts +0 -176
  77. package/src/mcp/skills/skills_bundle.json +0 -1
  78. package/src/mcp/skills/skills_bundle.spec.ts +0 -171
  79. package/src/mcp/tool_response.spec.ts +0 -108
  80. package/src/mcp/tool_response.ts +0 -138
  81. package/src/mcp/tools/compile_tool.spec.ts +0 -315
  82. package/src/mcp/tools/compile_tool.ts +0 -208
  83. package/src/mcp/tools/docs_search/build_docs_index.ts +0 -132
  84. package/src/mcp/tools/docs_search/malloy_docs_index.json +0 -1
  85. package/src/mcp/tools/docs_search_tool.spec.ts +0 -32
  86. package/src/mcp/tools/docs_search_tool.ts +0 -128
  87. package/src/mcp/tools/embedding_index.spec.ts +0 -1236
  88. package/src/mcp/tools/embedding_index.ts +0 -808
  89. package/src/mcp/tools/execute_query_tool.spec.ts +0 -294
  90. package/src/mcp/tools/execute_query_tool.ts +0 -316
  91. package/src/mcp/tools/get_context_eval.ts +0 -275
  92. package/src/mcp/tools/get_context_tool.spec.ts +0 -592
  93. package/src/mcp/tools/get_context_tool.ts +0 -585
  94. package/src/mcp/tools/reload_package_tool.spec.ts +0 -232
  95. package/src/mcp/tools/reload_package_tool.ts +0 -132
  96. package/src/mcp_config.spec.ts +0 -919
  97. package/src/mcp_config.ts +0 -425
  98. package/src/oom_guards.integration.spec.ts +0 -269
  99. package/src/package_load/package_load_pool.spec.ts +0 -252
  100. package/src/package_load/package_load_pool.ts +0 -931
  101. package/src/package_load/package_load_worker.ts +0 -1034
  102. package/src/package_load/protocol.ts +0 -364
  103. package/src/package_load/rpc_wait_accountant.spec.ts +0 -109
  104. package/src/package_load/rpc_wait_accountant.ts +0 -76
  105. package/src/package_load_metrics.spec.ts +0 -114
  106. package/src/package_load_metrics.ts +0 -127
  107. package/src/path_safety.spec.ts +0 -158
  108. package/src/path_safety.ts +0 -146
  109. package/src/pg_helpers.spec.ts +0 -223
  110. package/src/pg_helpers.ts +0 -52
  111. package/src/query_cap_metrics.spec.ts +0 -89
  112. package/src/query_cap_metrics.ts +0 -116
  113. package/src/query_concurrency.spec.ts +0 -247
  114. package/src/query_concurrency.ts +0 -237
  115. package/src/query_metadata_metrics.ts +0 -49
  116. package/src/query_param_utils.ts +0 -29
  117. package/src/query_timeout.spec.ts +0 -224
  118. package/src/query_timeout.ts +0 -179
  119. package/src/runtime/publisher.js +0 -328
  120. package/src/server-old.ts +0 -988
  121. package/src/server.ts +0 -2059
  122. package/src/service/annotations.spec.ts +0 -118
  123. package/src/service/annotations.ts +0 -91
  124. package/src/service/authorize.spec.ts +0 -154
  125. package/src/service/authorize.ts +0 -498
  126. package/src/service/authorize_integration.spec.ts +0 -2041
  127. package/src/service/build_plan.spec.ts +0 -685
  128. package/src/service/build_plan.ts +0 -797
  129. package/src/service/compile_authorize.spec.ts +0 -85
  130. package/src/service/compile_fragment_techniques.spec.ts +0 -156
  131. package/src/service/connection.spec.ts +0 -2360
  132. package/src/service/connection.ts +0 -2187
  133. package/src/service/connection_config.spec.ts +0 -734
  134. package/src/service/connection_config.ts +0 -899
  135. package/src/service/connection_federation.spec.ts +0 -184
  136. package/src/service/connection_fingerprint.spec.ts +0 -102
  137. package/src/service/connection_service.spec.ts +0 -711
  138. package/src/service/connection_service.ts +0 -291
  139. package/src/service/cron_evaluator.spec.ts +0 -79
  140. package/src/service/cron_evaluator.ts +0 -105
  141. package/src/service/db_utils.spec.ts +0 -1085
  142. package/src/service/db_utils.ts +0 -1502
  143. package/src/service/duckdb_instance_isolation.spec.ts +0 -137
  144. package/src/service/ducklake_lazy_attach.spec.ts +0 -110
  145. package/src/service/embedding_provider.spec.ts +0 -329
  146. package/src/service/embedding_provider.ts +0 -236
  147. package/src/service/environment.ts +0 -2191
  148. package/src/service/environment_admission.spec.ts +0 -344
  149. package/src/service/environment_compile.spec.ts +0 -197
  150. package/src/service/environment_store.spec.ts +0 -2274
  151. package/src/service/environment_store.ts +0 -2434
  152. package/src/service/environment_store_anchoring.spec.ts +0 -107
  153. package/src/service/environment_store_clone.spec.ts +0 -350
  154. package/src/service/explore_visibility.spec.ts +0 -434
  155. package/src/service/exports_probe.spec.ts +0 -107
  156. package/src/service/extension_fetch_policy.spec.ts +0 -256
  157. package/src/service/filter.spec.ts +0 -458
  158. package/src/service/filter.ts +0 -341
  159. package/src/service/filter_bypass.spec.ts +0 -418
  160. package/src/service/filter_integration.spec.ts +0 -1004
  161. package/src/service/freshness.spec.ts +0 -183
  162. package/src/service/freshness.ts +0 -112
  163. package/src/service/gcs_s3_utils.ts +0 -427
  164. package/src/service/given.ts +0 -105
  165. package/src/service/givens_integration.spec.ts +0 -219
  166. package/src/service/manifest_loader.spec.ts +0 -189
  167. package/src/service/manifest_loader.ts +0 -156
  168. package/src/service/materialization_build_session.spec.ts +0 -435
  169. package/src/service/materialization_build_session.ts +0 -681
  170. package/src/service/materialization_config_validation.spec.ts +0 -99
  171. package/src/service/materialization_config_validation.ts +0 -120
  172. package/src/service/materialization_eligibility.spec.ts +0 -158
  173. package/src/service/materialization_eligibility.ts +0 -305
  174. package/src/service/materialization_schedule_surface.spec.ts +0 -256
  175. package/src/service/materialization_scheduler.spec.ts +0 -289
  176. package/src/service/materialization_scheduler.ts +0 -304
  177. package/src/service/materialization_serve_transform.spec.ts +0 -1003
  178. package/src/service/materialization_serve_transform.ts +0 -779
  179. package/src/service/materialization_service.spec.ts +0 -2373
  180. package/src/service/materialization_service.ts +0 -2342
  181. package/src/service/materialization_test_fixtures.ts +0 -249
  182. package/src/service/model.spec.ts +0 -1157
  183. package/src/service/model.ts +0 -3370
  184. package/src/service/model_limits.spec.ts +0 -209
  185. package/src/service/model_limits.ts +0 -131
  186. package/src/service/model_storage_serve.spec.ts +0 -193
  187. package/src/service/model_storage_serve_joins.spec.ts +0 -193
  188. package/src/service/package.spec.ts +0 -612
  189. package/src/service/package.ts +0 -1799
  190. package/src/service/package_manifest.spec.ts +0 -230
  191. package/src/service/package_manifest.ts +0 -241
  192. package/src/service/package_memory_governor.spec.ts +0 -173
  193. package/src/service/package_memory_governor.ts +0 -233
  194. package/src/service/package_quote_bound_tables.spec.ts +0 -171
  195. package/src/service/package_race.spec.ts +0 -208
  196. package/src/service/package_reload_safety.spec.ts +0 -193
  197. package/src/service/package_rollback.spec.ts +0 -190
  198. package/src/service/package_worker_path.spec.ts +0 -645
  199. package/src/service/path_injection.spec.ts +0 -39
  200. package/src/service/persist_annotation_validation.spec.ts +0 -89
  201. package/src/service/persist_annotation_validation.ts +0 -59
  202. package/src/service/persistence_policy.spec.ts +0 -374
  203. package/src/service/proxy.spec.ts +0 -414
  204. package/src/service/proxy.ts +0 -248
  205. package/src/service/query_boundary.spec.ts +0 -470
  206. package/src/service/query_metadata.spec.ts +0 -408
  207. package/src/service/query_metadata.ts +0 -492
  208. package/src/service/query_metadata_identity.spec.ts +0 -149
  209. package/src/service/query_text.spec.ts +0 -79
  210. package/src/service/query_text.ts +0 -43
  211. package/src/service/quoting.spec.ts +0 -135
  212. package/src/service/quoting.ts +0 -78
  213. package/src/service/resolve_environment.ts +0 -15
  214. package/src/service/restricted_mode.spec.ts +0 -299
  215. package/src/service/source_extraction.ts +0 -225
  216. package/src/service/theme_store.ts +0 -199
  217. package/src/storage/DatabaseInterface.ts +0 -198
  218. package/src/storage/StorageManager.ts +0 -113
  219. package/src/storage/duckdb/ConnectionRepository.ts +0 -143
  220. package/src/storage/duckdb/DuckDBConnection.ts +0 -123
  221. package/src/storage/duckdb/DuckDBRepository.ts +0 -208
  222. package/src/storage/duckdb/EnvironmentRepository.ts +0 -148
  223. package/src/storage/duckdb/MaterializationRepository.spec.ts +0 -191
  224. package/src/storage/duckdb/MaterializationRepository.ts +0 -336
  225. package/src/storage/duckdb/PackageRepository.ts +0 -135
  226. package/src/storage/duckdb/schema.ts +0 -229
  227. package/src/stream_helpers.spec.ts +0 -280
  228. package/src/stream_helpers.ts +0 -162
  229. package/src/telemetry.ts +0 -20
  230. package/src/test_helpers/metrics_harness.ts +0 -166
  231. package/src/theme_key_parity.spec.ts +0 -57
  232. package/src/utils.ts +0 -29
  233. package/tests/fixtures/authorize-compile/model.malloy +0 -9
  234. package/tests/fixtures/authorize-compile/publisher.json +0 -4
  235. package/tests/fixtures/html-pages-nopublic/model.malloy +0 -1
  236. package/tests/fixtures/html-pages-nopublic/publisher.json +0 -5
  237. package/tests/fixtures/html-pages-test/data.csv +0 -3
  238. package/tests/fixtures/html-pages-test/public/assets/app.css +0 -3
  239. package/tests/fixtures/html-pages-test/public/barehtml.html +0 -4
  240. package/tests/fixtures/html-pages-test/public/bodymeta.html +0 -8
  241. package/tests/fixtures/html-pages-test/public/data.json +0 -1
  242. package/tests/fixtures/html-pages-test/public/fitcomment.html +0 -11
  243. package/tests/fixtures/html-pages-test/public/index.html +0 -9
  244. package/tests/fixtures/html-pages-test/public/nohead.html +0 -8
  245. package/tests/fixtures/html-pages-test/public/notfit.html +0 -13
  246. package/tests/fixtures/html-pages-test/public/slides.html +0 -12
  247. package/tests/fixtures/html-pages-test/public/sub/page2.html +0 -9
  248. package/tests/fixtures/html-pages-test/public/unterminated.html +0 -10
  249. package/tests/fixtures/html-pages-test/publisher.json +0 -5
  250. package/tests/fixtures/html-pages-test/report.malloy +0 -1
  251. package/tests/fixtures/persist-multi-level/data/orders.csv +0 -5
  252. package/tests/fixtures/persist-multi-level/multi_level.malloy +0 -18
  253. package/tests/fixtures/persist-multi-level/publisher.json +0 -5
  254. package/tests/fixtures/persist-schedule-test/data/orders.csv +0 -5
  255. package/tests/fixtures/persist-schedule-test/persist_schedule_test.malloy +0 -11
  256. package/tests/fixtures/persist-schedule-test/publisher.json +0 -7
  257. package/tests/fixtures/persist-schedule-warn-test/data/orders.csv +0 -5
  258. package/tests/fixtures/persist-schedule-warn-test/persist_warn_test.malloy +0 -11
  259. package/tests/fixtures/persist-schedule-warn-test/publisher.json +0 -6
  260. package/tests/fixtures/persist-test/data/orders.csv +0 -5
  261. package/tests/fixtures/persist-test/persist_test.malloy +0 -11
  262. package/tests/fixtures/persist-test/publisher.json +0 -5
  263. package/tests/fixtures/publisher.config.json +0 -15
  264. package/tests/fixtures/query-givens/data/orders.csv +0 -7
  265. package/tests/fixtures/query-givens/model.malloy +0 -34
  266. package/tests/fixtures/query-givens/publisher.json +0 -5
  267. package/tests/fixtures/xlsx/database.xlsx +0 -0
  268. package/tests/harness/mcp_test_setup.ts +0 -332
  269. package/tests/harness/rest_e2e.ts +0 -68
  270. package/tests/integration/authorize/compile_authorize_http.integration.spec.ts +0 -92
  271. package/tests/integration/concurrent_package/concurrent_package.integration.spec.ts +0 -280
  272. package/tests/integration/duckdb_storage/duckdb_storage.integration.spec.ts +0 -138
  273. package/tests/integration/first_boot/readiness_line.integration.spec.ts +0 -177
  274. package/tests/integration/html_pages/html_pages.integration.spec.ts +0 -440
  275. package/tests/integration/legacy_routes/legacy_routes.integration.spec.ts +0 -259
  276. package/tests/integration/materialization/environment_list.integration.spec.ts +0 -133
  277. package/tests/integration/materialization/freshness_gate.integration.spec.ts +0 -292
  278. package/tests/integration/materialization/manifest_binding.integration.spec.ts +0 -430
  279. package/tests/integration/materialization/materialization_lifecycle.integration.spec.ts +0 -441
  280. package/tests/integration/materialization/orchestrated_rebind.integration.spec.ts +0 -150
  281. package/tests/integration/materialization/reference_manifest.integration.spec.ts +0 -251
  282. package/tests/integration/materialization/schedule_edit.integration.spec.ts +0 -126
  283. package/tests/integration/materialization/scheduler.integration.spec.ts +0 -200
  284. package/tests/integration/materialization/scheduler_recovery.integration.spec.ts +0 -172
  285. package/tests/integration/materialization/scheduler_transitions.integration.spec.ts +0 -256
  286. package/tests/integration/materialization/scheduler_wiring.integration.spec.ts +0 -143
  287. package/tests/integration/mcp/mcp_execute_query_tool.integration.spec.ts +0 -438
  288. package/tests/integration/mcp/mcp_get_context_semantic.integration.spec.ts +0 -235
  289. package/tests/integration/mcp/mcp_transport.integration.spec.ts +0 -98
  290. package/tests/integration/query_givens/query_givens.integration.spec.ts +0 -146
  291. package/tests/integration/query_givens/query_givens_authorize.integration.spec.ts +0 -121
  292. package/tests/integration/sdk_givens/sdk_givens.integration.spec.ts +0 -110
  293. package/tests/integration/watch-mode/watch_mode.integration.spec.ts +0 -421
  294. package/tests/unit/controller/theme.controller.spec.ts +0 -141
  295. package/tests/unit/duckdb/attached_databases.test.ts +0 -1243
  296. package/tests/unit/duckdb/duckdb_connection.test.ts +0 -181
  297. package/tests/unit/duckdb/legacy_schema_migration.test.ts +0 -197
  298. package/tests/unit/duckdb/repositories.test.ts +0 -208
  299. package/tests/unit/ducklake/ducklake.test.ts +0 -952
  300. package/tests/unit/service/theme_store.spec.ts +0 -139
  301. package/tsconfig.json +0 -32
@@ -1,181 +0,0 @@
1
- /// <reference types="bun-types" />
2
-
3
- // Direct unit tests for the storage-layer DuckDBConnection -- the embedded
4
- // metadata DAO (publisher.db) that was migrated from the legacy `duckdb`
5
- // callback binding to `@duckdb/node-api` (the "neo" promise API). The
6
- // repositories depend on this class's public surface, so these assert it
7
- // behaves identically across the migration: lifecycle, parameterized CRUD with
8
- // `?` placeholders, row-shape parity, and error wrapping.
9
-
10
- import { afterEach, beforeEach, describe, expect, it } from "bun:test";
11
- import fs from "fs/promises";
12
- import os from "os";
13
- import path from "path";
14
- import { DuckDBConnection } from "../../../src/storage/duckdb/DuckDBConnection";
15
-
16
- describe("DuckDBConnection (storage DAO over @duckdb/node-api)", () => {
17
- let dbDir: string;
18
- let conn: DuckDBConnection;
19
-
20
- beforeEach(async () => {
21
- // Unique dir per test so file handles never collide across tests
22
- // (Windows runners are sensitive to reopening the same DuckDB file).
23
- dbDir = await fs.mkdtemp(path.join(os.tmpdir(), "duckdb-conn-"));
24
- conn = new DuckDBConnection(path.join(dbDir, "test.db"));
25
- await conn.initialize();
26
- });
27
-
28
- afterEach(async () => {
29
- try {
30
- await conn.close();
31
- } catch {
32
- // already closed / never opened
33
- }
34
- await fs.rm(dbDir, { recursive: true, force: true });
35
- });
36
-
37
- describe("lifecycle", () => {
38
- it("initializes and reports a usable connection", async () => {
39
- const rows = await conn.all<{ answer: number }>("SELECT 42 AS answer");
40
- expect(rows[0].answer).toBe(42);
41
- });
42
-
43
- it("isInitialized is false before the schema exists", async () => {
44
- // Fresh DB with no `environments` table yet.
45
- expect(await conn.isInitialized()).toBe(false);
46
- });
47
-
48
- it("isInitialized becomes true once the environments table exists", async () => {
49
- await conn.run("CREATE TABLE environments (id VARCHAR PRIMARY KEY)");
50
- expect(await conn.isInitialized()).toBe(true);
51
- });
52
-
53
- it("close() is idempotent and a second close does not throw", async () => {
54
- await conn.close();
55
- await expect(conn.close()).resolves.toBeUndefined();
56
- });
57
- });
58
-
59
- describe("run / all / get with ? placeholders", () => {
60
- beforeEach(async () => {
61
- await conn.run(`
62
- CREATE TABLE t (
63
- id VARCHAR PRIMARY KEY,
64
- name VARCHAR,
65
- flag BOOLEAN,
66
- metadata JSON,
67
- created_at TIMESTAMP
68
- )
69
- `);
70
- });
71
-
72
- it("binds positional ? parameters on INSERT and reads them back", async () => {
73
- const now = new Date().toISOString();
74
- await conn.run(
75
- "INSERT INTO t (id, name, flag, metadata, created_at) VALUES (?, ?, ?, ?, ?)",
76
- ["a", "alpha", true, JSON.stringify({ k: 1 }), now],
77
- );
78
-
79
- const row = await conn.get<{
80
- id: string;
81
- name: string;
82
- flag: boolean;
83
- metadata: string;
84
- created_at: unknown;
85
- }>("SELECT * FROM t WHERE id = ?", ["a"]);
86
-
87
- expect(row).not.toBeNull();
88
- expect(row!.id).toBe("a");
89
- expect(row!.name).toBe("alpha");
90
- // BOOLEAN round-trips as a JS boolean.
91
- expect(row!.flag).toBe(true);
92
- // JSON is stored/returned as a string the repositories JSON.parse().
93
- expect(typeof row!.metadata).toBe("string");
94
- expect(JSON.parse(row!.metadata)).toEqual({ k: 1 });
95
- // TIMESTAMP is consumable by `new Date(...)` (repositories do exactly this).
96
- expect(new Date(row!.created_at as string).toISOString()).toBe(now);
97
- });
98
-
99
- it("all() returns every matching row", async () => {
100
- await conn.run("INSERT INTO t (id, name) VALUES (?, ?)", ["a", "x"]);
101
- await conn.run("INSERT INTO t (id, name) VALUES (?, ?)", ["b", "y"]);
102
-
103
- const rows = await conn.all<{ id: string }>(
104
- "SELECT id FROM t ORDER BY id",
105
- );
106
- expect(rows.map((r) => r.id)).toEqual(["a", "b"]);
107
- });
108
-
109
- it("get() returns null when no row matches", async () => {
110
- const row = await conn.get("SELECT * FROM t WHERE id = ?", [
111
- "missing",
112
- ]);
113
- expect(row).toBeNull();
114
- });
115
-
116
- it("all() returns an empty array when no row matches", async () => {
117
- const rows = await conn.all("SELECT * FROM t WHERE id = ?", ["nope"]);
118
- expect(rows).toEqual([]);
119
- });
120
-
121
- it("supports statements with no parameters", async () => {
122
- await conn.run("INSERT INTO t (id) VALUES ('solo')");
123
- const rows = await conn.all<{ n: number | bigint }>(
124
- "SELECT count(*) AS n FROM t",
125
- );
126
- expect(Number(rows[0].n)).toBe(1);
127
- });
128
-
129
- it("UPDATE and DELETE bind ? parameters", async () => {
130
- await conn.run("INSERT INTO t (id, name) VALUES (?, ?)", ["a", "old"]);
131
- await conn.run("UPDATE t SET name = ? WHERE id = ?", ["new", "a"]);
132
- expect(
133
- (await conn.get<{ name: string }>(
134
- "SELECT name FROM t WHERE id = ?",
135
- ["a"],
136
- ))!.name,
137
- ).toBe("new");
138
-
139
- await conn.run("DELETE FROM t WHERE id = ?", ["a"]);
140
- expect(
141
- await conn.get("SELECT * FROM t WHERE id = ?", ["a"]),
142
- ).toBeNull();
143
- });
144
- });
145
-
146
- describe("error handling", () => {
147
- it("run() wraps a failing query with the query text", async () => {
148
- await expect(
149
- conn.run("INSERT INTO does_not_exist VALUES (1)"),
150
- ).rejects.toThrow(/Query execution failed[\s\S]*does_not_exist/);
151
- });
152
-
153
- it("all() wraps a failing query with the query text", async () => {
154
- await expect(conn.all("SELECT * FROM does_not_exist")).rejects.toThrow(
155
- /Query execution failed[\s\S]*does_not_exist/,
156
- );
157
- });
158
-
159
- it("run() throws when the connection is not initialized", async () => {
160
- const fresh = new DuckDBConnection(path.join(dbDir, "uninit.db"));
161
- await expect(fresh.run("SELECT 1")).rejects.toThrow(/not initialized/);
162
- });
163
- });
164
-
165
- describe("durability", () => {
166
- it("persists data to the file across reopen", async () => {
167
- const dbPath = path.join(dbDir, "persist.db");
168
- const first = new DuckDBConnection(dbPath);
169
- await first.initialize();
170
- await first.run("CREATE TABLE k (v VARCHAR)");
171
- await first.run("INSERT INTO k VALUES (?)", ["kept"]);
172
- await first.close();
173
-
174
- const second = new DuckDBConnection(dbPath);
175
- await second.initialize();
176
- const rows = await second.all<{ v: string }>("SELECT v FROM k");
177
- await second.close();
178
- expect(rows.map((r) => r.v)).toEqual(["kept"]);
179
- });
180
- });
181
- });
@@ -1,197 +0,0 @@
1
- /// <reference types="bun-types" />
2
-
3
- // TODO: Remove this during projects cleanup
4
-
5
- import { afterEach, beforeEach, describe, expect, it } from "bun:test";
6
- import fs from "fs/promises";
7
- import os from "os";
8
- import path from "path";
9
- import { DuckDBConnection } from "../../../src/storage/duckdb/DuckDBConnection";
10
- import { initializeSchema } from "../../../src/storage/duckdb/schema";
11
-
12
- const TEST_DB_DIR = path.join(os.tmpdir(), "duckdb-legacy-migration-tests");
13
-
14
- // Seed a pre-rename schema on an *already-open* connection. We deliberately
15
- // avoid opening, closing, and reopening the same DuckDB file within a test:
16
- // on Windows runners the second `duckdb.Database(path)` call sometimes fails
17
- // with `Invalid Error` because the OS hasn't released the file handle yet.
18
- // Sharing one connection between seed and assertion sidesteps that entirely.
19
- async function seedLegacySchema(db: DuckDBConnection): Promise<void> {
20
- // Seed a pre-rename schema: parent table named `projects` and child
21
- // tables with `project_id` foreign-key columns. Mirrors what an existing
22
- // installation looked like before the projects→environments rename.
23
- await db.run(`
24
- CREATE TABLE projects (
25
- id VARCHAR PRIMARY KEY,
26
- name VARCHAR NOT NULL UNIQUE,
27
- path VARCHAR NOT NULL,
28
- description VARCHAR,
29
- metadata JSON,
30
- created_at TIMESTAMP NOT NULL,
31
- updated_at TIMESTAMP NOT NULL
32
- )
33
- `);
34
- await db.run(`
35
- CREATE TABLE packages (
36
- id VARCHAR PRIMARY KEY,
37
- project_id VARCHAR NOT NULL,
38
- name VARCHAR NOT NULL,
39
- description VARCHAR,
40
- manifest_path VARCHAR NOT NULL,
41
- metadata JSON,
42
- created_at TIMESTAMP NOT NULL,
43
- updated_at TIMESTAMP NOT NULL,
44
- FOREIGN KEY (project_id) REFERENCES projects(id)
45
- )
46
- `);
47
- await db.run(`
48
- CREATE TABLE connections (
49
- id VARCHAR PRIMARY KEY,
50
- project_id VARCHAR NOT NULL,
51
- name VARCHAR NOT NULL,
52
- type VARCHAR NOT NULL,
53
- config JSON NOT NULL,
54
- created_at TIMESTAMP NOT NULL,
55
- updated_at TIMESTAMP NOT NULL,
56
- FOREIGN KEY (project_id) REFERENCES projects(id)
57
- )
58
- `);
59
- await db.run(`
60
- CREATE TABLE materializations (
61
- id VARCHAR PRIMARY KEY,
62
- project_id VARCHAR NOT NULL,
63
- package_name VARCHAR NOT NULL,
64
- status VARCHAR NOT NULL,
65
- active_key VARCHAR,
66
- started_at TIMESTAMP,
67
- completed_at TIMESTAMP,
68
- error TEXT,
69
- metadata JSON,
70
- created_at TIMESTAMP NOT NULL,
71
- updated_at TIMESTAMP NOT NULL,
72
- FOREIGN KEY (project_id) REFERENCES projects(id)
73
- )
74
- `);
75
- await db.run(`
76
- CREATE TABLE build_manifests (
77
- id VARCHAR PRIMARY KEY,
78
- project_id VARCHAR NOT NULL,
79
- package_name VARCHAR NOT NULL,
80
- build_id VARCHAR NOT NULL,
81
- table_name VARCHAR NOT NULL,
82
- source_name VARCHAR NOT NULL,
83
- connection_name VARCHAR NOT NULL,
84
- created_at TIMESTAMP NOT NULL,
85
- updated_at TIMESTAMP NOT NULL,
86
- FOREIGN KEY (project_id) REFERENCES projects(id)
87
- )
88
- `);
89
-
90
- await db.run(
91
- `INSERT INTO projects VALUES ('p1', 'proj-one', '/p1', 'd1', NULL,
92
- TIMESTAMP '2024-01-01 00:00:00', TIMESTAMP '2024-01-01 00:00:00')`,
93
- );
94
- await db.run(
95
- `INSERT INTO packages VALUES ('pkg1', 'p1', 'pkg-one', NULL, '/m', NULL,
96
- TIMESTAMP '2024-01-01 00:00:00', TIMESTAMP '2024-01-01 00:00:00')`,
97
- );
98
- }
99
-
100
- describe("DuckDB legacy projects schema cleanup", () => {
101
- beforeEach(async () => {
102
- await fs.mkdir(TEST_DB_DIR, { recursive: true });
103
- });
104
-
105
- afterEach(async () => {
106
- try {
107
- await fs.rm(TEST_DB_DIR, { recursive: true, force: true });
108
- } catch {
109
- // ignore
110
- }
111
- });
112
-
113
- it("drops legacy projects schema and creates the new environments schema cleanly", async () => {
114
- const dbPath = path.join(TEST_DB_DIR, "legacy.duckdb");
115
- const db = new DuckDBConnection(dbPath);
116
- await db.initialize();
117
-
118
- // Seed the legacy schema on the same connection, then run the
119
- // production schema-init path. This mirrors a server upgrade
120
- // (legacy data on disk, new code starting up) without forcing a
121
- // close+reopen, which is unreliable on Windows runners.
122
- await seedLegacySchema(db);
123
- await initializeSchema(db);
124
-
125
- // Legacy parent table is gone.
126
- const legacyProjects = await db.all<{ name: string }>(
127
- "SELECT name FROM sqlite_master WHERE type='table' AND name='projects'",
128
- );
129
- expect(legacyProjects.length).toBe(0);
130
-
131
- // New `environments` table exists and is empty (legacy data dropped).
132
- const envs = await db.all<{ id: string }>("SELECT id FROM environments");
133
- expect(envs.length).toBe(0);
134
-
135
- // Child tables are queryable by `environment_id` (the new column),
136
- // proving they were recreated with the new schema rather than left on
137
- // the old `project_id` column.
138
- const pkgs = await db.all<{ id: string }>(
139
- "SELECT id FROM packages WHERE environment_id = ?",
140
- ["p1"],
141
- );
142
- expect(pkgs.length).toBe(0);
143
- const conns = await db.all<{ id: string }>(
144
- "SELECT id FROM connections WHERE environment_id = ?",
145
- ["p1"],
146
- );
147
- expect(conns.length).toBe(0);
148
- const mats = await db.all<{ id: string }>(
149
- "SELECT id FROM materializations WHERE environment_id = ?",
150
- ["p1"],
151
- );
152
- expect(mats.length).toBe(0);
153
-
154
- // The legacy `build_manifests` table is dropped and not recreated: the
155
- // two-round refactor folds the manifest into a JSON column on the
156
- // `materializations` row, so no standalone manifest table remains.
157
- const manifestTable = await db.all<{ name: string }>(
158
- "SELECT name FROM sqlite_master WHERE type='table' AND name='build_manifests'",
159
- );
160
- expect(manifestTable.length).toBe(0);
161
-
162
- await db.close();
163
- });
164
-
165
- it("is idempotent: running initializeSchema twice on a migrated DB is a no-op", async () => {
166
- const dbPath = path.join(TEST_DB_DIR, "legacy_idempotent.duckdb");
167
- const db = new DuckDBConnection(dbPath);
168
- await db.initialize();
169
-
170
- await seedLegacySchema(db);
171
- await initializeSchema(db);
172
- // Second call should hit the early-return path (isInitialized() === true).
173
- await initializeSchema(db);
174
-
175
- const envs = await db.all<{ id: string }>("SELECT id FROM environments");
176
- expect(envs.length).toBe(0);
177
-
178
- await db.close();
179
- });
180
-
181
- it("creates a fresh schema unchanged when no legacy projects table is present", async () => {
182
- const dbPath = path.join(TEST_DB_DIR, "fresh.duckdb");
183
- const db = new DuckDBConnection(dbPath);
184
- await db.initialize();
185
- await initializeSchema(db);
186
-
187
- const envs = await db.all<{ id: string }>("SELECT id FROM environments");
188
- expect(envs.length).toBe(0);
189
-
190
- const legacy = await db.all<{ name: string }>(
191
- "SELECT name FROM sqlite_master WHERE type='table' AND name='projects'",
192
- );
193
- expect(legacy.length).toBe(0);
194
-
195
- await db.close();
196
- });
197
- });
@@ -1,208 +0,0 @@
1
- /// <reference types="bun-types" />
2
-
3
- // Round-trip tests for the DuckDB storage repositories against a REAL
4
- // DuckDBConnection (no mocks). The repositories were untested at the storage
5
- // layer -- existing service specs mock the repository methods -- so these run
6
- // their actual SQL through the connection migrated to @duckdb/node-api. They
7
- // guard the migration-sensitive behaviors: parameterized writes, JSON-column
8
- // round-trips (config/metadata via JSON.stringify -> JSON.parse), TIMESTAMP ->
9
- // Date mapping, and foreign-key relationships between the tables.
10
-
11
- import { afterEach, beforeEach, describe, expect, it } from "bun:test";
12
- import fs from "fs/promises";
13
- import os from "os";
14
- import path from "path";
15
- import { DuckDBConnection } from "../../../src/storage/duckdb/DuckDBConnection";
16
- import { initializeSchema } from "../../../src/storage/duckdb/schema";
17
- import { EnvironmentRepository } from "../../../src/storage/duckdb/EnvironmentRepository";
18
- import { PackageRepository } from "../../../src/storage/duckdb/PackageRepository";
19
- import { ConnectionRepository } from "../../../src/storage/duckdb/ConnectionRepository";
20
-
21
- describe("DuckDB storage repositories (real connection)", () => {
22
- let dbDir: string;
23
- let db: DuckDBConnection;
24
- let environments: EnvironmentRepository;
25
- let packages: PackageRepository;
26
- let connections: ConnectionRepository;
27
-
28
- beforeEach(async () => {
29
- dbDir = await fs.mkdtemp(path.join(os.tmpdir(), "duckdb-repos-"));
30
- db = new DuckDBConnection(path.join(dbDir, "test.db"));
31
- await db.initialize();
32
- await initializeSchema(db);
33
- environments = new EnvironmentRepository(db);
34
- packages = new PackageRepository(db);
35
- connections = new ConnectionRepository(db);
36
- });
37
-
38
- afterEach(async () => {
39
- try {
40
- await db.close();
41
- } catch {
42
- // ignore
43
- }
44
- await fs.rm(dbDir, { recursive: true, force: true });
45
- });
46
-
47
- describe("EnvironmentRepository", () => {
48
- it("creates and reads back an environment with metadata", async () => {
49
- const created = await environments.createEnvironment({
50
- name: "env-a",
51
- path: "/tmp/env-a",
52
- description: "first",
53
- metadata: { team: "data", tier: 2 },
54
- });
55
-
56
- expect(created.id).toBeTruthy();
57
- expect(created.createdAt).toBeInstanceOf(Date);
58
-
59
- const byId = await environments.getEnvironmentById(created.id);
60
- expect(byId).not.toBeNull();
61
- expect(byId!.name).toBe("env-a");
62
- expect(byId!.description).toBe("first");
63
- // metadata survives the JSON round-trip (stringify on write, parse on read)
64
- expect(byId!.metadata).toEqual({ team: "data", tier: 2 });
65
- // TIMESTAMP columns come back as Date
66
- expect(byId!.createdAt).toBeInstanceOf(Date);
67
- expect(byId!.updatedAt).toBeInstanceOf(Date);
68
- });
69
-
70
- it("looks up an environment by name", async () => {
71
- await environments.createEnvironment({
72
- name: "env-b",
73
- path: "/tmp/b",
74
- });
75
- const byName = await environments.getEnvironmentByName("env-b");
76
- expect(byName?.name).toBe("env-b");
77
- expect(await environments.getEnvironmentByName("nope")).toBeNull();
78
- });
79
-
80
- it("lists, updates, and deletes environments", async () => {
81
- const e = await environments.createEnvironment({
82
- name: "env-c",
83
- path: "/tmp/c",
84
- });
85
-
86
- expect(
87
- (await environments.listEnvironments()).map((x) => x.name),
88
- ).toContain("env-c");
89
-
90
- const updated = await environments.updateEnvironment(e.id, {
91
- description: "updated",
92
- metadata: { changed: true },
93
- });
94
- expect(updated.description).toBe("updated");
95
- expect(updated.metadata).toEqual({ changed: true });
96
-
97
- await environments.deleteEnvironment(e.id);
98
- expect(await environments.getEnvironmentById(e.id)).toBeNull();
99
- });
100
- });
101
-
102
- describe("PackageRepository", () => {
103
- it("creates a package under an environment and reads it back", async () => {
104
- const env = await environments.createEnvironment({
105
- name: "env-pkg",
106
- path: "/tmp/env-pkg",
107
- });
108
-
109
- const pkg = await packages.createPackage({
110
- environmentId: env.id,
111
- name: "pkg-1",
112
- manifestPath: "/tmp/env-pkg/pkg-1/publisher.json",
113
- metadata: { source: "test" },
114
- });
115
-
116
- const byId = await packages.getPackageById(pkg.id);
117
- expect(byId?.name).toBe("pkg-1");
118
- expect(byId?.manifestPath).toBe("/tmp/env-pkg/pkg-1/publisher.json");
119
- expect(byId?.metadata).toEqual({ source: "test" });
120
-
121
- const byName = await packages.getPackageByName(env.id, "pkg-1");
122
- expect(byName?.id).toBe(pkg.id);
123
-
124
- const list = await packages.listPackages(env.id);
125
- expect(list.map((p) => p.name)).toEqual(["pkg-1"]);
126
-
127
- await packages.deletePackage(pkg.id);
128
- expect(await packages.getPackageById(pkg.id)).toBeNull();
129
- });
130
- });
131
-
132
- describe("ConnectionRepository", () => {
133
- it("round-trips a connection's JSON config", async () => {
134
- const env = await environments.createEnvironment({
135
- name: "env-conn",
136
- path: "/tmp/env-conn",
137
- });
138
-
139
- const config = {
140
- account: "acct",
141
- warehouse: "wh",
142
- nested: { a: 1, b: ["x", "y"] },
143
- };
144
- const created = await connections.createConnection({
145
- environmentId: env.id,
146
- name: "snow",
147
- type: "snowflake",
148
- config,
149
- });
150
-
151
- const byId = await connections.getConnectionById(created.id);
152
- expect(byId).not.toBeNull();
153
- expect(byId!.type).toBe("snowflake");
154
- // The config column is stored as JSON and must deserialize identically.
155
- expect(byId!.config).toEqual(config);
156
-
157
- const byName = await connections.getConnectionByName(env.id, "snow");
158
- expect(byName?.id).toBe(created.id);
159
- });
160
-
161
- it("lists and deletes connections, including by environment", async () => {
162
- const env = await environments.createEnvironment({
163
- name: "env-conn2",
164
- path: "/tmp/env-conn2",
165
- });
166
- await connections.createConnection({
167
- environmentId: env.id,
168
- name: "c1",
169
- type: "postgres",
170
- config: { host: "h" },
171
- });
172
- await connections.createConnection({
173
- environmentId: env.id,
174
- name: "c2",
175
- type: "duckdb",
176
- config: {},
177
- });
178
-
179
- expect(
180
- (await connections.listConnections(env.id)).map((c) => c.name),
181
- ).toEqual(["c1", "c2"]);
182
-
183
- await connections.deleteConnectionsByEnvironmentId(env.id);
184
- expect(await connections.listConnections(env.id)).toEqual([]);
185
- });
186
-
187
- it("updates a connection's config", async () => {
188
- const env = await environments.createEnvironment({
189
- name: "env-conn3",
190
- path: "/tmp/env-conn3",
191
- });
192
- const c = await connections.createConnection({
193
- environmentId: env.id,
194
- name: "c",
195
- type: "postgres",
196
- config: { host: "old" },
197
- });
198
-
199
- const updated = await connections.updateConnection(c.id, {
200
- config: { host: "new", port: 5432 },
201
- });
202
- expect(updated.config).toEqual({ host: "new", port: 5432 });
203
-
204
- const reread = await connections.getConnectionById(c.id);
205
- expect(reread!.config).toEqual({ host: "new", port: 5432 });
206
- });
207
- });
208
- });