Skip to content

Commit

Permalink
Merge branch 'master' into SNOW-1825789-secure-token-cache
Browse files Browse the repository at this point in the history
  • Loading branch information
@sfc-gh-astachowski
sfc-gh-astachowski authored Mar 7, 2025
2 parents 86fd21b + 556e783 commit 3514cd7
Show file tree
Hide file tree
Showing 2 changed files with 258 additions and 1 deletion.
6 changes: 5 additions & 1 deletion system_test/connectionOptions.js
View file
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ let snowflakeTestProxyPort = process.env.SNOWFLAKE_TEST_PROXY_PORT;
const snowflakeTestProxyProtocol = process.env.SNOWFLAKE_TEST_PROXY_PROTOCOL;
const snowflakeTestProxyUser = process.env.SNOWFLAKE_TEST_PROXY_USER;
const snowflakeTestProxyPassword = process.env.SNOWFLAKE_TEST_PROXY_PASSWORD;
const snowflakeTestAccount = process.env.SNOWFLAKE_TEST_ACCOUNT;
let snowflakeTestAccount = process.env.SNOWFLAKE_TEST_ACCOUNT;
const snowflakeTestUser = process.env.SNOWFLAKE_TEST_USER;
const snowflakeTestDatabase = process.env.SNOWFLAKE_TEST_DATABASE;
const snowflakeTestWarehouse = process.env.SNOWFLAKE_TEST_WAREHOUSE;
Expand All @@ -30,6 +30,10 @@ if (snowflakeTestHost === undefined) {
snowflakeTestHost = snowflakeTestAccount + '.snowflakecomputing.com';
}

if (snowflakeTestHost.endsWith('reg.local')) {
snowflakeTestAccount = snowflakeTestHost.split('.')[0];
}

if (snowflakeTestPort === undefined) {
snowflakeTestPort = '443';
}
Expand Down
253 changes: 253 additions & 0 deletions test/integration/testBind.js
View file
Original file line number Diff line number Diff line change
Expand Up @@ -605,3 +605,256 @@ describe('Test Bind Varible', function () {
);
});
});

describe('Verify stage binding and array binding', () => {

function generateDate(year, month, day, hour, minute, second, millis) {
const date = new Date();
date.setFullYear(year, month - 1, day);
date.setHours(hour);
date.setMinutes(minute);
date.setSeconds(second);
date.setMilliseconds(millis);

return date;
}

function validateTimeValues(actual, expected) {
assert.strictEqual(actual.length, expected.length);

for (let i = 0; i < actual.length; i++) {
assert.strictEqual(Object.keys(actual[i]).length, Object.keys(expected[i]).length);
Object.keys(actual[i]).forEach((key) => key === 'C1' ? assert.strictEqual(actual[i][key], expected[i][key]) : assert.strictEqual(actual[i][key].toJSON(), expected[i][key]));
}
}

let connection;
const currentTimeZone = process.env.TZ;
process.env.TZ = 'UTC';
const columns = '(c1 int, c2 timestamp_ntz , c3 timestamp_ltz, c4 timestamp_tz, c5 date);';
const arrayBindingTable = 'arrBinding';
const stageBindingTable = 'stageBinding';
const alterTimeZoneQuery = (timeZone) => `alter session set TIMEZONE = '${timeZone}'`;
const getCreateTableQuery = (tableName) => `Create or Replace table ${tableName} ${columns}`;
const getInsertQuery = (tableName) => `insert into ${tableName} values (?, ?, ?, ?, ?)`;
const selectQuery = (tableName) => `SELECT * FROM ${tableName} order by c1`;

const timeTestingValues = [
generateDate(1, 1, 1, 23, 24, 25, 987),
generateDate(9999, 12, 30, 23, 24, 25, 987),
generateDate(2025, 2, 18, 11, 37, 25, 326),
//SummerTime
generateDate(2024, 6, 22, 23, 37, 25, 520),
new Date(2000, 0, 1)
];

const binding = [];
timeTestingValues.forEach((value, i) => binding.push([i, value, value, value, value]));

const testCases = [
{
timeZone: 'UTC',
expected:
[
{
C1: 0,
C2: '0001-01-01 23:24:25.987',
//Javascript Timezone limitation
C3: '0001-01-01 23:24:25.987 +0000',
C4: '0001-01-01 23:24:25.987 +0000',
C5: '0001-01-01'
},
{
C1: 1,
C2: '9999-12-30 23:24:25.987',
C3: '9999-12-30 23:24:25.987 +0000',
C4: '9999-12-30 23:24:25.987 +0000',
C5: '9999-12-30'

},
{
C1: 2,
C2: '2025-02-18 11:37:25.326',
C3: '2025-02-18 11:37:25.326 +0000',
C4: '2025-02-18 11:37:25.326 +0000',
C5: '2025-02-18',
},
{
C1: 3,
C2: '2024-06-22 23:37:25.520',
C3: '2024-06-22 23:37:25.520 +0000',
C4: '2024-06-22 23:37:25.520 +0000',
C5: '2024-06-22',
},
{
C1: 4,
C2: '2000-01-01 00:00:00.000',
C3: '2000-01-01 00:00:00.000 +0000',
C4: '2000-01-01 00:00:00.000 +0000',
C5: '2000-01-01',
},
],
},
{
timeZone: 'Europe/Warsaw',
expected:
[
{
C1: 0,
C2: '0001-01-01 23:24:25.987',
//Javascript Timezone limitation
C3: '0001-01-02 00:48:25.987 +0124',
C4: '0001-01-01 23:24:25.987 +0000',
C5: '0001-01-01',
},
{
C1: 1,
C2: '9999-12-30 23:24:25.987',
C3: '9999-12-31 00:24:25.987 +0100',
C4: '9999-12-30 23:24:25.987 +0000',
C5: '9999-12-30',
},
{
C1: 2,
C2: '2025-02-18 11:37:25.326',
C3: '2025-02-18 12:37:25.326 +0100',
C4: '2025-02-18 11:37:25.326 +0000',
C5: '2025-02-18',
},
{
C1: 3,
C2: '2024-06-22 23:37:25.520',
C3: '2024-06-23 01:37:25.520 +0200',
C4: '2024-06-22 23:37:25.520 +0000',
C5: '2024-06-22',
},
{
C1: 4,
C2: '2000-01-01 00:00:00.000',
C3: '2000-01-01 01:00:00.000 +0100',
C4: '2000-01-01 00:00:00.000 +0000',
C5: '2000-01-01',
},
],
},
{
timeZone: 'Asia/Tokyo',
expected:
[
{
C1: 0,
C2: '0001-01-01 23:24:25.987',
//Javascript Timezone limitation
C3: '0001-01-02 08:43:24.987 +0918',
C4: '0001-01-01 23:24:25.987 +0000',
C5: '0001-01-01',
},
{
C1: 1,
C2: '9999-12-30 23:24:25.987',
C3: '9999-12-31 08:24:25.987 +0900',
C4: '9999-12-30 23:24:25.987 +0000',
C5: '9999-12-30',
},
{
C1: 2,
C2: '2025-02-18 11:37:25.326',
C3: '2025-02-18 20:37:25.326 +0900',
C4: '2025-02-18 11:37:25.326 +0000',
C5: '2025-02-18',
},
{
C1: 3,
C2: '2024-06-22 23:37:25.520',
//Tokyo does not have Summer time
C3: '2024-06-23 08:37:25.520 +0900',
C4: '2024-06-22 23:37:25.520 +0000',
C5: '2024-06-22',
},
{
C1: 4,
C2: '2000-01-01 00:00:00.000',
C3: '2000-01-01 09:00:00.000 +0900',
C4: '2000-01-01 00:00:00.000 +0000',
C5: '2000-01-01',
},
]
},
{
timeZone: 'America/Los_Angeles',
expected:
[
{
C1: 0,
C2: '0001-01-01 23:24:25.987',
//Javascript Timezone limitation
C3: '0001-01-01 15:31:27.987 -0752',
C4: '0001-01-01 23:24:25.987 +0000',
C5: '0001-01-01',
},
{
C1: 1,
C2: '9999-12-30 23:24:25.987',
C3: '9999-12-30 15:24:25.987 -0800',
C4: '9999-12-30 23:24:25.987 +0000',
C5: '9999-12-30',
},
{
C1: 2,
C2: '2025-02-18 11:37:25.326',
//Summer Time
C3: '2025-02-18 03:37:25.326 -0800',
C4: '2025-02-18 11:37:25.326 +0000',
C5: '2025-02-18',
},
{
C1: 3,
C2: '2024-06-22 23:37:25.520',
C3: '2024-06-22 16:37:25.520 -0700',
C4: '2024-06-22 23:37:25.520 +0000',
C5: '2024-06-22',
},
{
C1: 4,
C2: '2000-01-01 00:00:00.000',
C3: '1999-12-31 16:00:00.000 -0800',
C4: '2000-01-01 00:00:00.000 +0000',
C5: '2000-01-01',
},
]
},
];

beforeEach(async () => {
connection = testUtil.createConnection();
await testUtil.connectAsync(connection);
await testUtil.executeCmdAsync(connection, getCreateTableQuery(arrayBindingTable));
await testUtil.executeCmdAsync(connection, getCreateTableQuery(stageBindingTable));
});

afterEach(async () => {
await testUtil.dropTablesIgnoringErrorsAsync(connection, arrayBindingTable);
await testUtil.dropTablesIgnoringErrorsAsync(connection, stageBindingTable);
await testUtil.destroyConnectionAsync(connection);
});

after(() => {
process.env.TZ = currentTimeZone;
});

testCases.forEach(({ timeZone, expected }) => {
it(`test binding values with timezone ${timeZone}`, async () => {
await testUtil.executeCmdAsync(connection, alterTimeZoneQuery(timeZone));
await testUtil.executeCmdAsync(connection, getInsertQuery(arrayBindingTable), binding);
await testUtil.destroyConnectionAsync(connection, arrayBindingTable);
connection = testUtil.createConnection({ arrayBindingThreshold: 3 });
await testUtil.connectAsync(connection);
await testUtil.executeCmdAsync(connection, alterTimeZoneQuery(timeZone));
await testUtil.executeCmdAsync(connection, getInsertQuery(stageBindingTable), binding);
let rows = await testUtil.executeCmdAsync(connection, selectQuery(arrayBindingTable));
validateTimeValues(rows, expected);
rows = await testUtil.executeCmdAsync(connection, selectQuery(stageBindingTable));
validateTimeValues(rows, expected);
});
});
});

0 comments on commit 3514cd7

Please sign in to comment.