import ReactEchartsCore from 'echarts-for-react/lib/core';

import {initializeOrg} from 'sentry-test/initializeOrg';
import {act, render, screen, userEvent, waitFor} from 'sentry-test/reactTestingLibrary';

import {ModalRenderProps} from 'sentry/actionCreators/modal';
import WidgetViewerModal from 'sentry/components/modals/widgetViewerModal';
import MemberListStore from 'sentry/stores/memberListStore';
import PageFiltersStore from 'sentry/stores/pageFiltersStore';
import space from 'sentry/styles/space';
import {Series} from 'sentry/types/echarts';
import {TableDataWithTitle} from 'sentry/utils/discover/discoverQuery';
import {
  DisplayType,
  Widget,
  WidgetQuery,
  WidgetType,
} from 'sentry/views/dashboardsV2/types';

jest.mock('echarts-for-react/lib/core', () => {
  return jest.fn(({style}) => {
    return <div style={{...style, background: 'green'}}>echarts mock</div>;
  });
});

jest.mock('sentry/components/tooltip', () => {
  return jest.fn(props => {
    return <div>{props.children}</div>;
  });
});

const stubEl = (props: {children?: React.ReactNode}) => <div>{props.children}</div>;

let eventsMetaMock;

const waitForMetaToHaveBeenCalled = async () => {
  await waitFor(() => {
    expect(eventsMetaMock).toHaveBeenCalled();
  });
};

async function renderModal({
  initialData: {organization, routerContext},
  widget,
  seriesData,
  tableData,
  pageLinks,
}: {
  initialData: any;
  widget: any;
  pageLinks?: string;
  seriesData?: Series[];
  tableData?: TableDataWithTitle[];
}) {
  const rendered = render(
    <div style={{padding: space(4)}}>
      <WidgetViewerModal
        Header={stubEl}
        Footer={stubEl as ModalRenderProps['Footer']}
        Body={stubEl as ModalRenderProps['Body']}
        CloseButton={stubEl}
        closeModal={() => undefined}
        organization={organization}
        widget={widget}
        onEdit={() => undefined}
        seriesData={seriesData}
        tableData={tableData}
        pageLinks={pageLinks}
      />
    </div>,
    {
      context: routerContext,
      organization,
    }
  );
  // Need to wait since WidgetViewerModal will make a request to events-meta
  // for total events count on mount
  if (widget.widgetType === WidgetType.DISCOVER) {
    await waitForMetaToHaveBeenCalled();
  }
  return rendered;
}

describe('Modals -> WidgetViewerModal', function () {
  let initialData, initialDataWithFlag;
  beforeEach(() => {
    initialData = initializeOrg({
      organization: {
        features: ['discover-query', 'widget-viewer-modal'],
        apdexThreshold: 400,
      },
      router: {
        location: {query: {}},
      },
      project: 1,
      projects: [],
    });

    initialDataWithFlag = {
      ...initialData,
      organization: {
        ...initialData.organization,
        features: [
          ...initialData.organization.features,
          'discover-frontend-use-events-endpoint',
        ],
      },
    };

    MockApiClient.addMockResponse({
      url: '/organizations/org-slug/projects/',
      body: [],
    });

    MockApiClient.addMockResponse({
      url: '/organizations/org-slug/releases/',
      body: [],
    });

    eventsMetaMock = MockApiClient.addMockResponse({
      url: '/organizations/org-slug/events-meta/',
      body: {count: 33323612},
    });

    PageFiltersStore.init();
    PageFiltersStore.onInitializeUrlState(
      {
        projects: [1, 2],
        environments: ['prod', 'dev'],
        datetime: {start: null, end: null, period: '24h', utc: null},
      },
      new Set()
    );
  });

  afterEach(() => {
    MockApiClient.clearMockResponses();
    PageFiltersStore.teardown();
  });

  describe('Discover Widgets', function () {
    describe('Area Chart Widget', function () {
      let mockQuery: WidgetQuery;
      let additionalMockQuery: WidgetQuery;
      let mockWidget: Widget;

      function mockEventsv2() {
        return MockApiClient.addMockResponse({
          url: '/organizations/org-slug/eventsv2/',
          body: {
            data: [
              {
                title: '/organizations/:orgId/dashboards/',
                id: '1',
                count: 1,
              },
            ],
            meta: {
              title: 'string',
              id: 'string',
              count: 1,
              isMetricsData: false,
            },
          },
        });
      }

      function mockEvents() {
        return MockApiClient.addMockResponse({
          url: '/organizations/org-slug/events/',
          body: {
            data: [
              {
                title: '/organizations/:orgId/dashboards/',
                id: '1',
                count: 1,
              },
            ],
            meta: {
              fields: {
                title: 'string',
                id: 'string',
                count: 1,
              },
              isMetricsData: false,
            },
          },
        });
      }

      beforeEach(function () {
        mockQuery = {
          conditions: 'title:/organizations/:orgId/performance/summary/',
          fields: ['count()'],
          aggregates: ['count()'],
          columns: [],
          name: 'Query Name',
          orderby: '',
        };
        additionalMockQuery = {
          conditions: '',
          fields: ['count()'],
          aggregates: ['count()'],
          columns: [],
          name: 'Another Query Name',
          orderby: '',
        };
        mockWidget = {
          id: '1',
          title: 'Test Widget',
          displayType: DisplayType.AREA,
          interval: '5m',
          queries: [mockQuery, additionalMockQuery],
          widgetType: WidgetType.DISCOVER,
        };
        (ReactEchartsCore as jest.Mock).mockClear();
        MockApiClient.addMockResponse({
          url: '/organizations/org-slug/events-stats/',
          body: {
            data: [
              [[1646100000], [{count: 1}]],
              [[1646120000], [{count: 1}]],
            ],
            start: 1646100000,
            end: 1646120000,
            isMetricsData: false,
          },
        });
      });

      describe('with eventsv2', function () {
        it('renders Edit and Open buttons', async function () {
          mockEventsv2();
          await renderModal({initialData, widget: mockWidget});
          expect(screen.getByText('Edit Widget')).toBeInTheDocument();
          expect(screen.getByText('Open in Discover')).toBeInTheDocument();
        });

        it('renders updated table columns and orderby', async function () {
          const eventsv2Mock = mockEventsv2();
          await renderModal({initialData, widget: mockWidget});
          expect(screen.getByText('title')).toBeInTheDocument();
          expect(
            screen.getByText('/organizations/:orgId/dashboards/')
          ).toBeInTheDocument();
          expect(eventsv2Mock).toHaveBeenCalledWith(
            '/organizations/org-slug/eventsv2/',
            expect.objectContaining({
              query: expect.objectContaining({sort: ['-count()']}),
            })
          );
        });

        it('renders area chart', async function () {
          mockEventsv2();
          await renderModal({initialData, widget: mockWidget});
          expect(screen.getByText('echarts mock')).toBeInTheDocument();
        });

        it('renders Discover area chart widget viewer', async function () {
          mockEventsv2();
          const {container} = await renderModal({initialData, widget: mockWidget});
          expect(container).toSnapshot();
        });

        it('redirects user to Discover when clicking Open in Discover', async function () {
          mockEventsv2();
          await renderModal({initialData, widget: mockWidget});
          userEvent.click(screen.getByText('Open in Discover'));
          expect(initialData.router.push).toHaveBeenCalledWith(
            '/organizations/org-slug/discover/results/?environment=prod&environment=dev&field=count%28%29&name=Test%20Widget&project=1&project=2&query=title%3A%2Forganizations%2F%3AorgId%2Fperformance%2Fsummary%2F&statsPeriod=24h&yAxis=count%28%29'
          );
        });

        it('zooms into the selected time range', async function () {
          mockEventsv2();
          await renderModal({initialData, widget: mockWidget});
          act(() => {
            // Simulate dataZoom event on chart
            (ReactEchartsCore as jest.Mock).mock.calls[0][0].onEvents.datazoom(
              undefined,
              {
                getModel: () => {
                  return {
                    _payload: {
                      batch: [{startValue: 1646100000000, endValue: 1646120000000}],
                    },
                  };
                },
              }
            );
          });
          expect(initialData.router.push).toHaveBeenCalledWith(
            expect.objectContaining({
              query: {
                viewerEnd: '2022-03-01T07:33:20',
                viewerStart: '2022-03-01T02:00:00',
              },
            })
          );
        });

        it('renders multiquery label and selector', async function () {
          mockEventsv2();
          await renderModal({initialData, widget: mockWidget});
          expect(
            screen.getByText(
              'This widget was built with multiple queries. Table data can only be displayed for one query at a time. To edit any of the queries, edit the widget.'
            )
          ).toBeInTheDocument();
          expect(screen.getByText('Query Name')).toBeInTheDocument();
        });

        it('updates selected query when selected in the query dropdown', async function () {
          mockEventsv2();
          const {rerender} = await renderModal({initialData, widget: mockWidget});
          userEvent.click(screen.getByText('Query Name'));
          userEvent.click(screen.getByText('Another Query Name'));
          expect(initialData.router.replace).toHaveBeenCalledWith({
            query: {query: 1},
          });
          // Need to manually set the new router location and rerender to simulate the dropdown selection click
          initialData.router.location.query = {query: ['1']};
          rerender(
            <WidgetViewerModal
              Header={stubEl}
              Footer={stubEl as ModalRenderProps['Footer']}
              Body={stubEl as ModalRenderProps['Body']}
              CloseButton={stubEl}
              closeModal={() => undefined}
              organization={initialData.organization}
              widget={mockWidget}
              onEdit={() => undefined}
            />
          );
          await waitForMetaToHaveBeenCalled();
          expect(screen.getByText('Another Query Name')).toBeInTheDocument();
        });

        it('renders the correct discover query link when there are multiple queries in a widget', async function () {
          mockEventsv2();
          initialData.router.location.query = {query: ['1']};
          await renderModal({initialData, widget: mockWidget});
          expect(screen.getByRole('button', {name: 'Open in Discover'})).toHaveAttribute(
            'href',
            '/organizations/org-slug/discover/results/?environment=prod&environment=dev&field=count%28%29&name=Test%20Widget&project=1&project=2&query=&statsPeriod=24h&yAxis=count%28%29'
          );
        });

        it('renders with first legend disabled by default', async function () {
          mockEventsv2();
          // Rerender with first legend disabled
          initialData.router.location.query = {legend: ['Query Name']};
          await renderModal({initialData, widget: mockWidget});
          expect(ReactEchartsCore).toHaveBeenLastCalledWith(
            expect.objectContaining({
              option: expect.objectContaining({
                legend: expect.objectContaining({
                  selected: {'Query Name': false},
                }),
              }),
            }),
            {}
          );
        });

        it('renders total results in footer', async function () {
          mockEventsv2();
          await renderModal({initialData, widget: mockWidget});
          expect(screen.getByText('33,323,612')).toBeInTheDocument();
        });

        it('renders highlighted query text and multiple queries in select dropdown', async function () {
          mockEventsv2();
          const {container} = await renderModal({
            initialData,
            widget: {
              ...mockWidget,
              queries: [{...mockQuery, name: ''}, additionalMockQuery],
            },
          });
          userEvent.click(screen.getByText('/organizations/:orgId/performance/summary/'));
          expect(container).toSnapshot();
        });

        it('renders widget chart minimap', async function () {
          initialData.organization.features.push('widget-viewer-modal-minimap');
          mockEventsv2();
          await renderModal({
            initialData,
            widget: {
              ...mockWidget,
              queries: [{...mockQuery, name: ''}, additionalMockQuery],
            },
          });

          expect(ReactEchartsCore).toHaveBeenLastCalledWith(
            expect.objectContaining({
              option: expect.objectContaining({
                dataZoom: expect.arrayContaining([
                  expect.objectContaining({
                    realtime: false,
                    showDetail: false,
                    end: 100,
                    start: 0,
                  }),
                ]),
              }),
            }),
            {}
          );
        });

        it('zooming on minimap updates location query and updates echart start and end values', async function () {
          initialData.organization.features.push('widget-viewer-modal-minimap');
          mockEventsv2();
          await renderModal({
            initialData,
            widget: {
              ...mockWidget,
              queries: [{...mockQuery, name: ''}, additionalMockQuery],
            },
          });
          const calls = (ReactEchartsCore as jest.Mock).mock.calls;
          act(() => {
            // Simulate dataZoom event on chart
            calls[calls.length - 1][0].onEvents.datazoom(
              {seriesStart: 1646100000000, seriesEnd: 1646120000000},
              {
                getModel: () => {
                  return {
                    _payload: {start: 30, end: 70},
                  };
                },
              }
            );
          });

          expect(initialData.router.push).toHaveBeenCalledWith(
            expect.objectContaining({
              query: {
                viewerEnd: '2022-03-01T05:53:20',
                viewerStart: '2022-03-01T03:40:00',
              },
            })
          );
        });

        it('includes group by in widget viewer table', async function () {
          mockEventsv2();
          mockWidget.queries = [
            {
              conditions: 'title:/organizations/:orgId/performance/summary/',
              fields: ['count()'],
              aggregates: ['count()'],
              columns: ['transaction'],
              name: 'Query Name',
              orderby: '-count()',
            },
          ];
          await renderModal({initialData, widget: mockWidget});
          screen.getByText('transaction');
        });

        it('includes order by in widget viewer table if not explicitly selected', async function () {
          mockEventsv2();
          mockWidget.queries = [
            {
              conditions: 'title:/organizations/:orgId/performance/summary/',
              fields: ['count()'],
              aggregates: ['count()'],
              columns: ['transaction'],
              name: 'Query Name',
              orderby: 'count_unique(user)',
            },
          ];
          await renderModal({initialData, widget: mockWidget});
          screen.getByText('count_unique(user)');
        });

        it('includes a custom equation order by in widget viewer table if not explicitly selected', async function () {
          mockEventsv2();
          mockWidget.queries = [
            {
              conditions: 'title:/organizations/:orgId/performance/summary/',
              fields: ['count()'],
              aggregates: ['count()'],
              columns: ['transaction'],
              name: 'Query Name',
              orderby: '-equation|count_unique(user) + 1',
            },
          ];
          await renderModal({initialData, widget: mockWidget});
          screen.getByText('count_unique(user) + 1');
        });
      });
      describe('with events', function () {
        it('renders updated table columns and orderby', async function () {
          const eventsMock = mockEvents();
          await renderModal({initialData: initialDataWithFlag, widget: mockWidget});
          expect(screen.getByText('title')).toBeInTheDocument();
          expect(
            screen.getByText('/organizations/:orgId/dashboards/')
          ).toBeInTheDocument();
          expect(eventsMock).toHaveBeenCalledWith(
            '/organizations/org-slug/events/',
            expect.objectContaining({
              query: expect.objectContaining({sort: ['-count()']}),
            })
          );
        });
      });
    });

    describe('TopN Chart Widget', function () {
      let mockQuery, mockWidget;

      function mockEventsStats() {
        return MockApiClient.addMockResponse({
          url: '/organizations/org-slug/events-stats/',
          body: {
            data: [
              [[1646100000], [{count: 1}]],
              [[1646120000], [{count: 1}]],
            ],
            start: 1646100000,
            end: 1646120000,
            isMetricsData: false,
          },
        });
      }

      const eventsMockData = [
        {
          'error.type': ['Test Error 1a', 'Test Error 1b', 'Test Error 1c'],
          count: 10,
        },
        {
          'error.type': ['Test Error 2'],
          count: 6,
        },
        {
          'error.type': ['Test Error 3'],
          count: 5,
        },
        {
          'error.type': ['Test Error 4'],
          count: 4,
        },
        {
          'error.type': ['Test Error 5'],
          count: 3,
        },
        {
          'error.type': ['Test Error 6'],
          count: 2,
        },
      ];

      function mockEventsv2() {
        return MockApiClient.addMockResponse({
          url: '/organizations/org-slug/eventsv2/',
          match: [MockApiClient.matchQuery({cursor: undefined})],
          headers: {
            Link:
              '<http://localhost/api/0/organizations/org-slug/eventsv2/?cursor=0:0:1>; rel="previous"; results="false"; cursor="0:0:1",' +
              '<http://localhost/api/0/organizations/org-slug/eventsv2/?cursor=0:10:0>; rel="next"; results="true"; cursor="0:10:0"',
          },
          body: {
            data: eventsMockData,
            meta: {
              'error.type': 'array',
              count: 'integer',
            },
          },
        });
      }

      function mockEvents() {
        return MockApiClient.addMockResponse({
          url: '/organizations/org-slug/events/',
          match: [MockApiClient.matchQuery({cursor: undefined})],
          headers: {
            Link:
              '<http://localhost/api/0/organizations/org-slug/eventsv2/?cursor=0:0:1>; rel="previous"; results="false"; cursor="0:0:1",' +
              '<http://localhost/api/0/organizations/org-slug/eventsv2/?cursor=0:10:0>; rel="next"; results="true"; cursor="0:10:0"',
          },
          body: {
            data: eventsMockData,
            meta: {
              fields: {
                'error.type': 'array',
                count: 'integer',
              },
            },
          },
        });
      }

      beforeEach(function () {
        mockQuery = {
          conditions: 'title:/organizations/:orgId/performance/summary/',
          fields: ['error.type', 'count()'],
          aggregates: ['count()'],
          columns: ['error.type'],
          id: '1',
          name: 'Query Name',
          orderby: '',
        };
        mockWidget = {
          title: 'Test Widget',
          displayType: DisplayType.TOP_N,
          interval: '5m',
          queries: [mockQuery],
          widgetType: WidgetType.DISCOVER,
        };

        MockApiClient.addMockResponse({
          url: '/organizations/org-slug/eventsv2/',
          match: [MockApiClient.matchQuery({cursor: '0:10:0'})],
          headers: {
            Link:
              '<http://localhost/api/0/organizations/org-slug/eventsv2/?cursor=0:0:1>; rel="previous"; results="false"; cursor="0:0:1",' +
              '<http://localhost/api/0/organizations/org-slug/eventsv2/?cursor=0:20:0>; rel="next"; results="true"; cursor="0:20:0"',
          },
          body: {
            data: [
              {
                'error.type': ['Next Page Test Error'],
                count: 1,
              },
            ],
            meta: {
              'error.type': 'array',
              count: 'integer',
            },
          },
        });
      });

      describe('with eventsv2', function () {
        it('renders Discover topn chart widget viewer', async function () {
          mockEventsStats();
          mockEventsv2();
          const {container} = await renderModal({initialData, widget: mockWidget});
          expect(container).toSnapshot();
        });

        it('sorts table when a sortable column header is clicked', async function () {
          const eventsStatsMock = mockEventsStats();
          const eventsv2Mock = mockEventsv2();
          const {rerender} = await renderModal({initialData, widget: mockWidget});
          userEvent.click(screen.getByText('count()'));
          expect(initialData.router.push).toHaveBeenCalledWith({
            query: {sort: ['-count()']},
          });
          // Need to manually set the new router location and rerender to simulate the sortable column click
          initialData.router.location.query = {sort: ['-count()']};
          rerender(
            <WidgetViewerModal
              Header={stubEl}
              Footer={stubEl as ModalRenderProps['Footer']}
              Body={stubEl as ModalRenderProps['Body']}
              CloseButton={stubEl}
              closeModal={() => undefined}
              organization={initialData.organization}
              widget={mockWidget}
              onEdit={() => undefined}
            />
          );
          await waitForMetaToHaveBeenCalled();
          expect(eventsv2Mock).toHaveBeenCalledWith(
            '/organizations/org-slug/eventsv2/',
            expect.objectContaining({
              query: expect.objectContaining({sort: ['-count()']}),
            })
          );
          expect(eventsStatsMock).toHaveBeenCalledWith(
            '/organizations/org-slug/events-stats/',
            expect.objectContaining({
              query: expect.objectContaining({orderby: '-count()'}),
            })
          );
        });

        it('renders pagination buttons', async function () {
          mockEventsStats();
          mockEventsv2();
          await renderModal({initialData, widget: mockWidget});
          expect(screen.getByRole('button', {name: 'Previous'})).toBeInTheDocument();
          expect(screen.getByRole('button', {name: 'Next'})).toBeInTheDocument();
        });

        it('does not render pagination buttons', async function () {
          mockEventsStats();
          mockEventsv2();
          MockApiClient.addMockResponse({
            url: '/organizations/org-slug/eventsv2/',
            headers: {
              Link:
                '<http://localhost/api/0/organizations/org-slug/eventsv2/?cursor=0:0:1>; rel="previous"; results="false"; cursor="0:0:1",' +
                '<http://localhost/api/0/organizations/org-slug/eventsv2/?cursor=0:20:0>; rel="next"; results="false"; cursor="0:20:0"',
            },
            body: {
              data: [
                {
                  'error.type': ['No Pagination'],
                  count: 1,
                },
              ],
              meta: {
                'error.type': 'array',
                count: 'integer',
              },
            },
          });
          await renderModal({initialData, widget: mockWidget});
          expect(
            screen.queryByRole('button', {name: 'Previous'})
          ).not.toBeInTheDocument();
          expect(screen.queryByRole('button', {name: 'Next'})).not.toBeInTheDocument();
        });

        it('paginates to the next page', async function () {
          mockEventsStats();
          mockEventsv2();
          const {rerender} = await renderModal({initialData, widget: mockWidget});
          expect(screen.getByText('Test Error 1c')).toBeInTheDocument();
          userEvent.click(screen.getByRole('button', {name: 'Next'}));
          expect(initialData.router.replace).toHaveBeenCalledWith(
            expect.objectContaining({
              query: {cursor: '0:10:0'},
            })
          );
          // Need to manually set the new router location and rerender to simulate the next page click
          initialData.router.location.query = {cursor: ['0:10:0']};
          rerender(
            <WidgetViewerModal
              Header={stubEl}
              Footer={stubEl as ModalRenderProps['Footer']}
              Body={stubEl as ModalRenderProps['Body']}
              CloseButton={stubEl}
              closeModal={() => undefined}
              organization={initialData.organization}
              widget={mockWidget}
              onEdit={() => undefined}
            />
          );
          await waitForMetaToHaveBeenCalled();
          expect(await screen.findByText('Next Page Test Error')).toBeInTheDocument();
        });

        it('uses provided seriesData and does not make an events-stats requests', async function () {
          const eventsStatsMock = mockEventsStats();
          mockEventsv2();
          await renderModal({initialData, widget: mockWidget, seriesData: []});
          expect(eventsStatsMock).not.toHaveBeenCalled();
        });

        it('makes events-stats requests when table is sorted', async function () {
          const eventsStatsMock = mockEventsStats();
          mockEventsv2();
          await renderModal({
            initialData,
            widget: mockWidget,
            seriesData: [],
          });
          expect(eventsStatsMock).not.toHaveBeenCalled();
          userEvent.click(screen.getByText('count()'));
          await waitForMetaToHaveBeenCalled();
          expect(eventsStatsMock).toHaveBeenCalledTimes(1);
        });

        it('renders widget chart minimap', async function () {
          mockEventsStats();
          mockEventsv2();
          initialData.organization.features.push('widget-viewer-modal-minimap');
          await renderModal({initialData, widget: mockWidget});

          expect(ReactEchartsCore).toHaveBeenLastCalledWith(
            expect.objectContaining({
              option: expect.objectContaining({
                dataZoom: expect.arrayContaining([
                  expect.objectContaining({
                    realtime: false,
                    showDetail: false,
                    end: 100,
                    start: 0,
                  }),
                ]),
              }),
            }),
            {}
          );
        });

        it('zooming on minimap updates location query and updates echart start and end values', async function () {
          mockEventsStats();
          mockEventsv2();
          initialData.organization.features.push('widget-viewer-modal-minimap');
          await renderModal({initialData, widget: mockWidget});
          const calls = (ReactEchartsCore as jest.Mock).mock.calls;
          act(() => {
            // Simulate dataZoom event on chart
            calls[calls.length - 1][0].onEvents.datazoom(
              {seriesStart: 1646100000000, seriesEnd: 1646120000000},
              {
                getModel: () => {
                  return {
                    _payload: {start: 30, end: 70},
                  };
                },
              }
            );
          });

          expect(initialData.router.push).toHaveBeenCalledWith(
            expect.objectContaining({
              query: {
                viewerEnd: '2022-03-01T05:53:20',
                viewerStart: '2022-03-01T03:40:00',
              },
            })
          );

          await waitFor(() => {
            expect(ReactEchartsCore).toHaveBeenLastCalledWith(
              expect.objectContaining({
                option: expect.objectContaining({
                  dataZoom: expect.arrayContaining([
                    expect.objectContaining({
                      realtime: false,
                      showDetail: false,
                      endValue: 1646114000000,
                      startValue: 1646106000000,
                    }),
                  ]),
                }),
              }),
              {}
            );
          });
        });
      });

      describe('with events', function () {
        it('sorts table when a sortable column header is clicked', async function () {
          const eventsStatsMock = mockEventsStats();
          const eventsMock = mockEvents();
          const {rerender} = await renderModal({
            initialData: initialDataWithFlag,
            widget: mockWidget,
          });
          userEvent.click(screen.getByText('count()'));
          expect(initialDataWithFlag.router.push).toHaveBeenCalledWith({
            query: {sort: ['-count()']},
          });
          // Need to manually set the new router location and rerender to simulate the sortable column click
          initialDataWithFlag.router.location.query = {sort: ['-count()']};
          rerender(
            <WidgetViewerModal
              Header={stubEl}
              Footer={stubEl as ModalRenderProps['Footer']}
              Body={stubEl as ModalRenderProps['Body']}
              CloseButton={stubEl}
              closeModal={() => undefined}
              organization={initialDataWithFlag.organization}
              widget={mockWidget}
              onEdit={() => undefined}
            />
          );
          await waitForMetaToHaveBeenCalled();
          expect(eventsMock).toHaveBeenCalledWith(
            '/organizations/org-slug/events/',
            expect.objectContaining({
              query: expect.objectContaining({sort: ['-count()']}),
            })
          );
          expect(eventsStatsMock).toHaveBeenCalledWith(
            '/organizations/org-slug/events-stats/',
            expect.objectContaining({
              query: expect.objectContaining({orderby: '-count()'}),
            })
          );
        });
      });
    });

    describe('World Map Chart Widget', function () {
      let mockQuery, mockWidget;

      const eventsMockData = [
        {
          'geo.country_code': 'ES',
          p75_measurements_lcp: 2000,
        },
        {
          'geo.country_code': 'SK',
          p75_measurements_lcp: 3000,
        },
        {
          'geo.country_code': 'CO',
          p75_measurements_lcp: 4000,
        },
      ];

      function mockEventsGeo() {
        return MockApiClient.addMockResponse({
          url: '/organizations/org-slug/events-geo/',
          body: {
            data: eventsMockData,
            meta: {
              'geo.country_code': 'string',
              p75_measurements_lcp: 'duration',
            },
          },
        });
      }
      function mockEventsv2() {
        return MockApiClient.addMockResponse({
          url: '/organizations/org-slug/eventsv2/',
          body: {
            data: eventsMockData,
            meta: {
              'geo.country_code': 'string',
              p75_measurements_lcp: 'duration',
            },
          },
        });
      }
      function mockEvents() {
        return MockApiClient.addMockResponse({
          url: '/organizations/org-slug/events/',
          body: {
            data: eventsMockData,
            meta: {
              fields: {
                'geo.country_code': 'string',
                p75_measurements_lcp: 'duration',
              },
            },
          },
        });
      }

      beforeEach(function () {
        mockQuery = {
          conditions: 'title:/organizations/:orgId/performance/summary/',
          fields: ['p75(measurements.lcp)'],
          aggregates: ['p75(measurements.lcp)'],
          columns: [],
          id: '1',
          name: 'Query Name',
          orderby: '',
        };
        mockWidget = {
          title: 'Test Widget',
          displayType: DisplayType.WORLD_MAP,
          interval: '5m',
          queries: [mockQuery],
          widgetType: WidgetType.DISCOVER,
        };
      });

      describe('with eventsv2', function () {
        it('always queries geo.country_code in the table chart', async function () {
          const eventsv2Mock = mockEventsv2();
          mockEventsGeo();
          await renderModal({initialData, widget: mockWidget});
          expect(eventsv2Mock).toHaveBeenCalledWith(
            '/organizations/org-slug/eventsv2/',
            expect.objectContaining({
              query: expect.objectContaining({
                field: ['geo.country_code', 'p75(measurements.lcp)'],
              }),
            })
          );
          expect(await screen.findByText('geo.country_code')).toBeInTheDocument();
        });

        it('renders Discover topn chart widget viewer', async function () {
          mockEventsv2();
          mockEventsGeo();
          const {container} = await renderModal({initialData, widget: mockWidget});
          expect(container).toSnapshot();
        });

        it('uses provided tableData and does not make an eventsv2 requests', async function () {
          const eventsGeoMock = mockEventsGeo();
          mockEventsv2();
          await renderModal({initialData, widget: mockWidget, tableData: []});
          expect(eventsGeoMock).not.toHaveBeenCalled();
        });
      });

      describe('with events', function () {
        it('always queries geo.country_code in the table chart', async function () {
          const eventsMock = mockEvents();
          mockEventsGeo();
          await renderModal({initialData: initialDataWithFlag, widget: mockWidget});
          expect(eventsMock).toHaveBeenCalledWith(
            '/organizations/org-slug/events/',
            expect.objectContaining({
              query: expect.objectContaining({
                field: ['geo.country_code', 'p75(measurements.lcp)'],
              }),
            })
          );
          expect(await screen.findByText('geo.country_code')).toBeInTheDocument();
        });
      });
    });

    describe('Table Widget', function () {
      const mockQuery = {
        conditions: 'title:/organizations/:orgId/performance/summary/',
        fields: ['title', 'count()'],
        aggregates: ['count()'],
        columns: ['title'],
        id: '1',
        name: 'Query Name',
        orderby: '',
      };
      const mockWidget = {
        title: 'Test Widget',
        displayType: DisplayType.TABLE,
        interval: '5m',
        queries: [mockQuery],
        widgetType: WidgetType.DISCOVER,
      };
      function mockEventsv2() {
        return MockApiClient.addMockResponse({
          url: '/organizations/org-slug/eventsv2/',
          body: {
            data: [
              {
                title: '/organizations/:orgId/dashboards/',
                id: '1',
                count: 1,
              },
            ],
            meta: {
              title: 'string',
              id: 'string',
              count: 1,
              isMetricsData: false,
            },
          },
        });
      }
      function mockEvents() {
        return MockApiClient.addMockResponse({
          url: '/organizations/org-slug/events/',
          body: {
            data: [
              {
                title: '/organizations/:orgId/dashboards/',
                id: '1',
                count: 1,
              },
            ],
            meta: {
              fields: {
                title: 'string',
                id: 'string',
                count: 1,
              },
              isMetricsData: false,
            },
          },
        });
      }
      describe('with eventsv2', function () {
        it('makes eventsv2 requests when table is paginated', async function () {
          const eventsv2Mock = mockEventsv2();
          await renderModal({
            initialData,
            widget: mockWidget,
            tableData: [],
            pageLinks:
              '<https://sentry.io>; rel="previous"; results="false"; cursor="0:0:1", <https://sentry.io>; rel="next"; results="true"; cursor="0:20:0"',
          });
          expect(eventsv2Mock).not.toHaveBeenCalled();
          userEvent.click(screen.getByLabelText('Next'));
          await waitFor(() => {
            expect(eventsv2Mock).toHaveBeenCalled();
          });
        });
      });

      describe('with events', function () {
        it('makes events requests when table is paginated', async function () {
          const eventsMock = mockEvents();
          await renderModal({
            initialData: initialDataWithFlag,
            widget: mockWidget,
            tableData: [],
            pageLinks:
              '<https://sentry.io>; rel="previous"; results="false"; cursor="0:0:1", <https://sentry.io>; rel="next"; results="true"; cursor="0:20:0"',
          });
          expect(eventsMock).not.toHaveBeenCalled();
          userEvent.click(screen.getByLabelText('Next'));
          await waitFor(() => {
            expect(eventsMock).toHaveBeenCalled();
          });
        });

        it('disables the Open in Discover button for a custom measurement widget', async function () {
          const customMeasurementWidget = {
            ...mockWidget,
            queries: [
              {
                conditions: '',
                fields: [],
                aggregates: ['p99(measurements.custom.measurement)'],
                columns: ['title'],
                id: '1',
                name: 'Query Name',
                orderby: '',
              },
            ],
          };
          await renderModal({
            initialData: initialDataWithFlag,
            widget: customMeasurementWidget,
            tableData: [],
            pageLinks:
              '<https://sentry.io>; rel="previous"; results="false"; cursor="0:0:1", <https://sentry.io>; rel="next"; results="true"; cursor="0:20:0"',
          });
          userEvent.click(screen.getByText('Open in Discover'));
          expect(initialData.router.push).not.toHaveBeenCalled();
        });
      });
    });
  });

  describe('Issue Table Widget', function () {
    let issuesMock;
    const mockQuery = {
      conditions: 'is:unresolved',
      fields: ['events', 'status', 'title'],
      columns: ['events', 'status', 'title'],
      aggregates: [],
      id: '1',
      name: 'Query Name',
      orderby: '',
    };
    const mockWidget = {
      id: '1',
      title: 'Issue Widget',
      displayType: DisplayType.TABLE,
      interval: '5m',
      queries: [mockQuery],
      widgetType: WidgetType.ISSUE,
    };
    beforeEach(function () {
      MemberListStore.loadInitialData([]);
      MockApiClient.addMockResponse({
        url: '/organizations/org-slug/issues/',
        method: 'GET',
        match: [
          MockApiClient.matchData({
            cursor: '0:10:0',
          }),
        ],
        headers: {
          Link:
            '<http://localhost/api/0/organizations/org-slug/issues/?cursor=0:0:1>; rel="previous"; results="false"; cursor="0:0:1",' +
            '<http://localhost/api/0/organizations/org-slug/issues/?cursor=0:20:0>; rel="next"; results="true"; cursor="0:20:0"',
        },
        body: [
          {
            id: '2',
            title: 'Another Error: Failed',
            project: {
              id: '3',
            },
            status: 'unresolved',
            lifetime: {count: 5},
            count: 3,
            userCount: 1,
          },
        ],
      });
      issuesMock = MockApiClient.addMockResponse({
        url: '/organizations/org-slug/issues/',
        method: 'GET',
        match: [
          MockApiClient.matchData({
            cursor: undefined,
          }),
        ],
        headers: {
          Link:
            '<http://localhost/api/0/organizations/org-slug/issues/?cursor=0:0:1>; rel="previous"; results="false"; cursor="0:0:1",' +
            '<http://localhost/api/0/organizations/org-slug/issues/?cursor=0:10:0>; rel="next"; results="true"; cursor="0:10:0"',
        },
        body: [
          {
            id: '1',
            title: 'Error: Failed',
            project: {
              id: '3',
            },
            status: 'unresolved',
            lifetime: {count: 10},
            count: 6,
            userCount: 3,
          },
        ],
      });
    });

    it('renders widget title', async function () {
      await renderModal({initialData, widget: mockWidget});
      expect(screen.getByText('Issue Widget')).toBeInTheDocument();
    });

    it('renders Edit and Open buttons', async function () {
      await renderModal({initialData, widget: mockWidget});
      expect(screen.getByText('Edit Widget')).toBeInTheDocument();
      expect(screen.getByText('Open in Issues')).toBeInTheDocument();
    });

    it('renders events, status, and title table columns', async function () {
      await renderModal({initialData, widget: mockWidget});
      expect(screen.getByText('title')).toBeInTheDocument();
      expect(await screen.findByText('Error: Failed')).toBeInTheDocument();
      expect(screen.getByText('events')).toBeInTheDocument();
      expect(screen.getByText('6')).toBeInTheDocument();
      expect(screen.getByText('status')).toBeInTheDocument();
      expect(screen.getByText('unresolved')).toBeInTheDocument();
    });

    it('renders Issue table widget viewer', async function () {
      const {container} = await renderModal({initialData, widget: mockWidget});
      await screen.findByText('Error: Failed');
      expect(container).toSnapshot();
    });

    it('redirects user to Issues when clicking Open in Issues', async function () {
      await renderModal({initialData, widget: mockWidget});
      userEvent.click(screen.getByText('Open in Issues'));
      expect(initialData.router.push).toHaveBeenCalledWith(
        '/organizations/org-slug/issues/?environment=prod&environment=dev&project=1&project=2&query=is%3Aunresolved&sort=&statsPeriod=24h'
      );
    });

    it('sorts table when a sortable column header is clicked', async function () {
      const {rerender} = await renderModal({initialData, widget: mockWidget});
      userEvent.click(screen.getByText('events'));
      expect(initialData.router.push).toHaveBeenCalledWith({
        query: {sort: 'freq'},
      });
      // Need to manually set the new router location and rerender to simulate the sortable column click
      initialData.router.location.query = {sort: ['freq']};
      rerender(
        <WidgetViewerModal
          Header={stubEl}
          Footer={stubEl as ModalRenderProps['Footer']}
          Body={stubEl as ModalRenderProps['Body']}
          CloseButton={stubEl}
          closeModal={() => undefined}
          organization={initialData.organization}
          widget={mockWidget}
          onEdit={() => undefined}
        />
      );
      expect(issuesMock).toHaveBeenCalledWith(
        '/organizations/org-slug/issues/',
        expect.objectContaining({
          data: {
            cursor: undefined,
            environment: ['prod', 'dev'],
            expand: ['owners'],
            limit: 20,
            project: [1, 2],
            query: 'is:unresolved',
            sort: 'date',
            statsPeriod: '24h',
          },
        })
      );
    });

    it('renders pagination buttons', async function () {
      await renderModal({initialData, widget: mockWidget});
      expect(await screen.findByRole('button', {name: 'Previous'})).toBeInTheDocument();
      expect(screen.getByRole('button', {name: 'Next'})).toBeInTheDocument();
    });

    it('paginates to the next page', async function () {
      const {rerender} = await renderModal({initialData, widget: mockWidget});
      expect(await screen.findByText('Error: Failed')).toBeInTheDocument();
      userEvent.click(screen.getByRole('button', {name: 'Next'}));
      expect(issuesMock).toHaveBeenCalledTimes(1);
      expect(initialData.router.replace).toHaveBeenCalledWith(
        expect.objectContaining({
          query: {cursor: '0:10:0', page: 1},
        })
      );
      // Need to manually set the new router location and rerender to simulate the next page click
      initialData.router.location.query = {cursor: ['0:10:0']};
      rerender(
        <WidgetViewerModal
          Header={stubEl}
          Footer={stubEl as ModalRenderProps['Footer']}
          Body={stubEl as ModalRenderProps['Body']}
          CloseButton={stubEl}
          closeModal={() => undefined}
          organization={initialData.organization}
          widget={mockWidget}
          onEdit={() => undefined}
        />
      );
      expect(await screen.findByText('Another Error: Failed')).toBeInTheDocument();
    });

    it('displays with correct table column widths', async function () {
      initialData.router.location.query = {width: ['-1', '-1', '575']};
      await renderModal({initialData, widget: mockWidget});
      expect(screen.getByTestId('grid-editable')).toHaveStyle({
        'grid-template-columns':
          ' minmax(90px, auto) minmax(90px, auto) minmax(575px, auto)',
      });
    });

    it('uses provided tableData and does not make an issues requests', async function () {
      await renderModal({initialData, widget: mockWidget, tableData: []});
      expect(issuesMock).not.toHaveBeenCalled();
    });

    it('makes issues requests when table is sorted', async function () {
      await renderModal({
        initialData,
        widget: mockWidget,
        tableData: [],
      });
      expect(issuesMock).not.toHaveBeenCalled();
      userEvent.click(screen.getByText('events'));
      await waitFor(() => {
        expect(issuesMock).toHaveBeenCalled();
      });
    });
  });

  describe('Release Health Widgets', function () {
    let metricsMock;
    const mockQuery = {
      conditions: '',
      fields: [`sum(session)`],
      columns: [],
      aggregates: [],
      id: '1',
      name: 'Query Name',
      orderby: '',
    };
    const mockWidget = {
      id: '1',
      title: 'Release Widget',
      displayType: DisplayType.LINE,
      interval: '5m',
      queries: [mockQuery],
      widgetType: WidgetType.RELEASE,
    };
    beforeEach(function () {
      metricsMock = MockApiClient.addMockResponse({
        url: '/organizations/org-slug/metrics/data/',
        body: TestStubs.MetricsTotalCountByReleaseIn24h(),
        headers: {
          link:
            '<http://localhost/api/0/organizations/org-slug/metrics/data/?cursor=0:0:1>; rel="previous"; results="false"; cursor="0:0:1",' +
            '<http://localhost/api/0/organizations/org-slug/metrics/data/?cursor=0:10:0>; rel="next"; results="true"; cursor="0:10:0"',
        },
      });
    });
    it('does a sessions query', async function () {
      await renderModal({initialData, widget: mockWidget});
      expect(metricsMock).toHaveBeenCalled();
    });

    it('renders widget title', async function () {
      await renderModal({initialData, widget: mockWidget});
      expect(screen.getByText('Release Widget')).toBeInTheDocument();
    });

    it('renders Edit and Open in Releases buttons', async function () {
      await renderModal({initialData, widget: mockWidget});
      expect(screen.getByText('Edit Widget')).toBeInTheDocument();
      expect(screen.getByText('Open in Releases')).toBeInTheDocument();
    });

    it('Open in Releases button redirects browser', async function () {
      await renderModal({initialData, widget: mockWidget});
      userEvent.click(screen.getByText('Open in Releases'));
      expect(initialData.router.push).toHaveBeenCalledWith(
        '/organizations/org-slug/releases/?environment=prod&environment=dev&project=1&project=2&statsPeriod=24h'
      );
    });

    it('renders table header and body', async function () {
      await renderModal({initialData, widget: mockWidget});
      expect(screen.getByText('release')).toBeInTheDocument();
      expect(await screen.findByText('e102abb2c46e')).toBeInTheDocument();
      expect(screen.getByText('sum(session)')).toBeInTheDocument();
      expect(screen.getByText('6.3k')).toBeInTheDocument();
    });

    it('renders Release widget viewer', async function () {
      const {container} = await renderModal({initialData, widget: mockWidget});
      expect(await screen.findByText('e102abb2c46e')).toBeInTheDocument();
      expect(container).toSnapshot();
    });

    it('renders pagination buttons', async function () {
      await renderModal({
        initialData,
        widget: mockWidget,
      });
      expect(await screen.findByRole('button', {name: 'Previous'})).toBeInTheDocument();
      expect(screen.getByRole('button', {name: 'Next'})).toBeInTheDocument();
    });

    it('does not render pagination buttons when sorting by release', async function () {
      await renderModal({
        initialData,
        widget: {...mockWidget, queries: [{...mockQuery, orderby: 'release'}]},
      });
      expect(screen.queryByRole('button', {name: 'Previous'})).not.toBeInTheDocument();
      expect(screen.queryByRole('button', {name: 'Next'})).not.toBeInTheDocument();
    });

    it('makes a new sessions request after sorting by a table column', async function () {
      const {rerender} = await renderModal({
        initialData,
        widget: mockWidget,
        tableData: [],
        seriesData: [],
      });
      expect(metricsMock).toHaveBeenCalledTimes(1);
      userEvent.click(screen.getByText(`sum(session)`));
      expect(initialData.router.push).toHaveBeenCalledWith({
        query: {sort: '-sum(session)'},
      });
      // Need to manually set the new router location and rerender to simulate the sortable column click
      initialData.router.location.query = {sort: '-sum(session)'};
      rerender(
        <WidgetViewerModal
          Header={stubEl}
          Footer={stubEl as ModalRenderProps['Footer']}
          Body={stubEl as ModalRenderProps['Body']}
          CloseButton={stubEl}
          closeModal={() => undefined}
          organization={initialData.organization}
          widget={mockWidget}
          onEdit={() => undefined}
          seriesData={[]}
          tableData={[]}
        />
      );
      await waitFor(() => {
        expect(metricsMock).toHaveBeenCalledTimes(2);
      });
    });
  });
});