ckaczor/azuredatastudio
1
0
Fork 0
mirror of https://github.com/ckaczor/azuredatastudio.git synced 2026-02-16 10:58:30 -05:00
Code Issues Packages Projects Releases Wiki Activity

Use column names as keys for table data in SQL cell outputs (#22688)

Browse Source
This commit is contained in:
Cory Rivera
2023-04-13 13:26:12 -07:00
committed by GitHub
parent 8202e1ec4e
commit 8aade56214
6 changed files with 86 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
extensions/integration-tests/src/test/notebook.test.ts
View File

@@ -52,8 +52,10 @@ suite('Notebook integration test suite', function () {
let actualOutput0 = (<azdata.nb.IDisplayData>cellOutputs[0]).data['text/html'];
console.log('Got first output');
assert(actualOutput0 === expectedOutput0, `Expected row count: ${expectedOutput0}, Actual: ${actualOutput0}`);
let actualOutput2 = (<azdata.nb.IExecuteResult>cellOutputs[2]).data['application/vnd.dataresource+json'].data[0];
assert(actualOutput2[0] === '1', `Expected result: 1, Actual: '${actualOutput2[0]}'`);
let dataResource = (<azdata.nb.IExecuteResult>cellOutputs[2]).data['application/vnd.dataresource+json'];
let actualOutput2 = dataResource.data[0];
let columnName = dataResource.schema.fields[0].name;
assert(actualOutput2[columnName] === '1', `Expected result: 1, Actual: '${actualOutput2[columnName]}'`);
});
test('Sql NB multiple cells test', async function () {
@@ -283,8 +285,9 @@ async function multipleCellsTest(relativeFilePath: string): Promise<void> {
assert(Object.keys(applicationDataResource).includes('data'), `Execute result did not include data key. It included ${Object.keys(applicationDataResource)}`);
const actualOutput2 = applicationDataResource.data[0];
const columnName = applicationDataResource.schema.fields[0].name;
assert(actualOutput2[0] === i.toString(), `Expected result: ${i.toString()}, Actual: '${actualOutput2[0]}'`);
assert(actualOutput2[columnName] === i.toString(), `Expected result: ${i.toString()}, Actual: '${actualOutput2[columnName]}'`);
console.log('Sql multiple cells NB done');
}
}