scottk/clickhouse
1
0
Fork 0
mirror of https://github.com/lingble/clickhouse.git synced 2025-12-24 22:47:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

added CSV and TSV Streams on CSVWithNames, TabSeparatedWithNames format query cases

Browse Source
This commit is contained in:
Pavel
2019-09-29 00:03:43 +04:00
parent f42b32907f
commit eedce5f094
5 changed files with 171 additions and 71 deletions
Download Patch File Download Diff File Expand all files Collapse all files

86
test/test.js
View File

@@ -135,6 +135,46 @@ describe('Select', () => {
clickhouse.query(sql).stream()
.on('data', () => ++i)
// TODO: on this case you should catch error
.on('error', err => callback(err))
.on('end', () => {
expect(error).to.not.be.ok();
expect(i).to.be(rowCount);
callback();
});
});
it('use stream with csv format', function(callback) {
// this.timeout(10000);
let i = 0;
let error = null;
clickhouse.query(`${sql} format CSVWithNames`).stream()
.on('data', () => {
++i
})
// TODO: on this case you should catch error
.on('error', err => callback(err))
.on('end', () => {
expect(error).to.not.be.ok();
expect(i).to.be(rowCount);
callback();
});
});
it('use stream with tsv format', function(callback) {
// this.timeout(10000);
let i = 0;
let error = null;
clickhouse.query(`${sql} format TabSeparatedWithNames`).stream()
.on('data', () => {
++i
})
// TODO: on this case you should catch error
.on('error', err => error = err)
.on('end', () => {
expect(error).to.not.be.ok();
@@ -191,6 +231,32 @@ describe('Select', () => {
expect(i).to.be(rowCount);
});
it('use async for with csv format', async function() {
let i = 0;
for await (const row of clickhouse.query(`${sql} format CSVWithNames`).stream()) {
++i;
expect(row).to.have.key('number');
expect(row).to.have.key('str');
expect(row).to.have.key('date');
}
expect(i).to.be(rowCount);
});
it('use async for with tsv format', async function() {
let i = 0;
for await (const row of clickhouse.query(`${sql} format TabSeparatedWithNames`).stream()) {
++i;
expect(row).to.have.key('number');
expect(row).to.have.key('str');
expect(row).to.have.key('date');
}
expect(i).to.be(rowCount);
});
}
@@ -445,17 +511,17 @@ describe('response codes', () => {
expect(err.code).to.be(60);
}
// try {
// let result = await clickhouse.query('DROP TABLE session_temp2').toPromise();
// expect(result).to.be.ok();
try {
let result = await clickhouse.query('DROP TABLE session_temp2').toPromise();
expect(result).to.be.ok();
// await clickhouse.query('SELECT COUNT(*) AS count FROM session_temp2').toPromise();
// expect().fail('You should not be here2');
// } catch (err) {
// expect(err).to.be.ok();
// expect(err).to.have.key('code');
// expect(err.code).to.be(60);
// }
await clickhouse.query('SELECT COUNT(*) AS count FROM session_temp2').toPromise();
expect().fail('You should not be here2');
} catch (err) {
expect(err).to.be.ok();
expect(err).to.have.key('code');
expect(err.code).to.be(60);
}
});
});