Skip to content

Commit

Permalink
fix: Use client.query in models.folder
Browse files Browse the repository at this point in the history
  • Loading branch information
doubleface committed Feb 13, 2024
1 parent 09fc1a8 commit 4db3e0c
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 30 deletions.
15 changes: 7 additions & 8 deletions packages/cozy-client/src/models/folder.js
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,8 @@ import sortBy from 'lodash/sortBy'
import CozyClient from '../CozyClient'
import { DOCTYPE_FILES } from '../const'

import { Q } from '../queries/dsl'

const APP_DOCTYPE = 'io.cozy.apps'
const administrative = 'administrative'
const photos = 'photos'
Expand Down Expand Up @@ -88,16 +90,13 @@ export const createFolderWithReference = async (client, path, document) => {
* @returns {Promise<import("../types").IOCozyFolder>} Folder referenced by the given document
*/
export const getReferencedFolder = async (client, document) => {
const { included } = await client
.collection(DOCTYPE_FILES)
.findReferencedBy(document)
const foldersOutsideTrash = included.filter(
folder => !/^\/\.cozy_trash/.test(folder.path)
const { included: folders } = await client.query(
Q(DOCTYPE_FILES)
.partialIndex({ type: 'directory', trashed: false })
.referencedBy(document)
)

// there can be multiple folders with the same reference in some edge cases
// when this happens we return the most recent one
return foldersOutsideTrash.length > 0
? sortBy(foldersOutsideTrash, 'created_at').pop()
: null
return folders.length > 0 ? sortBy(folders, 'created_at').pop() : null
}
26 changes: 4 additions & 22 deletions packages/cozy-client/src/models/folder.spec.js
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@ import { MAGIC_FOLDERS, ensureMagicFolder, getReferencedFolder } from './folder'
describe('Folder model', () => {
const mockClient = {
collection: () => mockClient,
query: jest.fn(),
findReferencedBy: jest.fn(),
ensureDirectoryExists: jest.fn(),
addReferencesTo: jest.fn(),
Expand Down Expand Up @@ -38,7 +39,7 @@ describe('Folder model', () => {
}
]

mockClient.findReferencedBy.mockResolvedValue({
mockClient.query.mockResolvedValueOnce({
included: existingMagicFolders
})

Expand Down Expand Up @@ -66,10 +67,9 @@ describe('Folder model', () => {
})

it('should create magic folder', async () => {
mockClient.findReferencedBy.mockResolvedValue({
mockClient.query.mockResolvedValueOnce({
included: []
})

mockClient.ensureDirectoryExists.mockResolvedValue('dir-id')
mockClient.get.mockResolvedValue({ data: {} })

Expand All @@ -90,24 +90,6 @@ describe('Folder model', () => {
})

describe('getReferencedFolder', () => {
it('should filter trashed folders', async () => {
const referencedFolder = {
path: '/Reference/Folder'
}

const trashFolder = {
path: '/.cozy_trash/Old/Reference/Folder'
}

mockClient.findReferencedBy.mockResolvedValue({
included: [referencedFolder, trashFolder]
})

const result = await getReferencedFolder(mockClient, 'ref')

expect(result).toBe(referencedFolder)
})

it('should return the most recently created folder', async () => {
const oldFolder = {
path: '/Old/folder',
Expand All @@ -118,7 +100,7 @@ describe('Folder model', () => {
created_at: '2020-03-02T14:57:07.661588+01:00'
}

mockClient.findReferencedBy.mockResolvedValue({
mockClient.query.mockResolvedValueOnce({
included: [oldFolder, recentFolder]
})

Expand Down

0 comments on commit 4db3e0c

Please sign in to comment.