fix(db-postgres): select query on upload fields with hasMany: true

payloadcms · Dec 17, 2024 · 5fdc0bc · 5fdc0bc
1 parent b0b2fc6
commit 5fdc0bc
Show file tree

Hide file tree

Showing 6 changed files with 176 additions and 2 deletions.
diff --git a/packages/drizzle/src/find/traverseFields.ts b/packages/drizzle/src/find/traverseFields.ts
@@ -526,7 +526,7 @@ export const traverseFields = ({
 
           if (
             !withTabledFields.rels &&
-            field.type === 'relationship' &&
+            (field.type === 'relationship' || field.type === 'upload') &&
             (field.hasMany || Array.isArray(field.relationTo))
           ) {
             withTabledFields.rels = true

diff --git a/test/select/collections/Posts/index.ts b/test/select/collections/Posts/index.ts
@@ -116,5 +116,33 @@ export const PostsCollection: CollectionConfig = {
         },
       ],
     },
+    {
+      type: 'relationship',
+      name: 'hasOne',
+      relationTo: 'posts',
+    },
+    {
+      type: 'relationship',
+      name: 'hasMany',
+      hasMany: true,
+      relationTo: 'posts',
+    },
+    {
+      type: 'upload',
+      name: 'hasManyUpload',
+      hasMany: true,
+      relationTo: 'posts',
+    },
+    {
+      type: 'relationship',
+      name: 'hasOnePoly',
+      relationTo: ['posts'],
+    },
+    {
+      type: 'relationship',
+      name: 'hasManyPoly',
+      hasMany: true,
+      relationTo: ['posts'],
+    },
   ],
 }
diff --git a/test/select/config.ts b/test/select/config.ts
@@ -23,6 +23,13 @@ export default buildConfigWithDefaults({
     DeepPostsCollection,
     Pages,
     Points,
+    {
+      slug: 'upload',
+      fields: [],
+      upload: {
+        staticDir: path.resolve(dirname, 'media'),
+      },
+    },
   ],
   globals: [
     {

diff --git a/test/select/image.jpg b/test/select/image.jpg
diff --git a/test/select/int.spec.ts b/test/select/int.spec.ts
@@ -133,6 +133,50 @@ describe('Select', () => {
         })
       })
 
+      it('should select relationships', async () => {
+        const res_1 = await payload.findByID({
+          collection: 'posts',
+          id: postId,
+          select: {
+            hasManyUpload: true,
+          },
+          depth: 0,
+        })
+
+        expect(res_1).toStrictEqual({
+          id: postId,
+          hasManyUpload: post.hasManyUpload,
+        })
+
+        const res_2 = await payload.findByID({
+          collection: 'posts',
+          id: postId,
+          select: {
+            hasOne: true,
+          },
+          depth: 0,
+        })
+
+        expect(res_2).toStrictEqual({
+          id: postId,
+          hasOne: post.hasOne,
+        })
+
+        const res_3 = await payload.findByID({
+          collection: 'posts',
+          id: postId,
+          select: {
+            hasManyPoly: true,
+          },
+          depth: 0,
+        })
+
+        expect(res_3).toStrictEqual({
+          id: postId,
+          hasManyPoly: post.hasManyPoly,
+        })
+      })
+
       it('should select all the fields inside of group', async () => {
         const res = await payload.findByID({
           collection: 'posts',
@@ -447,6 +491,29 @@ describe('Select', () => {
         expect(res).toStrictEqual(expected)
       })
 
+      it('should exclude relationships', async () => {
+        const res = await payload.findByID({
+          collection: 'posts',
+          id: postId,
+          select: {
+            hasOne: false,
+            hasMany: false,
+            hasManyPoly: false,
+            hasOnePoly: false,
+          },
+          depth: 0,
+        })
+
+        const expected = { ...post }
+
+        delete expected['hasOne']
+        delete expected['hasMany']
+        delete expected['hasManyPoly']
+        delete expected['hasOnePoly']
+
+        expect(res).toStrictEqual(expected)
+      })
+
       it('should exclude group', async () => {
         const res = await payload.findByID({
           collection: 'posts',
@@ -2129,7 +2196,15 @@ describe('Select', () => {
   })
 })
 
-function createPost() {
+async function createPost() {
+  const upload = await payload.create({
+    collection: 'upload',
+    data: {},
+    filePath: path.resolve(dirname, 'image.jpg'),
+  })
+
+  const relation = await payload.create({ depth: 0, collection: 'posts', data: {} })
+
   return payload.create({
     collection: 'posts',
     depth: 0,
@@ -2142,6 +2217,11 @@ function createPost() {
         number: 1,
         text: 'text',
       },
+      hasMany: [relation],
+      hasManyUpload: [upload],
+      hasOne: relation,
+      hasManyPoly: [{ relationTo: 'posts', value: relation }],
+      hasOnePoly: { relationTo: 'posts', value: relation },
       blocks: [
         {
           blockType: 'cta',

diff --git a/test/select/payload-types.ts b/test/select/payload-types.ts
@@ -17,6 +17,7 @@ export interface Config {
     'deep-posts': DeepPost;
     pages: Page;
     points: Point;
+    upload: Upload;
     users: User;
     'payload-locked-documents': PayloadLockedDocument;
     'payload-preferences': PayloadPreference;
@@ -30,6 +31,7 @@ export interface Config {
     'deep-posts': DeepPostsSelect<false> | DeepPostsSelect<true>;
     pages: PagesSelect<false> | PagesSelect<true>;
     points: PointsSelect<false> | PointsSelect<true>;
+    upload: UploadSelect<false> | UploadSelect<true>;
     users: UsersSelect<false> | UsersSelect<true>;
     'payload-locked-documents': PayloadLockedDocumentsSelect<false> | PayloadLockedDocumentsSelect<true>;
     'payload-preferences': PayloadPreferencesSelect<false> | PayloadPreferencesSelect<true>;
@@ -116,8 +118,39 @@ export interface Post {
   };
   unnamedTabText?: string | null;
   unnamedTabNumber?: number | null;
+  hasOne?: (string | null) | Post;
+  hasMany?: (string | Post)[] | null;
+  hasManyUpload?: (string | Upload)[] | null;
+  hasOnePoly?: {
+    relationTo: 'posts';
+    value: string | Post;
+  } | null;
+  hasManyPoly?:
+    | {
+        relationTo: 'posts';
+        value: string | Post;
+      }[]
+    | null;
+  updatedAt: string;
+  createdAt: string;
+}
+/**
+ * This interface was referenced by `Config`'s JSON-Schema
+ * via the `definition` "upload".
+ */
+export interface Upload {
+  id: string;
   updatedAt: string;
   createdAt: string;
+  url?: string | null;
+  thumbnailURL?: string | null;
+  filename?: string | null;
+  mimeType?: string | null;
+  filesize?: number | null;
+  width?: number | null;
+  height?: number | null;
+  focalX?: number | null;
+  focalY?: number | null;
 }
 /**
  * This interface was referenced by `Config`'s JSON-Schema
@@ -391,6 +424,10 @@ export interface PayloadLockedDocument {
         relationTo: 'points';
         value: string | Point;
       } | null)
+    | ({
+        relationTo: 'upload';
+        value: string | Upload;
+      } | null)
     | ({
         relationTo: 'users';
         value: string | User;
@@ -487,6 +524,11 @@ export interface PostsSelect<T extends boolean = true> {
       };
   unnamedTabText?: T;
   unnamedTabNumber?: T;
+  hasOne?: T;
+  hasMany?: T;
+  hasManyUpload?: T;
+  hasOnePoly?: T;
+  hasManyPoly?: T;
   updatedAt?: T;
   createdAt?: T;
 }
@@ -705,6 +747,23 @@ export interface PointsSelect<T extends boolean = true> {
   updatedAt?: T;
   createdAt?: T;
 }
+/**
+ * This interface was referenced by `Config`'s JSON-Schema
+ * via the `definition` "upload_select".
+ */
+export interface UploadSelect<T extends boolean = true> {
+  updatedAt?: T;
+  createdAt?: T;
+  url?: T;
+  thumbnailURL?: T;
+  filename?: T;
+  mimeType?: T;
+  filesize?: T;
+  width?: T;
+  height?: T;
+  focalX?: T;
+  focalY?: T;
+}
 /**
  * This interface was referenced by `Config`'s JSON-Schema
  * via the `definition` "users_select".