grafana/public/app/plugins/datasource/zipkin/QueryField.test.tsx

import { CascaderOption } from '@grafana/ui';
import { act, renderHook } from '@testing-library/react-hooks';
import { render, screen } from '@testing-library/react';
import React from 'react';
import { ZipkinDatasource } from './datasource';
import { ZipkinQueryField, useLoadOptions, useServices } from './QueryField';
import { ZipkinQuery } from './types';

describe('QueryField', () => {
  it('renders properly', () => {
    const ds = {} as ZipkinDatasource;
    render(
      <ZipkinQueryField
        history={[]}
        datasource={ds}
        query={{ query: '1234' } as ZipkinQuery}
        onRunQuery={() => {}}
        onChange={() => {}}
      />
    );

    expect(screen.getByText(/1234/i)).toBeInTheDocument();
    expect(screen.getByText(/Traces/i)).toBeInTheDocument();
  });
});

describe('useServices', () => {
  it('returns services from datasource', async () => {
    const ds = {
      async metadataRequest(url: string, params?: Record<string, any>): Promise<any> {
        if (url === '/api/v2/services') {
          return Promise.resolve(['service1', 'service2']);
        }
      },
    } as ZipkinDatasource;

    const { result, waitForNextUpdate } = renderHook(() => useServices(ds));
    await waitForNextUpdate();
    expect(result.current.value).toEqual([
      { label: 'service1', value: 'service1', isLeaf: false },
      { label: 'service2', value: 'service2', isLeaf: false },
    ]);
  });
});

describe('useLoadOptions', () => {
  it('loads spans and traces', async () => {
    const ds = {
      async metadataRequest(url: string, params?: Record<string, any>): Promise<any> {
        if (url === '/api/v2/spans' && params?.serviceName === 'service1') {
          return Promise.resolve(['span1', 'span2']);
        }

        if (url === '/api/v2/traces' && params?.serviceName === 'service1' && params?.spanName === 'span1') {
          return Promise.resolve([[{ name: 'trace1', duration: 10_000, traceId: 'traceId1' }]]);
        }
      },
    } as ZipkinDatasource;

    const { result, waitForNextUpdate } = renderHook(() => useLoadOptions(ds));
    expect(result.current.allOptions).toEqual({});

    act(() => {
      result.current.onLoadOptions([{ value: 'service1' } as CascaderOption]);
    });

    await waitForNextUpdate();

    expect(result.current.allOptions).toEqual({ service1: { span1: undefined, span2: undefined } });

    act(() => {
      result.current.onLoadOptions([{ value: 'service1' } as CascaderOption, { value: 'span1' } as CascaderOption]);
    });

    await waitForNextUpdate();

    expect(result.current.allOptions).toEqual({
      service1: { span1: { 'trace1 [10 ms]': 'traceId1' }, span2: undefined },
    });
  });
});