Restructure search.spec.ts to be much more readable.

This commit is contained in:
Sam Rose 2024-04-11 15:16:26 +01:00
parent eb56140ce2
commit c07882b452
No known key found for this signature in database
1 changed files with 200 additions and 206 deletions

View File

@ -7,15 +7,11 @@ import {
EmptyFilterOption,
FieldType,
RowSearchParams,
SearchFilters,
Table,
TableSchema,
} from "@budibase/types"
function leftContainsRight<
A extends Record<string, any>,
B extends Record<string, any>
>(left: A, right: B) {
return Object.entries(right).every(([k, v]) => left[k] === v)
}
import _ from "lodash"
jest.unmock("mssql")
@ -32,6 +28,7 @@ describe.each([
let envCleanup: (() => void) | undefined
let datasource: Datasource | undefined
let table: Table
beforeAll(async () => {
if (isSqs) {
@ -52,220 +49,217 @@ describe.each([
}
})
async function testSearch<RowType extends Record<string, any>>(
test: SearchTest<RowType>,
table: Table
) {
const expected = test.expectToFind
delete test.expectToFind
async function createTable(schema: TableSchema) {
table = await config.api.table.save(
tableForDatasource(datasource, { schema })
)
}
async function createRows(rows: Record<string, any>[]) {
await Promise.all(rows.map(r => config.api.row.save(table._id!, r)))
}
class SearchAssertion {
constructor(private readonly query: RowSearchParams) {}
async toFind(expectedRows: any[]) {
const { rows: foundRows } = await config.api.row.search(table._id!, {
...test,
...this.query,
tableId: table._id!,
})
if (!expected) {
return
}
expect(foundRows).toHaveLength(expected.length)
// eslint-disable-next-line jest/no-standalone-expect
expect(foundRows).toHaveLength(expectedRows.length)
// eslint-disable-next-line jest/no-standalone-expect
expect(foundRows).toEqual(
expect.arrayContaining(
expected.map(expectedRow =>
expectedRows.map((expectedRow: any) =>
expect.objectContaining(
foundRows.find(foundRow => leftContainsRight(foundRow, expectedRow))
foundRows.find(foundRow => _.isMatch(foundRow, expectedRow))
)
)
)
)
}
function searchTests<T extends Record<string, any>>(
name: string,
opts: {
table: (ds?: Datasource) => Promise<Table>
rows: T[]
tests: SearchTest<T>[]
}
) {
let table: Table
for (const test of opts.tests) {
test.toString = () => {
const queryStr = JSON.stringify({
query: test.query,
limit: test.limit,
sort: test.sort,
sortOrder: test.sortOrder,
})
const expectStr = JSON.stringify(test.expectToFind)
return `should run: ${queryStr} and find ${expectStr}`
async toFindNothing() {
await this.toFind([])
}
}
// eslint-disable-next-line jest/valid-title
describe(name, () => {
function expectSearch(query: Omit<RowSearchParams, "tableId">) {
return new SearchAssertion({ ...query, tableId: table._id! })
}
function expectQuery(query: SearchFilters) {
return expectSearch({ query })
}
describe("strings", () => {
beforeAll(async () => {
table = await opts.table(datasource)
await createTable({
name: { name: "name", type: FieldType.STRING },
})
await createRows([{ name: "foo" }, { name: "bar" }])
})
describe("misc", () => {
it("should return all if no query is passed", () =>
expectSearch({} as RowSearchParams).toFind([
{ name: "foo" },
{ name: "bar" },
]))
it("should return all if empty query is passed", () =>
expectQuery({}).toFind([{ name: "foo" }, { name: "bar" }]))
it("should return all if onEmptyFilter is RETURN_ALL", () =>
expectQuery({
onEmptyFilter: EmptyFilterOption.RETURN_ALL,
}).toFind([{ name: "foo" }, { name: "bar" }]))
it("should return nothing if onEmptyFilter is RETURN_NONE", () =>
expectQuery({
onEmptyFilter: EmptyFilterOption.RETURN_NONE,
}).toFindNothing())
})
describe("equal", () => {
it("successfully finds a row", () =>
expectQuery({ equal: { name: "foo" } }).toFind([{ name: "foo" }]))
it("fails to find nonexistent row", () =>
expectQuery({ equal: { name: "none" } }).toFindNothing())
})
describe("notEqual", () => {
it("successfully finds a row", () =>
expectQuery({ notEqual: { name: "foo" } }).toFind([{ name: "bar" }]))
it("fails to find nonexistent row", () =>
expectQuery({ notEqual: { name: "bar" } }).toFind([{ name: "foo" }]))
})
describe("oneOf", () => {
it("successfully finds a row", () =>
expectQuery({ oneOf: { name: ["foo"] } }).toFind([{ name: "foo" }]))
it("fails to find nonexistent row", () =>
expectQuery({ oneOf: { name: ["none"] } }).toFindNothing())
})
describe("fuzzy", () => {
it("successfully finds a row", () =>
expectQuery({ fuzzy: { name: "oo" } }).toFind([{ name: "foo" }]))
it("fails to find nonexistent row", () =>
expectQuery({ fuzzy: { name: "none" } }).toFindNothing())
})
})
describe("numbers", () => {
beforeAll(async () => {
await createTable({
age: { name: "age", type: FieldType.NUMBER },
})
await createRows([{ age: 1 }, { age: 10 }])
})
describe("equal", () => {
it("successfully finds a row", () =>
expectQuery({ equal: { age: 1 } }).toFind([{ age: 1 }]))
it("fails to find nonexistent row", () =>
expectQuery({ equal: { age: 2 } }).toFindNothing())
})
describe("notEqual", () => {
it("successfully finds a row", () =>
expectQuery({ notEqual: { age: 1 } }).toFind([{ age: 10 }]))
it("fails to find nonexistent row", () =>
expectQuery({ notEqual: { age: 10 } }).toFind([{ age: 1 }]))
})
describe("oneOf", () => {
it("successfully finds a row", () =>
expectQuery({ oneOf: { age: [1] } }).toFind([{ age: 1 }]))
it("fails to find nonexistent row", () =>
expectQuery({ oneOf: { age: [2] } }).toFindNothing())
})
describe("range", () => {
it("successfully finds a row", () =>
expectQuery({
range: { age: { low: 1, high: 5 } },
}).toFind([{ age: 1 }]))
it("successfully finds multiple rows", () =>
expectQuery({
range: { age: { low: 1, high: 10 } },
}).toFind([{ age: 1 }, { age: 10 }]))
it("successfully finds a row with a high bound", () =>
expectQuery({
range: { age: { low: 5, high: 10 } },
}).toFind([{ age: 10 }]))
})
})
describe("dates", () => {
const JAN_1ST = "2020-01-01T00:00:00.000Z"
const JAN_2ND = "2020-01-02T00:00:00.000Z"
const JAN_5TH = "2020-01-05T00:00:00.000Z"
const JAN_10TH = "2020-01-10T00:00:00.000Z"
beforeAll(async () => {
await Promise.all(
opts.rows.map(r => config.api.row.save(table._id!, r))
)
await createTable({
dob: { name: "dob", type: FieldType.DATETIME },
})
it.each(opts.tests)(`%s`, test => testSearch(test, table))
})
}
interface SearchTest<RowType extends Record<string, any>>
extends Omit<RowSearchParams, "tableId"> {
expectToFind?: RowType[]
}
searchTests("strings", {
table: async ds => {
return await config.api.table.save(
tableForDatasource(ds, {
schema: {
name: {
name: "name",
type: FieldType.STRING,
},
},
})
)
},
rows: [{ name: "foo" }, { name: "bar" }],
tests: [
// These test cases are generic and don't really need to be repeated for
// all data types, so we just do them here.
// @ts-expect-error - intentionally not passing a query to make sure the
// API can handle it.
{ expectToFind: [{ name: "foo" }, { name: "bar" }] },
{ query: {}, expectToFind: [{ name: "foo" }, { name: "bar" }] },
{
query: { onEmptyFilter: EmptyFilterOption.RETURN_ALL },
expectToFind: [{ name: "foo" }, { name: "bar" }],
},
{
query: { onEmptyFilter: EmptyFilterOption.RETURN_NONE },
expectToFind: [],
},
// The rest of these tests are specific to strings.
{ query: { string: { name: "foo" } }, expectToFind: [{ name: "foo" }] },
{ query: { string: { name: "none" } }, expectToFind: [] },
{ query: { fuzzy: { name: "oo" } }, expectToFind: [{ name: "foo" }] },
{ query: { equal: { name: "foo" } }, expectToFind: [{ name: "foo" }] },
{ query: { notEqual: { name: "foo" } }, expectToFind: [{ name: "bar" }] },
{ query: { oneOf: { name: ["foo"] } }, expectToFind: [{ name: "foo" }] },
],
await createRows([{ dob: JAN_1ST }, { dob: JAN_10TH }])
})
searchTests("numbers", {
table: async ds => {
return await config.api.table.save(
tableForDatasource(ds, {
schema: {
age: {
name: "age",
type: FieldType.NUMBER,
},
},
})
)
},
rows: [{ age: 1 }, { age: 10 }],
tests: [
{ query: { equal: { age: 1 } }, expectToFind: [{ age: 1 }] },
{ query: { equal: { age: 2 } }, expectToFind: [] },
{ query: { notEqual: { age: 1 } }, expectToFind: [{ age: 10 }] },
{ query: { oneOf: { age: [1] } }, expectToFind: [{ age: 1 }] },
{
query: { range: { age: { low: 1, high: 5 } } },
expectToFind: [{ age: 1 }],
},
{
query: { range: { age: { low: 0, high: 1 } } },
expectToFind: [{ age: 1 }],
},
{ query: { range: { age: { low: 3, high: 4 } } }, expectToFind: [] },
{
query: { range: { age: { low: 0, high: 11 } } },
expectToFind: [{ age: 1 }, { age: 10 }],
},
],
describe("equal", () => {
it("successfully finds a row", () =>
expectQuery({ equal: { dob: JAN_1ST } }).toFind([{ dob: JAN_1ST }]))
it("fails to find nonexistent row", () =>
expectQuery({ equal: { dob: JAN_2ND } }).toFindNothing())
})
searchTests("dates", {
table: async ds => {
return await config.api.table.save(
tableForDatasource(ds, {
schema: {
dob: {
name: "dob",
type: FieldType.DATETIME,
},
},
describe("notEqual", () => {
it("successfully finds a row", () =>
expectQuery({ notEqual: { dob: JAN_1ST } }).toFind([{ dob: JAN_10TH }]))
it("fails to find nonexistent row", () =>
expectQuery({ notEqual: { dob: JAN_10TH } }).toFind([{ dob: JAN_1ST }]))
})
describe("oneOf", () => {
it("successfully finds a row", () =>
expectQuery({ oneOf: { dob: [JAN_1ST] } }).toFind([{ dob: JAN_1ST }]))
it("fails to find nonexistent row", () =>
expectQuery({ oneOf: { dob: [JAN_2ND] } }).toFindNothing())
})
describe("range", () => {
it("successfully finds a row", () =>
expectQuery({
range: { dob: { low: JAN_1ST, high: JAN_5TH } },
}).toFind([{ dob: JAN_1ST }]))
it("successfully finds multiple rows", () =>
expectQuery({
range: { dob: { low: JAN_1ST, high: JAN_10TH } },
}).toFind([{ dob: JAN_1ST }, { dob: JAN_10TH }]))
it("successfully finds a row with a high bound", () =>
expectQuery({
range: { dob: { low: JAN_5TH, high: JAN_10TH } },
}).toFind([{ dob: JAN_10TH }]))
})
)
},
rows: [
{ dob: "2020-01-01T00:00:00.000Z" },
{ dob: "2020-01-10T00:00:00.000Z" },
],
tests: [
{
query: { equal: { dob: "2020-01-01T00:00:00.000Z" } },
expectToFind: [{ dob: "2020-01-01T00:00:00.000Z" }],
},
{
query: { equal: { dob: "2020-01-02T00:00:00.000Z" } },
expectToFind: [],
},
{
query: { notEqual: { dob: "2020-01-01T00:00:00.000Z" } },
expectToFind: [{ dob: "2020-01-10T00:00:00.000Z" }],
},
{
query: { oneOf: { dob: ["2020-01-01T00:00:00.000Z"] } },
expectToFind: [{ dob: "2020-01-01T00:00:00.000Z" }],
},
{
query: {
range: {
dob: {
low: "2020-01-01T00:00:00.000Z",
high: "2020-01-05T00:00:00.000Z",
},
},
},
expectToFind: [{ dob: "2020-01-01T00:00:00.000Z" }],
},
{
query: {
range: {
dob: {
low: "2020-01-01T00:00:00.000Z",
high: "2020-01-10T00:00:00.000Z",
},
},
},
expectToFind: [
{ dob: "2020-01-01T00:00:00.000Z" },
{ dob: "2020-01-10T00:00:00.000Z" },
],
},
{
query: {
range: {
dob: {
low: "2020-01-05T00:00:00.000Z",
high: "2020-01-10T00:00:00.000Z",
},
},
},
expectToFind: [{ dob: "2020-01-10T00:00:00.000Z" }],
},
],
})
})