navidrome/ui/src/dialogs/AddToPlaylistDialog.test.js

import * as React from 'react'
import { TestContext } from 'ra-test'
import { DataProviderContext } from 'react-admin'
import { cleanup, fireEvent, render, waitFor } from '@testing-library/react'
import { AddToPlaylistDialog } from './AddToPlaylistDialog'

describe('AddToPlaylistDialog', () => {
  afterEach(cleanup)

  let mockData = [
    { id: 'sample-id1', name: 'sample playlist 1', owner: 'admin' },
    { id: 'sample-id2', name: 'sample playlist 2', owner: 'admin' },
  ]
  let mockIndexedData = {
    'sample-id1': {
      id: 'sample-id1',
      name: 'sample playlist 1',
      owner: 'admin',
    },
    'sample-id2': {
      id: 'sample-id2',
      name: 'sample playlist 2',
      owner: 'admin',
    },
  }
  let selectedIds = ['song-1', 'song-2']

  it('adds distinct songs to already existing playlists', async () => {
    let mockDataProvider = {
      getList: jest.fn(() =>
        Promise.resolve({ data: mockData, total: mockData.length })
      ),
      getOne: jest.fn(() =>
        Promise.resolve({ data: { id: 'song-3' }, total: 1 })
      ),
      create: jest.fn(() =>
        Promise.resolve({ data: { id: 'created-id', name: 'created-name' } })
      ),
    }
    const testutils = render(
      <DataProviderContext.Provider value={mockDataProvider}>
        <TestContext
          initialState={{
            addToPlaylistDialog: {
              open: true,
              duplicateSong: false,
              selectedIds: selectedIds,
            },
            admin: {
              ui: { optimistic: false },
              resources: {
                playlist: {
                  data: mockIndexedData,
                  list: {
                    cachedRequests: {
                      '{"pagination":{"page":1,"perPage":-1},"sort":{"field":"name","order":"ASC"},"filter":{}}': {
                        ids: ['sample-id1', 'sample-id2'],
                        total: 2,
                      },
                    },
                  },
                },
              },
            },
          }}
        >
          <AddToPlaylistDialog />
        </TestContext>
      </DataProviderContext.Provider>
    )

    fireEvent.change(document.activeElement, { target: { value: 'sample' } })
    fireEvent.keyDown(document.activeElement, { key: 'ArrowDown' })
    fireEvent.keyDown(document.activeElement, { key: 'Enter' })
    fireEvent.keyDown(document.activeElement, { key: 'ArrowDown' })
    fireEvent.keyDown(document.activeElement, { key: 'Enter' })
    await waitFor(() => {
      expect(testutils.getByTestId('playlist-add')).not.toBeDisabled()
    })
    fireEvent.click(testutils.getByTestId('playlist-add'))
    await waitFor(() => {
      expect(mockDataProvider.create).toHaveBeenNthCalledWith(
        1,
        'playlistTrack',
        {
          data: { ids: selectedIds },
          filter: { playlist_id: 'sample-id1' },
        }
      )
    })
    await waitFor(() => {
      expect(mockDataProvider.create).toHaveBeenNthCalledWith(
        2,
        'playlistTrack',
        {
          data: { ids: selectedIds },
          filter: { playlist_id: 'sample-id2' },
        }
      )
    })
  })

  let mockDataProvider = {
    getList: jest.fn(() =>
      Promise.resolve({ data: mockData, total: mockData.length })
    ),
    getOne: jest.fn(() =>
      Promise.resolve({ data: { id: 'song-3' }, total: 1 })
    ),
    create: jest.fn(() =>
      Promise.resolve({ data: { id: 'created-id1', name: 'created-name' } })
    ),
  }

  it('adds distinct songs to a new playlist', async () => {
    const testutils = render(
      <DataProviderContext.Provider value={mockDataProvider}>
        <TestContext
          initialState={{
            addToPlaylistDialog: {
              open: true,
              duplicateSong: false,
              selectedIds: selectedIds,
            },
            admin: {
              ui: { optimistic: false },
              resources: {
                playlist: {
                  data: mockIndexedData,
                  list: {
                    cachedRequests: {
                      '{"pagination":{"page":1,"perPage":-1},"sort":{"field":"name","order":"ASC"},"filter":{}}': {
                        ids: ['sample-id1', 'sample-id2'],
                        total: 2,
                      },
                    },
                  },
                },
              },
            },
          }}
        >
          <AddToPlaylistDialog />
        </TestContext>
      </DataProviderContext.Provider>
    )

    fireEvent.change(document.activeElement, { target: { value: 'sample' } })
    fireEvent.keyDown(document.activeElement, { key: 'ArrowDown' })
    fireEvent.keyDown(document.activeElement, { key: 'ArrowDown' })
    fireEvent.keyDown(document.activeElement, { key: 'ArrowDown' })
    fireEvent.keyDown(document.activeElement, { key: 'Enter' })
    await waitFor(() => {
      expect(testutils.getByTestId('playlist-add')).not.toBeDisabled()
    })
    fireEvent.click(testutils.getByTestId('playlist-add'))
    await waitFor(() => {
      expect(mockDataProvider.create).toHaveBeenNthCalledWith(1, 'playlist', {
        data: { name: 'sample' },
      })
      expect(mockDataProvider.create).toHaveBeenNthCalledWith(
        2,
        'playlistTrack',
        {
          data: { ids: selectedIds },
          filter: { playlist_id: 'created-id1' },
        }
      )
    })
  })

  it('adds distinct songs to multiple new playlists', async () => {
    const testutils = render(
      <DataProviderContext.Provider value={mockDataProvider}>
        <TestContext
          initialState={{
            addToPlaylistDialog: {
              open: true,
              duplicateSong: false,
              selectedIds: selectedIds,
            },
            admin: {
              ui: { optimistic: false },
              resources: {
                playlist: {
                  data: mockIndexedData,
                  list: {
                    cachedRequests: {
                      '{"pagination":{"page":1,"perPage":-1},"sort":{"field":"name","order":"ASC"},"filter":{}}': {
                        ids: ['sample-id1', 'sample-id2'],
                        total: 2,
                      },
                    },
                  },
                },
              },
            },
          }}
        >
          <AddToPlaylistDialog />
        </TestContext>
      </DataProviderContext.Provider>
    )

    fireEvent.change(document.activeElement, { target: { value: 'sample' } })
    fireEvent.keyDown(document.activeElement, { key: 'ArrowDown' })
    fireEvent.keyDown(document.activeElement, { key: 'ArrowDown' })
    fireEvent.keyDown(document.activeElement, { key: 'ArrowDown' })
    fireEvent.keyDown(document.activeElement, { key: 'Enter' })
    fireEvent.change(document.activeElement, {
      target: { value: 'new playlist' },
    })
    fireEvent.keyDown(document.activeElement, { key: 'Enter' })
    await waitFor(() => {
      expect(testutils.getByTestId('playlist-add')).not.toBeDisabled()
    })
    fireEvent.click(testutils.getByTestId('playlist-add'))
    await waitFor(() => {
      expect(mockDataProvider.create).toHaveBeenCalledTimes(4)
    })
  })
})