Sync from a8e1c0a7402940e0fc5beef669522b315052df1blatest

9 files changed, 110 insertions, 0 deletions

diff --git a/packages/db/test/fixtures/integrations/astro.config.mjs b/packages/db/test/fixtures/integrations/astro.config.mjs
new file mode 100644
index 000000000..23f52739e
--- /dev/null
+++ b/packages/db/test/fixtures/integrations/astro.config.mjs
@@ -0,0 +1,8 @@
+import db from '@astrojs/db';
+import { defineConfig } from 'astro/config';
+import testIntegration from './integration';
+
+// https://astro.build/config
+export default defineConfig({
+	integrations: [db(), testIntegration()],
+});
diff --git a/packages/db/test/fixtures/integrations/db/config.ts b/packages/db/test/fixtures/integrations/db/config.ts
new file mode 100644
index 000000000..b8110406a
--- /dev/null
+++ b/packages/db/test/fixtures/integrations/db/config.ts
@@ -0,0 +1,12 @@
+import { column, defineDb, defineTable } from 'astro:db';
+
+const Author = defineTable({
+	columns: {
+		name: column.text(),
+		age2: column.number({ optional: true }),
+	},
+});
+
+export default defineDb({
+	tables: { Author },
+});
diff --git a/packages/db/test/fixtures/integrations/db/seed.ts b/packages/db/test/fixtures/integrations/db/seed.ts
new file mode 100644
index 000000000..56ffb5668
--- /dev/null
+++ b/packages/db/test/fixtures/integrations/db/seed.ts
@@ -0,0 +1,13 @@
+import { Author, db } from 'astro:db';
+
+export default async () => {
+	await db
+		.insert(Author)
+		.values([
+			{ name: 'Ben' },
+			{ name: 'Nate' },
+			{ name: 'Erika' },
+			{ name: 'Bjorn' },
+			{ name: 'Sarah' },
+		]);
+};
diff --git a/packages/db/test/fixtures/integrations/integration/config.ts b/packages/db/test/fixtures/integrations/integration/config.ts
new file mode 100644
index 000000000..71490be95
--- /dev/null
+++ b/packages/db/test/fixtures/integrations/integration/config.ts
@@ -0,0 +1,8 @@
+import { defineDb } from 'astro:db';
+import { menu } from './shared';
+
+export default defineDb({
+	tables: {
+		menu,
+	},
+});
diff --git a/packages/db/test/fixtures/integrations/integration/index.ts b/packages/db/test/fixtures/integrations/integration/index.ts
new file mode 100644
index 000000000..b249cc253
--- /dev/null
+++ b/packages/db/test/fixtures/integrations/integration/index.ts
@@ -0,0 +1,15 @@
+import { defineDbIntegration } from '@astrojs/db/utils';
+
+export default function testIntegration() {
+	return defineDbIntegration({
+		name: 'db-test-integration',
+		hooks: {
+			'astro:db:setup'({ extendDb }) {
+				extendDb({
+					configEntrypoint: './integration/config.ts',
+					seedEntrypoint: './integration/seed.ts',
+				});
+			},
+		},
+	});
+}
diff --git a/packages/db/test/fixtures/integrations/integration/seed.ts b/packages/db/test/fixtures/integrations/integration/seed.ts
new file mode 100644
index 000000000..ed2b2e2eb
--- /dev/null
+++ b/packages/db/test/fixtures/integrations/integration/seed.ts
@@ -0,0 +1,14 @@
+import { db } from 'astro:db';
+import { asDrizzleTable } from '@astrojs/db/utils';
+import { menu } from './shared';
+
+export default async function () {
+	const table = asDrizzleTable('menu', menu);
+
+	await db.insert(table).values([
+		{ name: 'Pancakes', price: 9.5, type: 'Breakfast' },
+		{ name: 'French Toast', price: 11.25, type: 'Breakfast' },
+		{ name: 'Coffee', price: 3, type: 'Beverages' },
+		{ name: 'Cappuccino', price: 4.5, type: 'Beverages' },
+	]);
+}
diff --git a/packages/db/test/fixtures/integrations/integration/shared.ts b/packages/db/test/fixtures/integrations/integration/shared.ts
new file mode 100644
index 000000000..d46ae65a6
--- /dev/null
+++ b/packages/db/test/fixtures/integrations/integration/shared.ts
@@ -0,0 +1,9 @@
+import { column, defineTable } from 'astro:db';
+
+export const menu = defineTable({
+	columns: {
+		name: column.text(),
+		type: column.text(),
+		price: column.number(),
+	},
+});
diff --git a/packages/db/test/fixtures/integrations/package.json b/packages/db/test/fixtures/integrations/package.json
new file mode 100644
index 000000000..1bb17a8c7
--- /dev/null
+++ b/packages/db/test/fixtures/integrations/package.json
@@ -0,0 +1,14 @@
+{
+  "name": "@test/db-integration",
+  "version": "0.0.0",
+  "private": true,
+  "scripts": {
+    "dev": "astro dev",
+    "build": "astro build",
+    "preview": "astro preview"
+  },
+  "dependencies": {
+    "@astrojs/db": "workspace:*",
+    "astro": "workspace:*"
+  }
+}
diff --git a/packages/db/test/fixtures/integrations/src/pages/index.astro b/packages/db/test/fixtures/integrations/src/pages/index.astro
new file mode 100644
index 000000000..3e9c30ef7
--- /dev/null
+++ b/packages/db/test/fixtures/integrations/src/pages/index.astro
@@ -0,0 +1,17 @@
+---
+/// <reference path="../../.astro/db-types.d.ts" />
+import { Author, db, menu } from 'astro:db';
+
+const authors = await db.select().from(Author);
+const menuItems = await db.select().from(menu);
+---
+
+<h2>Authors</h2>
+<ul class="authors-list">
+	{authors.map((author) => <li>{author.name}</li>)}
+</ul>
+
+<h2>Menu</h2>
+<ul class="menu">
+	{menuItems.map((item) => <li>{item.name}</li>)}
+</ul>