Zoltán Bedi
106 lines
3.5 KiB
TypeScript
106 lines
3.5 KiB
TypeScript
import { render, waitFor } from '@testing-library/react';
|
|
|
|
import { SQLExpression } from '../types';
|
|
import { makeVariable } from '../utils/testHelpers';
|
|
|
|
import { DatasetSelector } from './DatasetSelector';
|
|
import { buildMockDatasetSelectorProps, buildMockTableSelectorProps } from './SqlComponents.testHelpers';
|
|
import { TableSelector } from './TableSelector';
|
|
import { removeQuotesForMultiVariables } from './visual-query-builder/SQLWhereRow';
|
|
|
|
describe('DatasetSelector', () => {
|
|
it('should only query the database when needed', async () => {
|
|
const mockProps = buildMockDatasetSelectorProps();
|
|
render(<DatasetSelector {...mockProps} />);
|
|
|
|
await waitFor(() => {
|
|
expect(mockProps.db.datasets).toHaveBeenCalled();
|
|
});
|
|
});
|
|
|
|
it('should not query the database if Postgres instance, and no preconfigured database', async () => {
|
|
const mockProps = buildMockDatasetSelectorProps({ dialect: 'postgres' });
|
|
render(<DatasetSelector {...mockProps} />);
|
|
|
|
await waitFor(() => {
|
|
expect(mockProps.db.datasets).not.toHaveBeenCalled();
|
|
});
|
|
});
|
|
|
|
it('should not query the database if preconfigured', async () => {
|
|
const mockProps = buildMockDatasetSelectorProps({ preconfiguredDataset: 'database 1' });
|
|
render(<DatasetSelector {...mockProps} />);
|
|
|
|
await waitFor(() => {
|
|
expect(mockProps.db.datasets).not.toHaveBeenCalled();
|
|
});
|
|
});
|
|
});
|
|
|
|
describe('TableSelector', () => {
|
|
it('should only query the database when needed', async () => {
|
|
const mockProps = buildMockTableSelectorProps({ dataset: 'database 1' });
|
|
render(<TableSelector {...mockProps} />);
|
|
|
|
await waitFor(() => {
|
|
expect(mockProps.db.tables).toHaveBeenCalled();
|
|
});
|
|
});
|
|
|
|
it('should not query the database if no dataset is passed as a prop', async () => {
|
|
const mockProps = buildMockTableSelectorProps();
|
|
render(<TableSelector {...mockProps} />);
|
|
|
|
await waitFor(() => {
|
|
expect(mockProps.db.tables).not.toHaveBeenCalled();
|
|
});
|
|
});
|
|
});
|
|
|
|
describe('SQLWhereRow', () => {
|
|
it('should remove quotes in a where clause including multi-value variable', () => {
|
|
const exp: SQLExpression = {
|
|
whereString: "hostname IN ('${multiHost}')",
|
|
};
|
|
|
|
const multiVar = makeVariable('multiVar', 'multiHost', { multi: true });
|
|
const nonMultiVar = makeVariable('nonMultiVar', 'host', { multi: false });
|
|
|
|
const variables = [multiVar, nonMultiVar];
|
|
|
|
removeQuotesForMultiVariables(exp, variables);
|
|
|
|
expect(exp.whereString).toBe('hostname IN (${multiHost})');
|
|
});
|
|
|
|
it('should not remove quotes in a where clause including a non-multi variable', () => {
|
|
const exp: SQLExpression = {
|
|
whereString: "hostname IN ('${host}')",
|
|
};
|
|
|
|
const multiVar = makeVariable('multiVar', 'multiHost', { multi: true });
|
|
const nonMultiVar = makeVariable('nonMultiVar', 'host', { multi: false });
|
|
|
|
const variables = [multiVar, nonMultiVar];
|
|
|
|
removeQuotesForMultiVariables(exp, variables);
|
|
|
|
expect(exp.whereString).toBe("hostname IN ('${host}')");
|
|
});
|
|
|
|
it('should not remove quotes in a where clause not including any known variables', () => {
|
|
const exp: SQLExpression = {
|
|
whereString: "hostname IN ('${nonMultiHost}')",
|
|
};
|
|
|
|
const multiVar = makeVariable('multiVar', 'multiHost', { multi: true });
|
|
const nonMultiVar = makeVariable('nonMultiVar', 'host', { multi: false });
|
|
|
|
const variables = [multiVar, nonMultiVar];
|
|
|
|
removeQuotesForMultiVariables(exp, variables);
|
|
|
|
expect(exp.whereString).toBe("hostname IN ('${nonMultiHost}')");
|
|
});
|
|
});
|