@aurios/mizzle 1.1.2 → 1.1.3

package/test/unified-update.integration.test.ts

@@ -0,0 +1,139 @@
+import { describe, it, expect, beforeAll, afterAll } from "vitest";
+import { dynamoTable, dynamoEntity } from "@aurios/mizzle/table";
+import { string, uuid, number, stringSet } from "@aurios/mizzle/columns";
+import { prefixKey, staticKey } from "@aurios/mizzle";
+import { DynamoDBClient, CreateTableCommand, DeleteTableCommand, DescribeTableCommand } from "@aws-sdk/client-dynamodb";
+import { mizzle } from "@aurios/mizzle/db";
+import { eq, and } from "@aurios/mizzle/expressions/operators";
+
+const client = new DynamoDBClient({
+  endpoint: "http://localhost:8000",
+  region: "us-east-1",
+  credentials: {
+    accessKeyId: "local",
+    secretAccessKey: "local",
+  },
+});
+
+describe("Unified Update Integration", () => {
+  const tableName = "UnifiedUpdateTable";
+  const table = dynamoTable(tableName, {
+    pk: string("pk"),
+    sk: string("sk"),
+  });
+
+  const user = dynamoEntity(
+    table,
+    "User",
+    {
+      id: uuid(),
+      name: string(),
+      age: number(),
+      tags: stringSet(),
+    },
+    (cols) => ({
+      pk: prefixKey("USER#", cols.id),
+      sk: staticKey("METADATA"),
+    }),
+  );
+
+  beforeAll(async () => {
+    try {
+      await client.send(
+        new CreateTableCommand({
+          TableName: tableName,
+          KeySchema: [
+            { AttributeName: "pk", KeyType: "HASH" },
+            { AttributeName: "sk", KeyType: "RANGE" },
+          ],
+          AttributeDefinitions: [
+            { AttributeName: "pk", AttributeType: "S" },
+            { AttributeName: "sk", AttributeType: "S" },
+          ],
+          ProvisionedThroughput: {
+            ReadCapacityUnits: 5,
+            WriteCapacityUnits: 5,
+          },
+        }),
+      );
+    } catch {
+      /* ignore */
+    }
+  });
+
+  afterAll(async () => {
+    try {
+      await client.send(new DeleteTableCommand({ TableName: tableName }));
+    } catch {
+      /* ignore */
+    }
+  });
+
+  it("should support all unified update operations", async () => {
+    const db = mizzle(client);
+
+    // 1. Setup
+    const newUser = await db
+      .insert(user)
+      .values({
+        name: "Unified Alice",
+        age: 30,
+        tags: new Set(["initial"]),
+      })
+      .returning()
+      .execute();
+
+    // 2. SET multiple fields
+    await db
+      .update(user)
+      .set({ name: "Unified Alice Smith", age: 31 })
+      .where(eq(user.id, newUser.id))
+      .execute();
+
+    // 3. ADD to numeric field
+    await db.update(user).add({ age: 1 }).where(eq(user.id, newUser.id)).execute();
+
+    // 4. ADD to set field
+    await db
+      .update(user)
+      .add({ tags: new Set(["added"]) })
+      .where(eq(user.id, newUser.id))
+      .execute();
+
+    // 5. REMOVE field
+    await db.update(user).remove("name").where(eq(user.id, newUser.id)).execute();
+
+    // 6. DELETE from set
+    await db
+      .update(user)
+      .delete({ tags: new Set(["initial"]) })
+      .where(eq(user.id, newUser.id))
+      .execute();
+
+    // 7. Conditional update (should only update if age is 32)
+    // Current implementation might ignore the age condition if it's not part of the key
+    await db
+      .update(user)
+      .set({ name: "Conditional Alice" })
+      .where(and(eq(user.id, newUser.id), eq(user.age, 32)))
+      .execute();
+
+    // 8. Conditional update that should fail (age is not 40)
+    try {
+      await db
+        .update(user)
+        .set({ name: "Should Not Happen" })
+        .where(and(eq(user.id, newUser.id), eq(user.age, 40)))
+        .execute();
+    } catch {
+      // Expected to fail if ConditionExpression is implemented
+    }
+
+    // Verification
+    const final = await db.select().from(user).where(eq(user.id, newUser.id));
+    expect(final[0]!.name).toBe("Conditional Alice");
+    expect(final[0]!.age).toBe(32);
+    expect(final[0]!.tags).toContain("added");
+    expect(final[0]!.tags).not.toContain("initial");
+  });
+});

package/test/update.integration.test.ts

@@ -0,0 +1,132 @@
+import { describe, it, expect, beforeAll, afterAll } from "vitest";
+import { dynamoTable, dynamoEntity } from "@aurios/mizzle/table";
+import { string, uuid, number, list } from "@aurios/mizzle/columns";
+import { prefixKey, staticKey } from "@aurios/mizzle";
+import { DynamoDBClient, CreateTableCommand, DeleteTableCommand, DescribeTableCommand } from "@aws-sdk/client-dynamodb";
+import { mizzle } from "@aurios/mizzle/db";
+import { eq } from "@aurios/mizzle/expressions/operators";
+
+const client = new DynamoDBClient({
+  endpoint: "http://localhost:8000",
+  region: "us-east-1",
+  credentials: {
+    accessKeyId: "local",
+    secretAccessKey: "local",
+  },
+});
+
+describe("Update Integration", () => {
+  const tableName = "UpdateIntegrationTable";
+  const table = dynamoTable(tableName, {
+    pk: string("pk"),
+    sk: string("sk"),
+  });
+
+  const user = dynamoEntity(
+    table,
+    "User",
+    {
+      id: uuid(),
+      name: string(),
+      age: number(),
+      roles: list(),
+    },
+    (cols) => ({
+      pk: prefixKey("USER#", cols.id),
+      sk: staticKey("METADATA"),
+    }),
+  );
+
+  beforeAll(async () => {
+    try {
+      await client.send(
+        new CreateTableCommand({
+          TableName: tableName,
+          KeySchema: [
+            { AttributeName: "pk", KeyType: "HASH" },
+            { AttributeName: "sk", KeyType: "RANGE" },
+          ],
+          AttributeDefinitions: [
+            { AttributeName: "pk", AttributeType: "S" },
+            { AttributeName: "sk", AttributeType: "S" },
+          ],
+          ProvisionedThroughput: {
+            ReadCapacityUnits: 5,
+            WriteCapacityUnits: 5,
+          },
+        }),
+      );
+    } catch {
+      /* ignore */
+    }
+  });
+
+  afterAll(async () => {
+    try {
+      await client.send(new DeleteTableCommand({ TableName: tableName }));
+    } catch {
+      /* ignore */
+    }
+  });
+
+  it("should perform a full update lifecycle", async () => {
+    const db = mizzle(client);
+
+    // 1. Setup - Insert a user
+    const newUser = await db
+      .insert(user)
+      .values({
+        name: "Alice",
+        age: 25,
+        roles: ["user"],
+      })
+      .returning()
+      .execute();
+
+    expect(newUser.id).toBeDefined();
+
+    // 2. SET
+    const setRes = await db
+      .update(user)
+      .set({ name: "Alice Smith" })
+      .where(eq(user.id, newUser.id))
+      .returning("ALL_NEW")
+      .execute();
+
+    const res1 = setRes as Record<string, unknown>;
+    expect(res1.name).toBe("Alice Smith");
+    expect(res1.age).toBe(25);
+
+    // 3. ADD
+    const addRes = await db
+      .update(user)
+      .add({ age: 5 })
+      .where(eq(user.id, newUser.id))
+      .returning("UPDATED_NEW")
+      .execute();
+
+    const res2 = addRes as Record<string, unknown>;
+    expect(res2.age).toBe(30);
+
+    // 4. REMOVE
+    const removeRes = await db
+      .update(user)
+      .remove("roles")
+      .where(eq(user.id, newUser.id))
+      .returning("ALL_NEW")
+      .execute();
+
+    const res3 = removeRes as Record<string, unknown>;
+    expect(res3.roles).toBeUndefined();
+    expect(res3.name).toBe("Alice Smith");
+
+    // 5. Verification - Select
+    const final = await db.select().from(user).where(eq(user.id, newUser.id));
+    expect(final[0]!).toMatchObject({
+      id: newUser.id,
+      name: "Alice Smith",
+      age: 30,
+    });
+    expect(final[0]!.roles).toBeUndefined();
+  });
+});

package/tsconfig.json

@@ -1,10 +1,14 @@
 {
-  "extends": "@mizzle/tsconfig/base.json",
-  "compilerOptions": {
-    "outDir": "./dist",
-    "paths": {
-      "@mizzle/shared": ["./../shared/src/index.ts"]
-    }
-  },
-  "include": ["src"]
+    "extends": "@repo/typescript-config/base.json",
+    "compilerOptions": {
+        "outDir": "./dist",
+        "paths": {
+            "@repo/shared": [
+                "./../shared/src/index.ts"
+            ]
+        }
+    },
+    "include": [
+        "src"
+    ]
 }

package/tsup.config.ts

@@ -1,20 +1,20 @@
-import { defineConfig } from 'tsup';
+import { defineConfig } from "tsup";
 
 export default defineConfig({
   entry: {
-    index: 'src/index.ts',
-    columns: 'src/columns/index.ts',
-    table: 'src/core/table.ts',
-    snapshot: 'src/core/snapshot.ts',
-    diff: 'src/core/diff.ts',
-    introspection: 'src/core/introspection.ts',
-    db: 'src/db.ts',
+    index: "src/index.ts",
+    columns: "src/columns/index.ts",
+    table: "src/core/table.ts",
+    snapshot: "src/core/snapshot.ts",
+    diff: "src/core/diff.ts",
+    introspection: "src/core/introspection.ts",
+    db: "src/db.ts",
   },
-  format: 'esm',
+  format: "esm",
   dts: true,
   clean: true,
-  noExternal: ['@mizzle/shared'],
-  external: [/^@aws-sdk\//, 'uuid'],
+  noExternal: ["@repo/shared"],
+  external: [/^@aws-sdk\//, "uuid"],
   minify: true,
   splitting: true,
 });

package/vitest.config.ts

@@ -0,0 +1,8 @@
+import { defineConfig } from "vitest/config";
+import { sharedConfig } from "@repo/vitest-config";
+export default defineConfig({
+  ...sharedConfig,
+  test: {
+    ...sharedConfig.test,
+  },
+});