addressing PR comments

Author: ymc9

packages/language/src/utils.ts

@@ -445,7 +445,7 @@ export function getAllDeclarationsIncludingImports(documents: LangiumDocuments,
 export function getAuthDecl(decls: (DataModel | TypeDef)[]) {
     let authModel = decls.find((d) => hasAttribute(d, '@@auth'));
     if (!authModel) {
-        authModel = decls.find((d) => (isDataModel(d) || isTypeDef(d)) && d.name === 'User');
+        authModel = decls.find((d) => d.name === 'User');
     }
     return authModel;
 }

packages/runtime/src/client/client-impl.ts

@@ -234,7 +234,9 @@ export class ClientImpl<Schema extends SchemaDef> {
     }
 
     async $connect() {
-        await this.kysely.connection().execute(async () => {});
+        await this.kysely.connection().execute(async (conn) => {
+            await conn.executeQuery(sql`select 1`.compile(this.kysely));
+        });
     }
 
     async $disconnect() {

packages/runtime/src/client/executor/zenstack-query-executor.ts

@@ -60,7 +60,7 @@ export class ZenStackQueryExecutor<Schema extends SchemaDef> extends DefaultQuer
         }
     }
 
-    private schemaHasMappedNames<Schema extends SchemaDef>(schema: Schema) {
+    private schemaHasMappedNames(schema: Schema) {
         const hasMapAttr = (decl: ModelDef | TypeDefDef) => {
             if (decl.attributes?.some((attr) => attr.name === '@@map')) {
                 return true;

tests/e2e/orm/client-api/connect-disconnect.test.ts

@@ -0,0 +1,29 @@
+import { createTestClient } from '@zenstackhq/testtools';
+import { describe, expect, it } from 'vitest';
+
+describe('Client $connect and $disconnect tests', () => {
+    it('works with connect and disconnect', async () => {
+        const db = await createTestClient(
+            `
+        model User {
+            id String @id @default(cuid())
+            email String @unique
+        }
+        `,
+        );
+
+        // connect to the database
+        await db.$connect();
+
+        // perform a simple operation
+        await db.user.create({
+            data: {
+                email: '[email protected]',
+            },
+        });
+
+        await db.$disconnect();
+
+        await expect(db.user.findFirst()).rejects.toThrow();
+    });
+});