Reduce duplication in search.spec.ts

This commit is contained in:
Sam Rose 2024-04-11 11:58:21 +01:00
parent c53b824093
commit ba171bb5a2
No known key found for this signature in database
1 changed files with 144 additions and 148 deletions

View File

@ -6,11 +6,18 @@ import {
Datasource, Datasource,
EmptyFilterOption, EmptyFilterOption,
FieldType, FieldType,
Row, RowSearchParams,
SearchFilters, SortOrder,
Table, Table,
} from "@budibase/types" } 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)
}
jest.unmock("mssql") jest.unmock("mssql")
describe.each([ describe.each([
@ -47,10 +54,79 @@ describe.each([
} }
}) })
describe("strings", () => { async function testSearch<RowType extends Record<string, any>>(
beforeAll(async () => { test: SearchTest<RowType>,
table = await config.api.table.save( table: Table
tableForDatasource(datasource, { ) {
const expected = test.expectToFind
delete test.expectToFind
const { rows: foundRows } = await config.api.row.search(table._id!, {
...test,
tableId: table._id!,
})
if (!expected) {
return
}
expect(foundRows).toHaveLength(expected.length)
expect(foundRows).toEqual(
expect.arrayContaining(
expected.map(expectedRow =>
expect.objectContaining(
foundRows.find(foundRow => leftContainsRight(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}`
}
}
// eslint-disable-next-line jest/valid-title
describe(name, () => {
beforeAll(async () => {
table = await opts.table(datasource)
})
beforeAll(async () => {
await Promise.all(
opts.rows.map(r => config.api.row.save(table._id!, r))
)
})
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: { schema: {
name: { name: {
name: "name", name: "name",
@ -59,66 +135,38 @@ describe.each([
}, },
}) })
) )
}) },
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.
const rows = [{ name: "foo" }, { name: "bar" }] // @ts-expect-error - intentionally not passing a query to make sure the
let savedRows: Row[] // API can handle it.
{ expectToFind: [{ name: "foo" }, { name: "bar" }] },
beforeAll(async () => { { query: {}, expectToFind: [{ name: "foo" }, { name: "bar" }] },
savedRows = await Promise.all(
rows.map(r => config.api.row.save(table._id!, r))
)
})
interface StringSearchTest {
query: SearchFilters
expected: (typeof rows)[number][]
}
const stringSearchTests: StringSearchTest[] = [
// These three test cases are generic and don't really need
// to be repeated for all data types, so we just do them here.
{ query: {}, expected: rows },
{ {
query: { onEmptyFilter: EmptyFilterOption.RETURN_ALL }, query: { onEmptyFilter: EmptyFilterOption.RETURN_ALL },
expected: rows, expectToFind: [{ name: "foo" }, { name: "bar" }],
}, },
{ {
query: { onEmptyFilter: EmptyFilterOption.RETURN_NONE }, query: { onEmptyFilter: EmptyFilterOption.RETURN_NONE },
expected: [], expectToFind: [],
}, },
// The rest of these tests are specific to strings. // The rest of these tests are specific to strings.
{ query: { string: { name: "foo" } }, expected: [rows[0]] }, { query: { string: { name: "foo" } }, expectToFind: [{ name: "foo" }] },
{ query: { string: { name: "none" } }, expected: [] }, { query: { string: { name: "none" } }, expectToFind: [] },
{ query: { fuzzy: { name: "oo" } }, expected: [rows[0]] }, { query: { fuzzy: { name: "oo" } }, expectToFind: [{ name: "foo" }] },
{ query: { equal: { name: "foo" } }, expected: [rows[0]] }, { query: { equal: { name: "foo" } }, expectToFind: [{ name: "foo" }] },
{ query: { notEqual: { name: "foo" } }, expected: [rows[1]] }, { query: { notEqual: { name: "foo" } }, expectToFind: [{ name: "bar" }] },
{ query: { oneOf: { name: ["foo"] } }, expected: [rows[0]] }, { query: { oneOf: { name: ["foo"] } }, expectToFind: [{ name: "foo" }] },
] ],
it.each(stringSearchTests)(
`should be able to run query: $query`,
async ({ query, expected }) => {
const { rows: foundRows } = await config.api.row.search(table._id!, {
tableId: table._id!,
query,
})
expect(foundRows).toHaveLength(expected.length)
expect(foundRows).toEqual(
expect.arrayContaining(
expected.map(r =>
expect.objectContaining(savedRows.find(sr => sr.name === r.name)!)
)
)
)
}
)
}) })
describe("number", () => { searchTests("numbers", {
beforeAll(async () => { table: async ds => {
table = await config.api.table.save( return await config.api.table.save(
tableForDatasource(datasource, { tableForDatasource(ds, {
schema: { schema: {
age: { age: {
name: "age", name: "age",
@ -127,56 +175,33 @@ describe.each([
}, },
}) })
) )
}) },
rows: [{ age: 1 }, { age: 10 }],
const rows = [{ age: 1 }, { age: 10 }] tests: [
let savedRows: Row[] { query: { equal: { age: 1 } }, expectToFind: [{ age: 1 }] },
{ query: { equal: { age: 2 } }, expectToFind: [] },
beforeAll(async () => { { query: { notEqual: { age: 1 } }, expectToFind: [{ age: 10 }] },
savedRows = await Promise.all( { query: { oneOf: { age: [1] } }, expectToFind: [{ age: 1 }] },
rows.map(r => config.api.row.save(table._id!, r)) {
) query: { range: { age: { low: 1, high: 5 } } },
}) expectToFind: [{ age: 1 }],
},
interface NumberSearchTest { {
query: SearchFilters query: { range: { age: { low: 0, high: 1 } } },
expected: (typeof rows)[number][] expectToFind: [{ age: 1 }],
} },
{ query: { range: { age: { low: 3, high: 4 } } }, expectToFind: [] },
const numberSearchTests: NumberSearchTest[] = [ {
{ query: { equal: { age: 1 } }, expected: [rows[0]] }, query: { range: { age: { low: 0, high: 11 } } },
{ query: { equal: { age: 2 } }, expected: [] }, expectToFind: [{ age: 1 }, { age: 10 }],
{ query: { notEqual: { age: 1 } }, expected: [rows[1]] }, },
{ query: { oneOf: { age: [1] } }, expected: [rows[0]] }, ],
{ query: { range: { age: { low: 1, high: 5 } } }, expected: [rows[0]] },
{ query: { range: { age: { low: 0, high: 1 } } }, expected: [rows[0]] },
{ query: { range: { age: { low: 3, high: 4 } } }, expected: [] },
{ query: { range: { age: { low: 0, high: 11 } } }, expected: rows },
]
it.each(numberSearchTests)(
`should be able to run query: $query`,
async ({ query, expected }) => {
const { rows: foundRows } = await config.api.row.search(table._id!, {
tableId: table._id!,
query,
})
expect(foundRows).toHaveLength(expected.length)
expect(foundRows).toEqual(
expect.arrayContaining(
expected.map(r =>
expect.objectContaining(savedRows.find(sr => sr.age === r.age)!)
)
)
)
}
)
}) })
describe("dates", () => { searchTests("dates", {
beforeEach(async () => { table: async ds => {
table = await config.api.table.save( return await config.api.table.save(
tableForDatasource(datasource, { tableForDatasource(ds, {
schema: { schema: {
dob: { dob: {
name: "dob", name: "dob",
@ -185,41 +210,27 @@ describe.each([
}, },
}) })
) )
}) },
rows: [
const rows = [
{ dob: new Date("2020-01-01").toISOString() }, { dob: new Date("2020-01-01").toISOString() },
{ dob: new Date("2020-01-10").toISOString() }, { dob: new Date("2020-01-10").toISOString() },
] ],
let savedRows: Row[] tests: [
beforeEach(async () => {
savedRows = await Promise.all(
rows.map(r => config.api.row.save(table._id!, r))
)
})
interface DateSearchTest {
query: SearchFilters
expected: (typeof rows)[number][]
}
const dateSearchTests: DateSearchTest[] = [
{ {
query: { equal: { dob: new Date("2020-01-01").toISOString() } }, query: { equal: { dob: new Date("2020-01-01").toISOString() } },
expected: [rows[0]], expectToFind: [{ dob: new Date("2020-01-01").toISOString() }],
}, },
{ {
query: { equal: { dob: new Date("2020-01-02").toISOString() } }, query: { equal: { dob: new Date("2020-01-02").toISOString() } },
expected: [], expectToFind: [],
}, },
{ {
query: { notEqual: { dob: new Date("2020-01-01").toISOString() } }, query: { notEqual: { dob: new Date("2020-01-01").toISOString() } },
expected: [rows[1]], expectToFind: [{ dob: new Date("2020-01-10").toISOString() }],
}, },
{ {
query: { oneOf: { dob: [new Date("2020-01-01").toISOString()] } }, query: { oneOf: { dob: [new Date("2020-01-01").toISOString()] } },
expected: [rows[0]], expectToFind: [{ dob: new Date("2020-01-01").toISOString() }],
}, },
{ {
query: { query: {
@ -230,7 +241,7 @@ describe.each([
}, },
}, },
}, },
expected: [rows[0]], expectToFind: [{ dob: new Date("2020-01-01").toISOString() }],
}, },
{ {
query: { query: {
@ -241,7 +252,10 @@ describe.each([
}, },
}, },
}, },
expected: rows, expectToFind: [
{ dob: new Date("2020-01-01").toISOString() },
{ dob: new Date("2020-01-10").toISOString() },
],
}, },
{ {
query: { query: {
@ -252,26 +266,8 @@ describe.each([
}, },
}, },
}, },
expected: [rows[1]], expectToFind: [{ dob: new Date("2020-01-10").toISOString() }],
}, },
] ],
it.each(dateSearchTests)(
`should be able to run query: $query`,
async ({ query, expected }) => {
const { rows: foundRows } = await config.api.row.search(table._id!, {
tableId: table._id!,
query,
})
expect(foundRows).toHaveLength(expected.length)
expect(foundRows).toEqual(
expect.arrayContaining(
expected.map(r =>
expect.objectContaining(savedRows.find(sr => sr.dob === r.dob)!)
)
)
)
}
)
}) })
}) })