plugin-sdk-javascript/src/schema/table.test.ts at v0.1.32 · cloudquery/plugin-sdk-javascript · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
import test from 'ava';

import { filterTables, createTable, flattenTables } from './table.js';

const tableA = createTable({ name: 'a' });
const tableC = createTable({ name: 'c' });
const tableB = createTable({ name: 'b', relations: [tableC] });

const allTables = [tableA, tableB];

const testCases = [
  {
    name: 'should return all tables when * is specified',
    allTables,
    tables: ['*'],
    skipTables: [],
    skipDependentTables: false,
    expected: [tableA, tableB, tableC],
  },
  {
    name: 'should skip tables when * and skipTables are specified',
    allTables,
    tables: ['*'],
    skipTables: ['a'],
    skipDependentTables: false,
    expected: [tableB, tableC],
  },
  {
    name: 'should skip tables when skipTables is specified',
    allTables,
    tables: ['a'],
    skipTables: ['b'],
    skipDependentTables: false,
    expected: [tableA],
  },
  {
    name: 'should return only specified tables',
    allTables,
    tables: ['a'],
    skipTables: [],
    skipDependentTables: false,
    expected: [tableA],
  },
  {
    name: 'should return parent and child if child is specified',
    allTables,
    tables: ['c'],
    skipTables: [],
    skipDependentTables: false,
    expected: [tableB, tableC],
  },
  {
    name: 'should skip dependent tables when skipDependentTables is specified',
    allTables,
    tables: ['a', 'b'],
    skipTables: [],
    skipDependentTables: true,
    expected: [tableA, tableB],
  },
  {
    name: 'should error when child is included, but parent skipped',
    allTables,
    tables: ['c'],
    skipTables: ['b'],
    skipDependentTables: false,
    expected: [],
    expectedError: `Can't skip parent table when child table is included. Skipped parents are: b`,
  },
];

testCases.forEach((testCase) => {
  test(`filterTables - ${testCase.name}`, (t) => {
    const { allTables, tables, skipTables, skipDependentTables, expected, expectedError } = testCase;
    if (expectedError) {
      t.throws(() => filterTables(allTables, tables, skipTables, skipDependentTables), { message: expectedError });
      return;
    }
    const filtered = filterTables(allTables, tables, skipTables, skipDependentTables);
    const actual = flattenTables(filtered);
    t.deepEqual(
      actual.map(({ name }) => name),
      expected.map(({ name }) => name),
    );
  });
});