Skip to content

Commit 36814a5

Browse files
committed
fix tests
1 parent c59878f commit 36814a5

File tree

4 files changed

+18
-36
lines changed

4 files changed

+18
-36
lines changed

packages/cubejs-schema-compiler/test/integration/postgres/views-join-order-2.test.ts

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
import { prepareJsCompiler } from '../../unit/PrepareCompiler';
22
import { dbRunner } from './PostgresDBRunner';
3-
import { transformResultsForTesseractIfNeeded } from '../../unit/utils';
43

54
describe('Views Join Order 2', () => {
65
jest.setTimeout(200000);
@@ -161,7 +160,7 @@ cube('D', {
161160
total: true,
162161
renewQuery: false,
163162
limit: 1
164-
}, transformResultsForTesseractIfNeeded([{
163+
}, [{
165164
view___a_id: 1,
166165
view___a_name: 'a',
167166
view___b_id: 2,
@@ -170,5 +169,5 @@ cube('D', {
170169
view___d_name: 'd',
171170
view___e_id: 4,
172171
view___e_name: 'e',
173-
}]), { compiler, joinGraph, cubeEvaluator }));
172+
}], { compiler, joinGraph, cubeEvaluator }));
174173
});

packages/cubejs-schema-compiler/test/integration/postgres/views-join-order-3.test.ts

Lines changed: 10 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
import { prepareJsCompiler } from '../../unit/PrepareCompiler';
22
import { dbRunner } from './PostgresDBRunner';
3-
import { transformResultsForTesseractIfNeeded } from '../../unit/utils';
43

54
/**
65
* This tests the cube join correctness for cases, when there are
@@ -126,9 +125,9 @@ view(\`V\`, {
126125
timeDimensions: [],
127126
segments: [],
128127
filters: [],
129-
}, transformResultsForTesseractIfNeeded([{
128+
}, [{
130129
b___activity_balance: 125,
131-
}]), { compiler, joinGraph, cubeEvaluator });
130+
}], { compiler, joinGraph, cubeEvaluator });
132131

133132
expect(sql).toMatch(/AS "b"/);
134133
expect(sql).toMatch(/AS "d"/);
@@ -143,36 +142,36 @@ view(\`V\`, {
143142
timeDimensions: [],
144143
segments: [],
145144
filters: [],
146-
}, transformResultsForTesseractIfNeeded([{
145+
}, [{
147146
v___activity_balance: 125,
148-
}]), { compiler, joinGraph, cubeEvaluator }));
147+
}], { compiler, joinGraph, cubeEvaluator }));
149148

150149
it('correct join for simple view F-dimension', async () => dbRunner.runQueryTest({
151150
dimensions: ['V.PlanCode'],
152151
timeDimensions: [],
153152
segments: [],
154153
filters: [],
155-
}, transformResultsForTesseractIfNeeded([{
154+
}, [{
156155
v___plan_code: 'PLAN_CODE',
157-
}]), { compiler, joinGraph, cubeEvaluator }));
156+
}], { compiler, joinGraph, cubeEvaluator }));
158157

159158
it('correct join for view F-dimension + B-dimension', async () => dbRunner.runQueryTest({
160159
dimensions: ['V.PlanCode', 'V.ActivityBalance'],
161160
timeDimensions: [],
162161
segments: [],
163162
filters: [],
164-
}, transformResultsForTesseractIfNeeded([{
163+
}, [{
165164
v___plan_code: 'PLAN_CODE',
166165
v___activity_balance: 125,
167-
}]), { compiler, joinGraph, cubeEvaluator }));
166+
}], { compiler, joinGraph, cubeEvaluator }));
168167

169168
it('correct join for view B-dimension + F-dimension', async () => dbRunner.runQueryTest({
170169
dimensions: ['V.ActivityBalance', 'V.PlanCode'],
171170
timeDimensions: [],
172171
segments: [],
173172
filters: [],
174-
}, transformResultsForTesseractIfNeeded([{
173+
}, [{
175174
v___activity_balance: 125,
176175
v___plan_code: 'PLAN_CODE',
177-
}]), { compiler, joinGraph, cubeEvaluator }));
176+
}], { compiler, joinGraph, cubeEvaluator }));
178177
});

packages/cubejs-schema-compiler/test/integration/postgres/views-join-order-join-maps.test.ts

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
import { prepareJsCompiler } from '../../unit/PrepareCompiler';
22
import { dbRunner } from './PostgresDBRunner';
3-
import { transformResultsForTesseractIfNeeded } from '../../unit/utils';
43

54
describe('Views Join Order using join maps', () => {
65
jest.setTimeout(200000);
@@ -140,11 +139,11 @@ cube('D', {
140139
segments: [],
141140
filters: [],
142141
total: true,
143-
}, transformResultsForTesseractIfNeeded([{
142+
}, [{
144143
view___a_id: 1,
145144
view___a_name: 'a',
146145
view___a_d_name: 'd3',
147-
}]), { compiler, joinGraph, cubeEvaluator });
146+
}], { compiler, joinGraph, cubeEvaluator });
148147

149148
expect(sql).toMatch(/AS "b"/);
150149
expect(sql).toMatch(/AS "c"/);
@@ -166,11 +165,11 @@ cube('D', {
166165
segments: [],
167166
filters: [],
168167
total: true,
169-
}, transformResultsForTesseractIfNeeded([{
168+
}, [{
170169
view___a_id: 1,
171170
view___a_name: 'a',
172171
view___a_c_name: 'c1',
173-
}]), { compiler, joinGraph, cubeEvaluator });
172+
}], { compiler, joinGraph, cubeEvaluator });
174173

175174
expect(sql).toMatch(/AS "b"/);
176175
expect(sql).toMatch(/AS "c"/);
@@ -193,12 +192,12 @@ cube('D', {
193192
segments: [],
194193
filters: [],
195194
total: true,
196-
}, transformResultsForTesseractIfNeeded([{
195+
}, [{
197196
view___a_id: 1,
198197
view___a_name: 'a',
199198
view___a_c_name: 'c1',
200199
view___a_d_name: 'd3',
201-
}]), { compiler, joinGraph, cubeEvaluator });
200+
}], { compiler, joinGraph, cubeEvaluator });
202201

203202
expect(sql).toMatch(/AS "b"/);
204203
expect(sql).toMatch(/AS "c"/);

packages/cubejs-schema-compiler/test/unit/utils.ts

Lines changed: 0 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -769,18 +769,3 @@ export function createJoinedCubesSchema(): string {
769769
});
770770
`;
771771
}
772-
773-
export function transformResultsForTesseractIfNeeded(results: Record<string, any>[]) {
774-
if (getEnv('nativeSqlPlanner')) {
775-
return results.map(record => {
776-
const fixedRecord: Record<string, any> = {};
777-
for (const [key, value] of Object.entries(record)) {
778-
const fixedKey = key.replace(/___/g, '__');
779-
fixedRecord[fixedKey] = value;
780-
}
781-
return fixedRecord;
782-
});
783-
}
784-
785-
return results;
786-
}

0 commit comments

Comments
 (0)