-
Notifications
You must be signed in to change notification settings - Fork 4
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Secrets manager integration test (#127)
* create secrets in java util
* simple sm test
* simplify setup
* fix integration
* comment out toxiporxy
* comment out toxiporxy
* sm test not inherit from base anymore
* fix build
* fix set error message, fix dsn not picked up
* add {} in connection string
* fix curly braces
* fix log
* fix build
* more logging
* more logging
* fix test
* better error handling
* uncomment
* address comments
* revert- Loading branch information
1 parent
b5ae0e5
commit 4d240ad
Showing
10 changed files
with
250 additions
and
25 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,152 @@ | ||
| // Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. | ||
| // | ||
| // This program is free software; you can redistribute it and/or modify | ||
| // it under the terms of the GNU General Public License, version 2.0 | ||
| // (GPLv2), as published by the Free Software Foundation, with the | ||
| // following additional permissions: | ||
| // | ||
| // This program is distributed with certain software that is licensed | ||
| // under separate terms, as designated in a particular file or component | ||
| // or in the license documentation. Without limiting your rights under | ||
| // the GPLv2, the authors of this program hereby grant you an additional | ||
| // permission to link the program and your derivative works with the | ||
| // separately licensed software that they have included with the program. | ||
| // | ||
| // Without limiting the foregoing grant of rights under the GPLv2 and | ||
| // additional permission as to separately licensed software, this | ||
| // program is also subject to the Universal FOSS Exception, version 1.0, | ||
| // a copy of which can be found along with its FAQ at | ||
| // http://oss.oracle.com/licenses/universal-foss-exception. | ||
| // | ||
| // This program is distributed in the hope that it will be useful, but | ||
| // WITHOUT ANY WARRANTY; without even the implied warranty of | ||
| // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. | ||
| // See the GNU General Public License, version 2.0, for more details. | ||
| // | ||
| // You should have received a copy of the GNU General Public License | ||
| // along with this program. If not, see | ||
| // http://www.gnu.org/licenses/gpl-2.0.html. | ||
|
|
||
| #include <gtest/gtest.h> | ||
| #include <sql.h> | ||
| #include <sqlext.h> | ||
|
|
||
| #include <cassert> | ||
| #include <chrono> | ||
| #include <climits> | ||
| #include <cstdlib> | ||
| #include <iostream> | ||
| #include <random> | ||
| #include <stdexcept> | ||
|
|
||
| #include "connection_string_builder.cc" | ||
|
|
||
| #define MAX_NAME_LEN 255 | ||
|
|
||
| #define AS_SQLCHAR(str) const_cast<SQLCHAR*>(reinterpret_cast<const SQLCHAR*>(str)) | ||
|
|
||
| static int str_to_int(const char* str) { | ||
| const long int x = strtol(str, nullptr, 10); | ||
| assert(x <= INT_MAX); | ||
| assert(x >= INT_MIN); | ||
| return static_cast<int>(x); | ||
| } | ||
|
|
||
| class SecretsManagerIntegrationTest : public testing::Test { | ||
| protected: | ||
| std::string SECRETS_ARN = std::getenv("SECRETS_ARN"); | ||
| char* dsn = std::getenv("TEST_DSN"); | ||
|
|
||
| int MYSQL_PORT = str_to_int(std::getenv("MYSQL_PORT")); | ||
|
|
||
| std::string MYSQL_CLUSTER_URL = std::getenv("TEST_SERVER"); | ||
|
|
||
| SQLHENV env = nullptr; | ||
| SQLHDBC dbc = nullptr; | ||
|
|
||
| ConnectionStringBuilder builder; | ||
| std::string connection_string; | ||
|
|
||
| static void SetUpTestSuite() { | ||
| } | ||
|
|
||
| static void TearDownTestSuite() { | ||
| } | ||
|
|
||
| void SetUp() override { | ||
| SQLAllocHandle(SQL_HANDLE_ENV, nullptr, &env); | ||
| SQLSetEnvAttr(env, SQL_ATTR_ODBC_VERSION, reinterpret_cast<SQLPOINTER>(SQL_OV_ODBC3), 0); | ||
| SQLAllocHandle(SQL_HANDLE_DBC, env, &dbc); | ||
|
|
||
| builder = ConnectionStringBuilder(); | ||
| builder.withPort(MYSQL_PORT).withLogQuery(true); | ||
| } | ||
|
|
||
| void TearDown() override { | ||
| if (nullptr != dbc) { | ||
| SQLFreeHandle(SQL_HANDLE_DBC, dbc); | ||
| } | ||
| if (nullptr != env) { | ||
| SQLFreeHandle(SQL_HANDLE_ENV, env); | ||
| } | ||
| } | ||
| }; | ||
|
|
||
| TEST_F(SecretsManagerIntegrationTest, EnableSecretsManager) { | ||
| connection_string = builder | ||
| .withDSN(dsn) | ||
| .withServer(MYSQL_CLUSTER_URL) | ||
| .withAuthMode("SECRETS MANAGER") | ||
| .withAuthRegion("us-east-2") | ||
| .withSecretId(SECRETS_ARN) | ||
| .build(); | ||
| SQLCHAR conn_out[4096] = "\0"; | ||
| SQLSMALLINT len; | ||
|
|
||
| EXPECT_EQ(SQL_SUCCESS, SQLDriverConnect(dbc, nullptr, AS_SQLCHAR(connection_string.c_str()), SQL_NTS, conn_out, MAX_NAME_LEN, &len, SQL_DRIVER_NOPROMPT)); | ||
| EXPECT_EQ(SQL_SUCCESS, SQLDisconnect(dbc)); | ||
| } | ||
|
|
||
| TEST_F(SecretsManagerIntegrationTest, EnableSecretsManagerWrongRegion) { | ||
| connection_string = builder | ||
| .withDSN(dsn) | ||
| .withServer(MYSQL_CLUSTER_URL) | ||
| .withAuthMode("SECRETS MANAGER") | ||
| .withAuthRegion("us-east-1") | ||
| .withSecretId(SECRETS_ARN) | ||
| .build(); | ||
| SQLCHAR conn_out[4096] = "\0"; | ||
| SQLSMALLINT len; | ||
|
|
||
| EXPECT_EQ(SQL_ERROR, SQLDriverConnect(dbc, nullptr, AS_SQLCHAR(connection_string.c_str()), SQL_NTS, conn_out, MAX_NAME_LEN, &len, SQL_DRIVER_NOPROMPT)); | ||
|
|
||
| // Check state | ||
| SQLCHAR sqlstate[6] = "\0", message[SQL_MAX_MESSAGE_LENGTH] = "\0";; | ||
| SQLINTEGER native_error = 0; | ||
| SQLSMALLINT stmt_length; | ||
| EXPECT_EQ(SQL_SUCCESS, SQLError(env, dbc, nullptr, sqlstate, &native_error, message, SQL_MAX_MESSAGE_LENGTH - 1, &stmt_length)); | ||
| const std::string state = reinterpret_cast<char*>(sqlstate); | ||
| EXPECT_EQ("HY000", state); | ||
| } | ||
|
|
||
| TEST_F(SecretsManagerIntegrationTest, EnableSecretsManagerInvalidSecretID) { | ||
| connection_string = builder | ||
| .withDSN(dsn) | ||
| .withServer(MYSQL_CLUSTER_URL) | ||
| .withAuthMode("SECRETS MANAGER") | ||
| .withAuthRegion("us-east-2") | ||
| .withSecretId("invalid-id") | ||
| .build(); | ||
| SQLCHAR conn_out[4096] = "\0"; | ||
| SQLSMALLINT len; | ||
|
|
||
| EXPECT_EQ(SQL_ERROR, SQLDriverConnect(dbc, nullptr, AS_SQLCHAR(connection_string.c_str()), SQL_NTS, conn_out, MAX_NAME_LEN, &len, SQL_DRIVER_NOPROMPT)); | ||
|
|
||
| // Check state | ||
| SQLCHAR sqlstate[6] = "\0", message[SQL_MAX_MESSAGE_LENGTH] = "\0";; | ||
| SQLINTEGER native_error = 0; | ||
| SQLSMALLINT stmt_length; | ||
| EXPECT_EQ(SQL_SUCCESS, SQLError(env, dbc, nullptr, sqlstate, &native_error, message, SQL_MAX_MESSAGE_LENGTH - 1, &stmt_length)); | ||
| const std::string state = reinterpret_cast<char*>(sqlstate); | ||
| EXPECT_EQ("HY000", state); | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.