import { AsynchronousJobStatus } from '../generated/index'
import { delay } from './delay'
import { SynapseClientError } from './SynapseClientError'
import { waitForAsyncResult } from './waitForAsyncResult'

vi.mock('./delay')

const mockDelay = vi.mocked(delay)

describe('waitForAsyncResult', () => {
  test('waits for the async result to be no longer PROCESSING', async () => {
    const mockGetAsyncResult = vi.fn()
    mockGetAsyncResult
      .mockResolvedValueOnce({ jobState: 'PROCESSING' })
      .mockResolvedValueOnce({ jobState: 'PROCESSING' })
      .mockResolvedValueOnce({ jobState: 'PROCESSING' })
      .mockResolvedValueOnce({ jobState: 'PROCESSING' })
      .mockResolvedValueOnce({ jobState: 'PROCESSING' })
      .mockResolvedValueOnce({ jobState: 'COMPLETED', result: 'foo' })

    const result = await waitForAsyncResult(mockGetAsyncResult)

    expect(result).toEqual({ jobState: 'COMPLETED', result: 'foo' })
    expect(mockGetAsyncResult).toHaveBeenCalledTimes(6)
    expect(mockDelay).toHaveBeenCalledTimes(5)
    expect(mockDelay).toHaveBeenNthCalledWith(1, 125)
    expect(mockDelay).toHaveBeenNthCalledWith(2, 250)
    expect(mockDelay).toHaveBeenNthCalledWith(3, 500)
    expect(mockDelay).toHaveBeenNthCalledWith(4, 1000)
    expect(mockDelay).toHaveBeenNthCalledWith(4, 1000)
  })

  test('throws error when the job fails', async () => {
    const mockFailedResult: AsynchronousJobStatus = {
      requestBody: {
        concreteType: 'org.sagebionetworks.repo.model.table.QueryBundleRequest',
      },
      jobState: 'FAILED',
      errorMessage: 'Job failed',
      errorDetails: 'Some error details',
    }
    const mockGetAsyncResult = vi.fn()
    mockGetAsyncResult.mockResolvedValueOnce(mockFailedResult)

    const result = waitForAsyncResult(mockGetAsyncResult)

    await expect(result).rejects.toThrow(
      new SynapseClientError(
        400,
        'Job failed',
        'waitForAsyncResult - org.sagebionetworks.repo.model.table.QueryBundleRequest',
      ),
    )
  })

  test('include errorDetails when errorMessage is missing', async () => {
    const mockFailedResult: AsynchronousJobStatus = {
      requestBody: {
        concreteType: 'org.sagebionetworks.repo.model.table.QueryBundleRequest',
      },
      jobState: 'FAILED',
      errorMessage: undefined,
      errorDetails: 'Some error details',
    }
    const mockGetAsyncResult = vi.fn()
    mockGetAsyncResult.mockResolvedValueOnce(mockFailedResult)

    const result = waitForAsyncResult(mockGetAsyncResult)

    await expect(result).rejects.toThrow(
      new SynapseClientError(
        400,
        'Some error details',
        'waitForAsyncResult - org.sagebionetworks.repo.model.table.QueryBundleRequest',
      ),
    )
  })
})