| #include "rocketmq/CredentialsProvider.h" |
| #include "MixAll.h" |
| #include "ghc/filesystem.hpp" |
| #include "google/protobuf/struct.pb.h" |
| #include "google/protobuf/util/json_util.h" |
| #include "gtest/gtest.h" |
| #include <cstdlib> |
| #include <fstream> |
| #include <iostream> |
| |
| ROCKETMQ_NAMESPACE_BEGIN |
| |
| class CredentialsProviderTest : public testing::Test {}; |
| |
| TEST_F(CredentialsProviderTest, testStaticCredentialsProvider) { |
| std::string access_key("abc"); |
| std::string access_secret("def"); |
| StaticCredentialsProvider credentials_provider(access_key, access_secret); |
| Credentials&& credentials = credentials_provider.getCredentials(); |
| ASSERT_EQ(credentials.accessKey(), access_key); |
| ASSERT_EQ(credentials.accessSecret(), access_secret); |
| } |
| |
| TEST_F(CredentialsProviderTest, testEnvironmentVariable) { |
| const char* access_key = "abc"; |
| const char* access_secret = "def"; |
| |
| #ifdef _WIN32 |
| std::string env_access_key; |
| env_access_key.append(EnvironmentVariablesCredentialsProvider::ENVIRONMENT_ACCESS_KEY); |
| env_access_key.push_back('='); |
| env_access_key.append(access_key); |
| |
| _putenv(env_access_key.c_str()); |
| |
| std::string env_access_secret; |
| env_access_secret.append(EnvironmentVariablesCredentialsProvider::ENVIRONMENT_ACCESS_SECRET); |
| env_access_secret.push_back('='); |
| env_access_secret.append(access_secret); |
| _putenv(env_access_secret.c_str()); |
| #else |
| setenv(EnvironmentVariablesCredentialsProvider::ENVIRONMENT_ACCESS_KEY, access_key, 1); |
| setenv(EnvironmentVariablesCredentialsProvider::ENVIRONMENT_ACCESS_SECRET, access_secret, 1); |
| #endif |
| |
| EnvironmentVariablesCredentialsProvider provider; |
| const Credentials& credentials = provider.getCredentials(); |
| EXPECT_STREQ(access_key, credentials.accessKey().c_str()); |
| EXPECT_STREQ(access_secret, credentials.accessSecret().c_str()); |
| } |
| |
| ROCKETMQ_NAMESPACE_END |