supabase/studio/tests/pages/projects/LogsPreviewer.test.js

import dayjs from 'dayjs'
import utc from 'dayjs/plugin/utc'

dayjs.extend(utc)

import { get } from 'lib/common/fetch'
import { useRouter } from 'next/router'

const defaultRouterMock = () => {
  const router = jest.fn()
  router.query = {}
  router.push = jest.fn()
  router.pathname = 'logs/path'
  return router
}
useRouter.mockReturnValue(defaultRouterMock())

import LogsPreviewer from 'components/interfaces/Settings/Logs/LogsPreviewer'
import { useProjectSubscription } from 'hooks'
import { fireEvent, waitFor, screen, act } from '@testing-library/react'
import { render } from '../../helpers'
import userEvent from '@testing-library/user-event'
import { wait } from '@testing-library/user-event/dist/utils'
import { logDataFixture } from '../../fixtures'
import { LogsTableName } from 'components/interfaces/Settings/Logs'

beforeEach(() => {
  // reset mocks between tests
  get.mockReset()
  useRouter.mockReset()
  useRouter.mockReturnValue(defaultRouterMock())
})

test.each([
  {
    queryType: 'api',
    tableName: undefined,
    tableLog: logDataFixture({
      path: 'some-path',
      method: 'POST',
      status_code: '400',
      metadata: undefined,
    }),
    selectionLog: logDataFixture({
      metadata: [{ request: [{ method: 'POST' }] }],
    }),
    tableTexts: [/POST/, /some\-path/, /400/],
    selectionTexts: [/POST/, /Timestamp/, RegExp(`${new Date().getFullYear()}.+`, 'g')],
  },
  {
    queryType: 'database',
    tableName: undefined,
    tableLog: logDataFixture({
      error_severity: 'ERROR',
      event_message: 'some db event',
      metadata: undefined,
    }),
    selectionLog: logDataFixture({
      metadata: [
        {
          parsed: [
            {
              application_type: 'client backend',
              error_severity: 'ERROR',
              hint: 'some pg hint',
            },
          ],
        },
      ],
    }),
    tableTexts: [/ERROR/, /some db event/],
    selectionTexts: [
      /client backend/,
      /some pg hint/,
      /ERROR/,
      /Timestamp/,
      RegExp(`${new Date().getFullYear()}.+`, 'g'),
    ],
  },
  {
    queryType: 'auth',
    tableName: undefined,
    tableLog: logDataFixture({
      event_message: JSON.stringify({
        msg: 'some message',
        path: '/auth-path',
        level: 'info',
        status: 300,
      }),
    }),
    selectionLog: logDataFixture({
      event_message: JSON.stringify({
        msg: 'some message',
        path: '/auth-path',
        level: 'info',
        status: 300,
      }),
    }),
    tableTexts: [/auth\-path/, /some message/, /INFO/],
    selectionTexts: [
      /auth\-path/,
      /some message/,
      /INFO/,
      /300/,
      /Timestamp/,
      RegExp(`${new Date().getFullYear()}.+`, 'g'),
    ],
  },
  // these all use teh default selection/table renderers
  ...['pgbouncer', 'postgrest', 'storage', 'realtime'].map((queryType) => ({
    queryType,
    tableName: undefined,
    tableLog: logDataFixture({
      event_message: 'some message',
      metadata: undefined,
    }),
    selectionLog: logDataFixture({
      metadata: [{ some: [{ nested: 'value' }] }],
    }),
    tableTexts: [/some message/],
    selectionTexts: [/some/, /nested/, /value/, RegExp(`${new Date().getFullYear()}.+`, 'g')],
  })),
])(
  'selection $queryType $tableName , can display log data and metadata',
  async ({ queryType, tableName, tableLog, selectionLog, tableTexts, selectionTexts }) => {
    get.mockImplementation((url) => {
      // counts
      if (url.includes('count')) {
        return { result: [{ count: 0 }] }
      }
      // single
      if (url.includes('where+id')) {
        return { result: [selectionLog] }
      }
      // table
      return { result: [tableLog] }
    })
    render(<LogsPreviewer projectRef="123" queryType={queryType} tableName={tableName} />)

    await waitFor(() => {
      expect(get).toHaveBeenCalledWith(expect.stringContaining('iso_timestamp_start'))
      expect(get).not.toHaveBeenCalledWith(expect.stringContaining('iso_timestamp_end'))
    })
    // reset mock so that we can check for selection call
    get.mockClear()

    for (const text of tableTexts) {
      await screen.findByText(text)
    }
    const row = await screen.findByText(tableTexts[0])
    fireEvent.click(row)

    await waitFor(() => {
      expect(get).toHaveBeenCalledWith(expect.stringContaining('iso_timestamp_start'))
      expect(get).not.toHaveBeenCalledWith(expect.stringContaining('iso_timestamp_end'))
    })

    for (const text of selectionTexts) {
      await screen.findAllByText(text)
    }
  }
)

test('Search will trigger a log refresh', async () => {
  get.mockImplementation((url) => {
    if (url.includes('something')) {
      return {
        result: [logDataFixture({ id: 'some-event-id' })],
      }
    }
    return { result: [] }
  })
  render(<LogsPreviewer projectRef="123" tableName={LogsTableName.EDGE} />)

  userEvent.type(screen.getByPlaceholderText(/Search events/), 'something{enter}')

  await waitFor(
    () => {
      expect(get).toHaveBeenCalledWith(expect.stringContaining('something'))

      // updates router query params
      const router = useRouter()
      expect(router.push).toHaveBeenCalledWith(
        expect.objectContaining({
          pathname: expect.any(String),
          query: expect.objectContaining({
            s: expect.stringContaining('something'),
          }),
        })
      )
    },
    { timeout: 1500 }
  )
  await screen.findByText(/some-event-id/)
})

test('poll count for new messages', async () => {
  get.mockImplementation((url) => {
    if (url.includes('count')) {
      return { result: [{ count: 125 }] }
    }
    return {
      result: [logDataFixture({ id: 'some-uuid123' })],
    }
  })
  render(<LogsPreviewer projectRef="123" tableName={LogsTableName.EDGE} />)
  await waitFor(() => screen.queryByText(/some-uuid123/) === null)
  // should display new logs count
  await waitFor(() => screen.getByText(/125/))

  userEvent.click(screen.getByText(/Refresh/))
  await waitFor(() => screen.queryByText(/125/) === null)
  await screen.findByText(/some-uuid123/)
})
test('log event chart', async () => {
  get.mockImplementation((url) => {
    // truncate
    if (url.includes('trunc')) {
      return { result: [{ timestamp: new Date().toISOString(), count: 125 }] }
    }
    return {
      result: [logDataFixture({ id: 'some-uuid123' })],
    }
  })
  render(<LogsPreviewer projectRef="123" tableName={LogsTableName.EDGE} />)

  await waitFor(() => screen.queryByText(/some-uuid123/) === null)
  expect(get).toBeCalledWith(expect.stringContaining('trunc'))
})

test('s= query param will populate the search bar', async () => {
  const router = defaultRouterMock()
  router.query = { ...router.query, s: 'someSearch' }
  useRouter.mockReturnValue(router)
  render(<LogsPreviewer projectRef="123" tableName={LogsTableName.EDGE} />)
  // should populate search input with the search param
  await screen.findByDisplayValue('someSearch')
  expect(get).toHaveBeenCalledWith(expect.stringContaining('someSearch'))
})

test('te= query param will populate the timestamp to input', async () => {
  // get time 20 mins before
  const newDate = new Date()
  newDate.setMinutes(new Date().getMinutes() - 20)
  const iso = newDate.toISOString()
  const router = defaultRouterMock()
  router.query = { ...router.query, ite: iso }
  useRouter.mockReturnValue(router)
  render(<LogsPreviewer projectRef="123" tableName={LogsTableName.EDGE} />)

  await waitFor(() => {
    expect(get).toHaveBeenCalledWith(
      expect.stringContaining(`iso_timestamp_end=${encodeURIComponent(iso)}`)
    )
  })
  userEvent.click(await screen.findByTitle('Custom'))
})
test('ts= query param will populate the timestamp from input', async () => {
  // get time 20 mins before
  const newDate = new Date()
  newDate.setMinutes(new Date().getMinutes() - 20)
  const iso = newDate.toISOString()
  const router = defaultRouterMock()
  router.query = { ...router.query, its: iso }
  useRouter.mockReturnValue(router)
  render(<LogsPreviewer projectRef="123" tableName={LogsTableName.EDGE} />)

  await waitFor(() => {
    expect(get).toHaveBeenCalledWith(
      expect.stringContaining(`iso_timestamp_start=${encodeURIComponent(iso)}`)
    )
  })
  userEvent.click(await screen.findByTitle('Custom'))
  await screen.findByText(new RegExp(newDate.getFullYear()))
})

test('load older btn will fetch older logs', async () => {
  get.mockImplementation((url) => {
    if (url.includes('count')) {
      return {}
    }
    return {
      result: [logDataFixture({ id: 'first event' })],
    }
  })
  render(<LogsPreviewer projectRef="123" tableName={LogsTableName.EDGE} />)
  // should display first log but not second
  await waitFor(() => screen.getByText('first event'))
  await expect(screen.findByText('second event')).rejects.toThrow()

  get.mockResolvedValueOnce({
    result: [logDataFixture({ id: 'second event' })],
  })
  // should display first and second log
  userEvent.click(await screen.findByText('Load older'))
  await screen.findByText('first event')
  await screen.findByText('second event')
  expect(get).toHaveBeenCalledWith(expect.stringContaining('timestamp_end='))
})

test('bug: load older btn does not error out when previous page is empty', async () => {
  // bugfix for https://sentry.io/organizations/supabase/issues/2903331460/?project=5459134&referrer=slack
  get.mockImplementation((url) => {
    if (url.includes('count')) {
      return {}
    }
    return { result: [] }
  })
  render(<LogsPreviewer projectRef="123" tableName={LogsTableName.EDGE} />)

  userEvent.click(await screen.findByText('Load older'))
  // NOTE: potential race condition, since we are asserting that something DOES NOT EXIST
  // wait for 500s to make sure all ui logic is complete
  // need to wrap in act because internal react state is changing during this time.
  await act(async () => await wait(100))

  // clicking load older multiple times should not give error
  await waitFor(() => {
    expect(screen.queryByText(/Sorry/)).toBeNull()
    expect(screen.queryByText(/An error occurred/)).toBeNull()
    expect(screen.queryByText(/undefined/)).toBeNull()
  })
})

test('log event chart hide', async () => {
  get.mockImplementation((url) => {
    return { result: [] }
  })
  render(<LogsPreviewer projectRef="123" tableName={LogsTableName.EDGE} />)
  await screen.findByText(/Logs \/ Time/)
  const toggle = await screen.findByText(/Chart/)
  userEvent.click(toggle)
  await expect(screen.findByText('Events')).rejects.toThrow()
})

test('bug: nav backwards with params change results in ui changing', async () => {
  // bugfix for https://sentry.io/organizations/supabase/issues/2903331460/?project=5459134&referrer=slack
  get.mockImplementation((url) => {
    if (url.includes('count')) {
      return {}
    }
    return { data: [] }
  })
  const { container, rerender } = render(
    <LogsPreviewer projectRef="123" tableName={LogsTableName.EDGE} />
  )

  await expect(screen.findByDisplayValue('simple-query')).rejects.toThrow()

  const router = defaultRouterMock()
  router.query = { ...router.query, s: 'simple-query' }
  useRouter.mockReturnValue(router)
  rerender(<LogsPreviewer projectRef="123" tableName={LogsTableName.EDGE} />)

  await screen.findByDisplayValue('simple-query')
})

test('bug: nav to explorer preserves newlines', async () => {
  render(<LogsPreviewer projectRef="123" tableName={LogsTableName.EDGE} />)
  const router = useRouter()
  userEvent.click(await screen.findByText(/Explore/))
  await expect(router.push).toBeCalledWith(expect.stringContaining(encodeURIComponent('\n')))
})
test('filters alter generated query', async () => {
  render(<LogsPreviewer projectRef="123" tableName={LogsTableName.EDGE} />)
  userEvent.click(await screen.findByRole('button', { name: 'Status' }))
  userEvent.click(await screen.findByText(/500 error codes/))
  userEvent.click(await screen.findByText(/200 codes/))
  userEvent.click(await screen.findByText(/Save/))

  await waitFor(() => {
    // counts are adjusted
    expect(get).toHaveBeenCalledWith(expect.stringMatching(/count.+\*.+as.count.+where.+500.+599/))

    expect(get).toHaveBeenCalledWith(expect.stringContaining('500'))
    expect(get).toHaveBeenCalledWith(expect.stringContaining('599'))
    expect(get).toHaveBeenCalledWith(expect.stringContaining('200'))
    expect(get).toHaveBeenCalledWith(expect.stringContaining('299'))
    expect(get).toHaveBeenCalledWith(expect.stringContaining('where'))
    expect(get).toHaveBeenCalledWith(expect.stringContaining('and'))
  })

  // should be able to clear the filters
  userEvent.click(await screen.findByRole('button', { name: 'Status' }))
  userEvent.click(await screen.findByRole('button', { name: 'Clear' }))
  get.mockClear()

  userEvent.click(await screen.findByRole('button', { name: 'Status' }))
  userEvent.click(await screen.findByText(/400 codes/))
  userEvent.click(await screen.findByText(/Save/))

  await waitFor(() => {
    // counts are adjusted
    expect(get).not.toHaveBeenCalledWith(
      expect.stringMatching(/count.+\*.+as.count.+where.+500.+599/)
    )
    expect(get).toHaveBeenCalledWith(expect.stringMatching(/count.+\*.+as.count.+where.+400.+499/))

    expect(get).not.toHaveBeenCalledWith(expect.stringContaining('500'))
    expect(get).not.toHaveBeenCalledWith(expect.stringContaining('599'))
    expect(get).not.toHaveBeenCalledWith(expect.stringContaining('200'))
    expect(get).not.toHaveBeenCalledWith(expect.stringContaining('299'))
    expect(get).toHaveBeenCalledWith(expect.stringContaining('400'))
    expect(get).toHaveBeenCalledWith(expect.stringContaining('499'))
    expect(get).toHaveBeenCalledWith(expect.stringContaining('where'))
    expect(get).toHaveBeenCalledWith(expect.stringContaining('and'))
  })
})
test('filters accept filterOverride', async () => {
  render(
    <LogsPreviewer
      projectRef="123"
      tableName={LogsTableName.FUNCTIONS}
      filterOverride={{ 'my.nestedkey': 'myvalue' }}
    />
  )

  await waitFor(() => {
    expect(get).toHaveBeenCalledWith(expect.stringContaining('my.nestedkey'))
    expect(get).toHaveBeenCalledWith(expect.stringContaining('myvalue'))
  })
})

describe.each(['FREE', 'PRO', 'ENTERPRISE'])('upgrade modal for %s', (key) => {
  beforeEach(() => {
    useProjectSubscription.mockReturnValue({
      subscription: {
        tier: {
          supabase_prod_id: `tier_${key.toLocaleLowerCase()}`,
          key,
        },
      },
    })
  })
  test('based on query params', async () => {
    const router = defaultRouterMock()
    router.query = {
      ...router.query,
      q: 'some_query',
      its: dayjs().subtract(4, 'months').toISOString(),
      ite: dayjs().toISOString(),
    }
    useRouter.mockReturnValue(router)
    render(<LogsPreviewer projectRef="123" tableName={LogsTableName.EDGE} />)
    await screen.findByText('Log retention') // assert modal title is present
  })
})