/* * This file is a part of the CaosDB Project. * * Copyright (C) 2021 Timm Fitschen <t.fitschen@indiscale.com> * Copyright (C) 2021 IndiScale GmbH <info@indiscale.com> * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU Affero General Public License as * published by the Free Software Foundation, either version 3 of the * License, or (at your option) any later version. * * 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 Affero General Public License for more details. * * You should have received a copy of the GNU Affero General Public License * along with this program. If not, see <https://www.gnu.org/licenses/>. * */ #include <gtest/gtest-message.h> // for Message #include <gtest/gtest-test-part.h> // for SuiteApiResolver, TestPartResult #include <memory> // for make_shared, allocator, shared_ptr #include <string> // for string #include "caosdb/authentication.h" // for PlainPasswordAuthenticator #include "caosdb/connection.h" // for InsecureCaosDBConnectionConfig #include "caosdb/exceptions.h" // for AuthenticationError, ConnectionError #include "caosdb/info.h" // for VersionInfo #include "caosdb/utils.h" // for get_env_var #include "gtest/gtest_pred_impl.h" // for Test, TEST, EXPECT_EQ, EXPECT_THROW #include "caosdb_test_utility.h" namespace caosdb::connection { using caosdb::authentication::PlainPasswordAuthenticator; /** * Return a fresh CaosDBConnection pointer. The caller has the ownership. */ inline auto create_test_connection() -> CaosDBConnection * { auto port_str = caosdb::utils::get_env_var("CAOSDB_SERVER_GRPC_PORT_HTTPS", "8443"); auto port = std::stoi(port_str); const auto host = caosdb::utils::get_env_var("CAOSDB_SERVER_HOST", "localhost"); const auto path = caosdb::utils::get_env_var("CAOSDB_SERVER_CERT", std::string()); const auto user = caosdb::utils::get_env_var("CAOSDB_USER", "admin"); const auto password = caosdb::utils::get_env_var("CAOSDB_PASSWORD", "caosdb"); auto auth = std::make_shared<PlainPasswordAuthenticator>(user, password); auto ssloptions = std::make_shared<PemFileCACertProvider>(path); auto config = std::make_shared<SslCaosDBConnectionConfig>(host, port, ssloptions, auth); return new CaosDBConnection(config); }; /** * Singleton which holds a single global CaosDBConnection */ class ConnectionProvider { private: std::shared_ptr<CaosDBConnection> connection; ConnectionProvider() : connection(create_test_connection()){}; public: static ConnectionProvider &GetInstance() { static ConnectionProvider instance; return instance; }; inline auto GetConnection() -> std::shared_ptr<CaosDBConnection> & { return this->connection; } ConnectionProvider(ConnectionProvider const &) = delete; void operator=(ConnectionProvider const &) = delete; }; /** * Return a connection for testing purposes. */ inline auto get_test_connection() -> const std::shared_ptr<CaosDBConnection> & { return ConnectionProvider::GetInstance().GetConnection(); }; } // namespace caosdb::connection