sql: extract frontend code into separate package (#81109)

* sql: extract frontend code into separate package * updated package version
2024-01-26 11:38:29 +01:00
parent 2febbec758
commit 29e8a355cb
87 changed files with 187 additions and 104 deletions
@@ -0,0 +1,94 @@
+import { TimeRange, PluginType } from '@grafana/data';
+
+import { DB, SQLQuery, SQLSelectableValue, ValidationResults } from '../types';
+
+import { DatasetSelectorProps } from './DatasetSelector';
+import { TableSelectorProps } from './TableSelector';
+
+const buildMockDB = (): DB => ({
+  datasets: jest.fn(() => Promise.resolve(['dataset1', 'dataset2'])),
+  tables: jest.fn((_ds: string | undefined) => Promise.resolve(['table1', 'table2'])),
+  fields: jest.fn((_query: SQLQuery, _order?: boolean) => Promise.resolve<SQLSelectableValue[]>([])),
+  validateQuery: jest.fn((_query: SQLQuery, _range?: TimeRange) =>
+    Promise.resolve<ValidationResults>({ query: { refId: '123' }, error: '', isError: false, isValid: true })
+  ),
+  dsID: jest.fn(() => 1234),
+  getEditorLanguageDefinition: jest.fn(() => ({ id: '4567' })),
+  toRawSql: (_query: SQLQuery) => '',
+});
+
+// This data is of type `SqlDatasource`
+export const buildMockDatasource = (hasDefaultDatabaseConfigured?: boolean) => {
+  return {
+    id: Infinity,
+    type: '',
+    name: '',
+    uid: '',
+    responseParser: { transformMetricFindResponse: jest.fn() },
+    interval: '',
+    db: buildMockDB(),
+    preconfiguredDatabase: hasDefaultDatabaseConfigured ? 'default database' : '',
+    getDB: () => buildMockDB(),
+    getQueryModel: jest.fn(),
+    getResponseParser: jest.fn(),
+    interpolateVariable: jest.fn(),
+    interpolateVariablesInQueries: jest.fn(),
+    filterQuery: jest.fn(),
+    applyTemplateVariables: jest.fn(),
+    metricFindQuery: jest.fn(),
+    templateSrv: {
+      getVariables: jest.fn(),
+      replace: jest.fn(),
+      containsTemplate: jest.fn(),
+      updateTimeRange: jest.fn(),
+    },
+    runSql: jest.fn(),
+    runMetaQuery: jest.fn(),
+    targetContainsTemplate: jest.fn(),
+    query: jest.fn(),
+    getRequestHeaders: jest.fn(),
+    streamOptionsProvider: jest.fn(),
+    getResource: jest.fn(),
+    postResource: jest.fn(),
+    callHealthCheck: jest.fn(),
+    testDatasource: jest.fn(),
+    getRef: jest.fn(),
+    meta: {
+      id: '',
+      name: '',
+      type: PluginType.panel,
+      info: {
+        author: { name: '' },
+        description: '',
+        links: [],
+        logos: { large: '', small: '' },
+        screenshots: [],
+        updated: '',
+        version: '',
+      },
+      module: '',
+      baseUrl: '',
+    },
+  };
+};
+
+export function buildMockDatasetSelectorProps(overrides?: Partial<DatasetSelectorProps>): DatasetSelectorProps {
+  return {
+    db: buildMockDB(),
+    dataset: '',
+    dialect: 'other',
+    onChange: jest.fn(),
+    preconfiguredDataset: '',
+    ...overrides,
+  };
+}
+
+export function buildMockTableSelectorProps(overrides?: Partial<TableSelectorProps>): TableSelectorProps {
+  return {
+    db: buildMockDB(),
+    dataset: '',
+    table: '',
+    onChange: jest.fn(),
+    ...overrides,
+  };
+}