system-test/mysql2-connect.ts

// Copyright 2023 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

import t from 'tap';
import {RowDataPacket} from 'mysql2';
import mysql from 'mysql2/promise';
import {
  Connector,
  AuthTypes,
  IpAddressTypes,
} from '@google-cloud/cloud-sql-connector';

interface Now extends RowDataPacket {
  'NOW()': Date;
}

t.test('open connection and run basic mysql commands', async t => {
  const connector = new Connector();
  const clientOpts = await connector.getOptions({
    instanceConnectionName: String(process.env.MYSQL_CONNECTION_NAME),
    ipType: process.env.IP_TYPE || IpAddressTypes.PUBLIC,
  });
  const conn = await mysql.createConnection({
    ...clientOpts,
    user: String(process.env.MYSQL_USER),
    password: String(process.env.MYSQL_PASS),
    database: String(process.env.MYSQL_DB),
  });

  const [result] = await conn.query<Now[]>('SELECT NOW();');
  const [row] = result;
  const returnedDate = row['NOW()'];
  t.ok(returnedDate.getTime(), 'should have valid returned date object');

  await conn.end();
  connector.close();
});

t.test('open IAM connection and run basic mysql commands', async t => {
  const connector = new Connector();
  const clientOpts = await connector.getOptions({
    instanceConnectionName: String(process.env.MYSQL_CONNECTION_NAME),
    ipType: process.env.IP_TYPE || IpAddressTypes.PUBLIC,
    authType: AuthTypes.IAM,
  });
  const conn = await mysql.createConnection({
    ...clientOpts,
    user: String(process.env.MYSQL_IAM_USER),
    database: String(process.env.MYSQL_DB),
  });

  const [[result]] = await conn.query<Now[]>('SELECT NOW();');
  const returnedDate = result['NOW()'];
  t.ok(returnedDate.getTime(), 'should have valid returned date object');

  await conn.end();
  connector.close();
});

t.test('open connection and run basic mysql commands', async t => {
  const connector = new Connector();
  const clientOpts = await connector.getOptions({
    instanceConnectionName: String(process.env.MYSQL_MCP_CONNECTION_NAME),
    ipType: process.env.IP_TYPE || IpAddressTypes.PUBLIC,
  });
  const conn = await mysql.createConnection({
    ...clientOpts,
    user: String(process.env.MYSQL_USER),
    password: String(process.env.MYSQL_MCP_PASS),
    database: String(process.env.MYSQL_DB),
  });

  const [result] = await conn.query<Now[]>('SELECT NOW();');
  const [row] = result;
  const returnedDate = row['NOW()'];
  t.ok(returnedDate.getTime(), 'should have valid returned date object');

  await conn.end();
  connector.close();
});

t.test('open IAM connection and run basic mysql commands', async t => {
  const connector = new Connector();
  const clientOpts = await connector.getOptions({
    instanceConnectionName: String(process.env.MYSQL_MCP_CONNECTION_NAME),
    ipType: process.env.IP_TYPE || IpAddressTypes.PUBLIC,
    authType: AuthTypes.IAM,
  });
  const conn = await mysql.createConnection({
    ...clientOpts,
    user: String(process.env.MYSQL_IAM_USER),
    database: String(process.env.MYSQL_DB),
  });

  const [[result]] = await conn.query<Now[]>('SELECT NOW();');
  const returnedDate = result['NOW()'];
  t.ok(returnedDate.getTime(), 'should have valid returned date object');

  await conn.end();
  connector.close();
});