sentinel/static/app/utils/performance/contexts/onDemandControl.spec.tsx

import {_isOnDemandMetricWidget} from 'sentry/utils/performance/contexts/onDemandControl';
import type {Widget} from 'sentry/views/dashboards/types';
import {WidgetType} from 'sentry/views/dashboards/types';

function widget(
  aggregates: string[],
  conditions: string,
  type: WidgetType = WidgetType.DISCOVER
) {
  const queries = aggregates.map(_ => ({
    aggregates,
    columns: [],
    conditions,
    name: '',
    orderby: '',
  }));

  return {
    widgetType: type,
    displayType: 'line',
    title: 'title',
    interval: '5m',
    queries,
  } as Widget;
}

describe('isOnDemandMetricWidget', () => {
  it('should return true for a widget that contains non standard fields', () => {
    expect(
      _isOnDemandMetricWidget(widget(['count()'], 'transaction.duration:>1'))
    ).toBeTruthy();
    expect(_isOnDemandMetricWidget(widget(['count()'], 'device.name:foo'))).toBeTruthy();
    expect(_isOnDemandMetricWidget(widget(['count()'], 'geo.region:>US'))).toBeTruthy();
  });

  it('should return false for a widget that has only standard fields', () => {
    expect(_isOnDemandMetricWidget(widget(['count()'], 'release:1.0'))).toBeFalsy();
    expect(_isOnDemandMetricWidget(widget(['count()'], 'platform:foo'))).toBeFalsy();
  });

  it('should return false for a widget that has multiple or unsupported aggregates', () => {
    expect(
      _isOnDemandMetricWidget(
        widget(['count()', 'count_unique()'], 'transaction.duration:>1')
      )
    ).toBeFalsy();
    expect(
      _isOnDemandMetricWidget(widget(['apdex(100)'], 'transaction.duration:>1'))
    ).toBeFalsy();
  });
});