test: migrate more migration cases, a few minor fixes (#293)

* test: migrate more migration cases, a few minor fixes

* update

* update

* update

* update

* update

* update

Author: ymc9

packages/language/package.json

@@ -11,6 +11,7 @@
     "type": "module",
     "scripts": {
         "build": "pnpm langium:generate && tsc --noEmit && tsup-node",
+        "watch": "tsup-node --watch",
         "lint": "eslint src --ext ts",
         "langium:generate": "langium generate",
         "langium:generate:production": "langium generate --mode=production",

packages/language/res/stdlib.zmodel

@@ -676,7 +676,7 @@ attribute @@allow(_ operation: String @@@completionHint(["'create'", "'read'", "
  * @param condition: a boolean expression that controls if the operation should be allowed.
  * @param override: a boolean value that controls if the field-level policy should override the model-level policy.
  */
-attribute @allow(_ operation: String @@@completionHint(["'create'", "'read'", "'update'", "'delete'", "'all'"]), _ condition: Boolean, _ override: Boolean?)
+// attribute @allow(_ operation: String @@@completionHint(["'create'", "'read'", "'update'", "'delete'", "'all'"]), _ condition: Boolean, _ override: Boolean?)
 
 /**
  * Defines an access policy that denies a set of operations when the given condition is true.
@@ -692,7 +692,7 @@ attribute @@deny(_ operation: String @@@completionHint(["'create'", "'read'", "'
  * @param operation: comma-separated list of "create", "read", "update", "delete". Use "all" to denote all operations.
  * @param condition: a boolean expression that controls if the operation should be denied.
  */
-attribute @deny(_ operation: String @@@completionHint(["'create'", "'read'", "'update'", "'delete'", "'all'"]), _ condition: Boolean)
+// attribute @deny(_ operation: String @@@completionHint(["'create'", "'read'", "'update'", "'delete'", "'all'"]), _ condition: Boolean)
 
 /**
  * Checks if the current user can perform the given operation on the given field.

packages/language/src/zmodel-scope.ts

@@ -36,8 +36,8 @@ import {
     getAuthDecl,
     getRecursiveBases,
     isAuthInvocation,
-    isCollectionPredicate,
     isBeforeInvocation,
+    isCollectionPredicate,
     resolveImportUri,
 } from './utils';
 
@@ -75,7 +75,7 @@ export class ZModelScopeComputation extends DefaultScopeComputation {
 
     override processNode(node: AstNode, document: LangiumDocument<AstNode>, scopes: PrecomputedScopes) {
         super.processNode(node, document, scopes);
-        if (isDataModel(node)) {
+        if (isDataModel(node) || isTypeDef(node)) {
             // add base fields to the scope recursively
             const bases = getRecursiveBases(node);
             for (const base of bases) {

packages/runtime/src/client/crud/operations/base.ts

@@ -939,15 +939,18 @@ export abstract class BaseOperationHandler<Schema extends SchemaDef> {
             combinedWhere = Object.keys(combinedWhere).length > 0 ? { AND: [parentWhere, combinedWhere] } : parentWhere;
         }
 
-        // fill in automatically updated fields
         const modelDef = this.requireModel(model);
         let finalData = data;
+
+        // fill in automatically updated fields
+        const autoUpdatedFields: string[] = [];
         for (const [fieldName, fieldDef] of Object.entries(modelDef.fields)) {
             if (fieldDef.updatedAt) {
                 if (finalData === data) {
                     finalData = clone(data);
                 }
                 finalData[fieldName] = this.dialect.transformPrimitive(new Date(), 'DateTime', false);
+                autoUpdatedFields.push(fieldName);
             }
         }
 
@@ -1027,7 +1030,13 @@ export abstract class BaseOperationHandler<Schema extends SchemaDef> {
             }
         }
 
-        if (Object.keys(updateFields).length === 0) {
+        let hasFieldUpdate = Object.keys(updateFields).length > 0;
+        if (hasFieldUpdate) {
+            // check if only updating auto-updated fields, if so, we can skip the update
+            hasFieldUpdate = Object.keys(updateFields).some((f) => !autoUpdatedFields.includes(f));
+        }
+
+        if (!hasFieldUpdate) {
             // nothing to update, return the filter so that the caller can identify the entity
             return combinedWhere;
         } else {
@@ -2073,22 +2082,11 @@ export abstract class BaseOperationHandler<Schema extends SchemaDef> {
         }
     }
 
-    // Given a unique filter of a model, return the entity ids by trying to
-    // reused the filter if it's a complete id filter (without extra fields)
-    // otherwise, read the entity by the filter
+    // Given a unique filter of a model, load the entity and return its id fields
     private getEntityIds(kysely: ToKysely<Schema>, model: GetModels<Schema>, uniqueFilter: any) {
-        const idFields: string[] = requireIdFields(this.schema, model);
-        if (
-            // all id fields are provided
-            idFields.every((f) => f in uniqueFilter && uniqueFilter[f] !== undefined) &&
-            // no non-id filter exists
-            Object.keys(uniqueFilter).every((k) => idFields.includes(k))
-        ) {
-            return uniqueFilter;
-        }
-
         return this.readUnique(kysely, model, {
             where: uniqueFilter,
+            select: this.makeIdSelect(model),
         });
     }
 

packages/runtime/src/client/crud/validator/index.ts

@@ -976,9 +976,14 @@ export class InputValidator<Schema extends SchemaDef> {
                       ])
                       .optional();
 
+            let upsertWhere = this.makeWhereSchema(fieldType, true);
+            if (!fieldDef.array) {
+                // to-one relation, can upsert without where clause
+                upsertWhere = upsertWhere.optional();
+            }
             fields['upsert'] = this.orArray(
                 z.strictObject({
-                    where: this.makeWhereSchema(fieldType, true),
+                    where: upsertWhere,
                     create: this.makeCreateDataSchema(fieldType, false, withoutFields),
                     update: this.makeUpdateDataSchema(fieldType, withoutFields),
                 }),

packages/testtools/src/client.ts

@@ -58,7 +58,6 @@ export async function createTestClient<Schema extends SchemaDef>(
     const provider = options?.provider ?? getTestDbProvider() ?? 'sqlite';
 
     const dbName = options?.dbName ?? getTestDbName(provider);
-    console.log(`Using provider: ${provider}, db: ${dbName}`);
 
     const dbUrl =
         provider === 'sqlite'

pnpm-lock.yaml

@@ -38,7 +38,8 @@ describe('Client update tests', () => {
                 email: user.email,
                 name: user.name,
             });
-            expect(updated.updatedAt.getTime()).toBeGreaterThan(user.updatedAt.getTime());
+            // should not update updatedAt
+            expect(updated.updatedAt.getTime()).toEqual(user.updatedAt.getTime());
 
             // id as filter
             updated = await client.user.update({
@@ -114,6 +115,21 @@ describe('Client update tests', () => {
             ).resolves.toMatchObject({ id: 'user2' });
         });
 
+        it('does not update updatedAt if no other scalar fields are updated', async () => {
+            const user = await createUser(client, '[email protected]');
+            const originalUpdatedAt = user.updatedAt;
+
+            await client.user.update({
+                where: { id: user.id },
+                data: {
+                    posts: { create: { title: 'Post1' } },
+                },
+            });
+
+            const updatedUser = await client.user.findUnique({ where: { id: user.id } });
+            expect(updatedUser?.updatedAt).toEqual(originalUpdatedAt);
+        });
+
         it('works with numeric incremental update', async () => {
             await createUser(client, '[email protected]', {
                 profile: { create: { id: '1', bio: 'bio' } },

tests/e2e/orm/client-api/update.test.ts

@@ -294,7 +294,7 @@ model Bar {
 
             await db.$unuseAll().foo.create({ data: { id: 1 } });
             await expect(db.foo.findMany()).resolves.toHaveLength(0);
-            await db.foo.update({ where: { id: 1 }, data: { bar: { create: { id: 1, y: 0 } } } });
+            await db.$unuseAll().foo.update({ where: { id: 1 }, data: { bar: { create: { id: 1, y: 0 } } } });
             await expect(db.foo.findMany()).resolves.toHaveLength(1);
         });
 
@@ -321,7 +321,7 @@ model Bar {
 
             await db.$unuseAll().foo.create({ data: { id: 1, bars: { create: [{ id: 1, y: 0 }] } } });
             await expect(db.foo.findMany()).resolves.toHaveLength(0);
-            await db.foo.update({ where: { id: 1 }, data: { bars: { create: { id: 2, y: 1 } } } });
+            await db.$unuseAll().foo.update({ where: { id: 1 }, data: { bars: { create: { id: 2, y: 1 } } } });
             await expect(db.foo.findMany()).resolves.toHaveLength(1);
         });
 

tests/e2e/orm/policy/crud/read.test.ts

@@ -10,7 +10,7 @@ describe('prisma omit', () => {
               name String
               profile Profile?
               age Int
-              value Int @allow('read', age > 20)
+              value Int
               @@allow('all', age > 18)
             }