This commit is contained in:
Jan Prochazka
2021-06-03 12:33:02 +02:00
parent 2a34a3b48b
commit bebdf3f43b
2 changed files with 15 additions and 3 deletions

View File

@@ -119,7 +119,7 @@ describe('Query', () => {
); );
test.each(engines.map(engine => [engine.label, engine]))( test.each(engines.map(engine => [engine.label, engine]))(
'Script - %s', 'Script - return data - %s',
testWrapper(async (conn, driver, engine) => { testWrapper(async (conn, driver, engine) => {
const results = await executeStream( const results = await executeStream(
driver, driver,
@@ -134,6 +134,18 @@ describe('Query', () => {
}) })
); );
test.each(engines.map(engine => [engine.label, engine]))(
'Script - no data - %s',
testWrapper(async (conn, driver, engine) => {
const results = await executeStream(
driver,
conn,
'CREATE TABLE t1 (id int); INSERT INTO t1 (id) VALUES (1); INSERT INTO t1 (id) VALUES (2) '
);
expect(results.length).toEqual(0);
})
);
test.each(engines.map(engine => [engine.label, engine]))( test.each(engines.map(engine => [engine.label, engine]))(
'Save data query - %s', 'Save data query - %s',
testWrapper(async (conn, driver, engine) => { testWrapper(async (conn, driver, engine) => {
@@ -144,7 +156,7 @@ describe('Query', () => {
'INSERT INTO t1 (id) VALUES (3);INSERT INTO t1 (id) VALUES (4);UPDATE t1 SET id=10 WHERE id=1;DELETE FROM t1 WHERE id=2;' 'INSERT INTO t1 (id) VALUES (3);INSERT INTO t1 (id) VALUES (4);UPDATE t1 SET id=10 WHERE id=1;DELETE FROM t1 WHERE id=2;'
); );
const res = await driver.query(conn, 'SELECT COUNT(*) AS cnt FROM t1'); const res = await driver.query(conn, 'SELECT COUNT(*) AS cnt FROM t1');
console.log(res); // console.log(res);
expect(res.rows[0].cnt == 3).toBeTruthy(); expect(res.rows[0].cnt == 3).toBeTruthy();
}) })
); );

View File

@@ -125,7 +125,7 @@ async function handleRunScript({ msgid, sql }) {
await waitConnected(); await waitConnected();
const driver = requireEngineDriver(storedConnection); const driver = requireEngineDriver(storedConnection);
try { try {
await driver.script(sql); await driver.script(systemConnection, sql);
process.send({ msgtype: 'response', msgid }); process.send({ msgtype: 'response', msgid });
} catch (err) { } catch (err) {
process.send({ msgtype: 'response', msgid, errorMessage: err.message }); process.send({ msgtype: 'response', msgid, errorMessage: err.message });