Skip to content
Snippets Groups Projects
Commit 06968a09 authored by Henrik tom Wörden's avatar Henrik tom Wörden
Browse files

Merge branch 'f-files' into 'main' 

F files

See merge request !8
parents bd035ce1 72cfe1d0
No related branches found
No related tags found
1 merge request!8F files
Pipeline #13298 failed
Stage: info
Stage: setup
Stage: build
Stage: test
Hide whitespace changes
Inline Side-by-side
CMakeLists.txt
+ 1
− 0
View file @ 06968a09
...@@ -26,6 +26,7 @@ project(libcaosdb_inttests ...@@ -26,6 +26,7 @@ project(libcaosdb_inttests
set(CMAKE_CXX_STANDARD 17) set(CMAKE_CXX_STANDARD 17)
set(CMAKE_CXX_STANDARD_REQUIRED ON) set(CMAKE_CXX_STANDARD_REQUIRED ON)
set(CMAKE_EXPORT_COMPILE_COMMANDS ON)
# dependency management with conan # dependency management with conan
......
README.md
+ 8
− 2
View file @ 06968a09
...@@ -12,11 +12,17 @@ Integration tests for caosdb-cpplib and the caosdb-server ...@@ -12,11 +12,17 @@ Integration tests for caosdb-cpplib and the caosdb-server
# Run tests # Run tests
## Prerequisite
Create a local conan package from the caosdb-cpplib repository
## Build & Execution
1. `mkdir build && cd build/` 1. `mkdir build && cd build/`
2. `conan install .. -s "compiler.libcxx=libstdc++11"` 2. `conan install .. -s "compiler.libcxx=libstdc++11"`
3. `cmake -B . ..` 3. `cmake -B . ..`
4. `cmake --build .` 4. `cmake --build .`
5. Setup env vars 5. Setup env vars
- CAOSDB_SERVER_GRPC_PORT_HTTP
- CAOSDB_SERVER_HOST
- CAOSDB_SERVER_GRPC_PORT_HTTPS
- CAOSDB_SERVER_CERT
6. Run with `ctest` in the build directory. 6. Run with `ctest` in the build directory.
conanfile.txt
+ 1
− 1
View file @ 06968a09
[requires] [requires]
caosdb/0.0.10 caosdb/0.0.11
gtest/1.11.0 gtest/1.11.0
[generators] [generators]
......
test/CMakeLists.txt
+ 4
− 2
View file @ 06968a09
...@@ -24,6 +24,7 @@ ...@@ -24,6 +24,7 @@
set(test_cases set(test_cases
test_connection test_connection
test_transaction test_transaction
#test_file_transmission TODO add once fixed
test_ccaosdb test_ccaosdb
) )
...@@ -82,7 +83,7 @@ if(LINTING) ...@@ -82,7 +83,7 @@ if(LINTING)
"--warnings-as-errors=*" "--warnings-as-errors=*"
"--fix") "--fix")
set(_CMAKE_CXX_CLANG_TIDY_CHECKS set(_CMAKE_CXX_CLANG_TIDY_CHECKS
"--checks=*,-fuchsia-*,-llvmlibc-*,-cert-err58-cpp,-cppcoreguidelines-avoid-non-const-global-variables,-cppcoreguidelines-owning-memory,-modernize-use-trailing-return-type,-google-readability-avoid-underscore-in-googletest-name,-cppcoreguidelines-avoid-magic-numbers,-readability-magic-numbers,-cppcoreguidelines-avoid-goto,-hicpp-avoid-goto,-readability-function-cognitive-complexity") "--checks=*,-fuchsia-*,-llvmlibc-*,-cert-err58-cpp,-cppcoreguidelines-avoid-non-const-global-variables,-cppcoreguidelines-owning-memory,-modernize-use-trailing-return-type,-google-readability-avoid-underscore-in-googletest-name,-cppcoreguidelines-avoid-magic-numbers,-readability-magic-numbers,-cppcoreguidelines-avoid-goto,-hicpp-avoid-goto,-readability-function-cognitive-complexity,-cppcoreguidelines-non-private-member-variables-in-classes,-misc-non-private-member-variables-in-classes,-cppcoreguidelines-pro-type-vararg,-cppcoreguidelines-pro-bounds-array-to-pointer-decay,-hicpp-no-array-decay")
endif() endif()
else() else()
message(STATUS "LINTING is OFF") message(STATUS "LINTING is OFF")
...@@ -105,7 +106,8 @@ foreach (i RANGE "${len_test_cases}") ...@@ -105,7 +106,8 @@ foreach (i RANGE "${len_test_cases}")
target_link_libraries(${test_case_name} PRIVATE ${CONAN_LIBS_CAOSDB} target_link_libraries(${test_case_name} PRIVATE ${CONAN_LIBS_CAOSDB}
${CONAN_LIBS_GTEST} ${CONAN_LIBS_GRPC} ${CONAN_LIBS_ABSEIL} ${CONAN_LIBS_GTEST} ${CONAN_LIBS_GRPC} ${CONAN_LIBS_ABSEIL}
${CONAN_LIBS_OPENSSL} ${CONAN_LIBS_C-ARES} ${CONAN_LIBS_BZIP2} ${CONAN_LIBS_OPENSSL} ${CONAN_LIBS_C-ARES} ${CONAN_LIBS_BZIP2}
${CONAN_LIBS_PROTOBUF} ${CONAN_LIBS_ZLIB} ${CONAN_LIBS_RE2}) ${CONAN_LIBS_PROTOBUF} ${CONAN_LIBS_ZLIB} ${CONAN_LIBS_RE2}
${CONAN_LIBS_BOOST})
target_include_directories(${test_case_name} target_include_directories(${test_case_name}
PUBLIC ${CONAN_INCLUDE_DIRS} ${CMAKE_CURRENT_SOURCE_DIR}) PUBLIC ${CONAN_INCLUDE_DIRS} ${CMAKE_CURRENT_SOURCE_DIR})
if(LINTING) if(LINTING)
......
test/caosdb_test_utility.h
+ 10
− 10
View file @ 06968a09
...@@ -33,16 +33,16 @@ ...@@ -33,16 +33,16 @@
*/ */
#define EXPECT_THROW_MESSAGE(statement, exeption_type, message) \ #define EXPECT_THROW_MESSAGE(statement, exeption_type, message) \
EXPECT_THROW( \ EXPECT_THROW( \
try { statement; } catch (const exeption_type &e) { \ try { statement; } catch (const exeption_type &e) { \
EXPECT_EQ(std::string(e.what()), message); \ EXPECT_EQ(std::string(e.what()), message); \
throw; \ throw; \
}, \ }, \
exeption_type) exeption_type)
#define ASSERT_THROW_MESSAGE(statement, exeption_type, message) \ #define ASSERT_THROW_MESSAGE(statement, exeption_type, message) \
ASSERT_THROW( \ ASSERT_THROW( \
try { statement; } catch (const exeption_type &e) { \ try { statement; } catch (const exeption_type &e) { \
ASSERT_EQ(std::string(e.what()), message); \ ASSERT_EQ(std::string(e.what()), message); \
throw; \ throw; \
}, \ }, \
exeption_type) exeption_type)
#endif #endif
test/test_ccaosdb.cpp
+ 177
− 1
View file @ 06968a09
...@@ -2,6 +2,7 @@ ...@@ -2,6 +2,7 @@
* This file is a part of the CaosDB Project. * This file is a part of the CaosDB Project.
* *
* Copyright (C) 2021 Timm Fitschen <t.fitschen@indiscale.com> * Copyright (C) 2021 Timm Fitschen <t.fitschen@indiscale.com>
* Copyright (C) 2021 Florian Spreckelsen <f.spreckelsen@indiscale.com>
* Copyright (C) 2021 IndiScale GmbH <info@indiscale.com> * Copyright (C) 2021 IndiScale GmbH <info@indiscale.com>
* *
* This program is free software: you can redistribute it and/or modify * This program is free software: you can redistribute it and/or modify
...@@ -17,6 +18,8 @@ ...@@ -17,6 +18,8 @@
* You should have received a copy of the GNU Affero General Public License * 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/>. * along with this program. If not, see <https://www.gnu.org/licenses/>.
*/ */
#include "caosdb/message_code.h" // for MessageCode
#include "caosdb/status_code.h" // for StatusCode
#include "ccaosdb.h" // for caosdb_info_version_info, caosdb_... #include "ccaosdb.h" // for caosdb_info_version_info, caosdb_...
#include "gtest/gtest-message.h" // for Message #include "gtest/gtest-message.h" // for Message
#include "gtest/gtest-test-part.h" // for TestPartResult #include "gtest/gtest-test-part.h" // for TestPartResult
...@@ -34,9 +37,182 @@ TEST(test_ccaosdb, connection_ssl_authentication_success) { ...@@ -34,9 +37,182 @@ TEST(test_ccaosdb, connection_ssl_authentication_success) {
auto major = caosdb_constants_COMPATIBLE_SERVER_VERSION_MAJOR(); auto major = caosdb_constants_COMPATIBLE_SERVER_VERSION_MAJOR();
auto minor = caosdb_constants_COMPATIBLE_SERVER_VERSION_MINOR(); auto minor = caosdb_constants_COMPATIBLE_SERVER_VERSION_MINOR();
const auto *const pre_release = const auto *const pre_release =
caosdb_constants_COMPATIBLE_SERVER_VERSION_PRE_RELEASE(); caosdb_constants_COMPATIBLE_SERVER_VERSION_PRE_RELEASE();
EXPECT_EQ(major, version_info.major); EXPECT_EQ(major, version_info.major);
EXPECT_EQ(minor, version_info.minor); EXPECT_EQ(minor, version_info.minor);
EXPECT_STREQ(pre_release, version_info.pre_release); EXPECT_STREQ(pre_release, version_info.pre_release);
} }
TEST(test_ccaosdb, test_count_query) {
caosdb_connection_connection connection;
int return_code(
caosdb_connection_connection_manager_get_default_connection(&connection));
EXPECT_EQ(return_code, 0);
caosdb_transaction_transaction transaction;
return_code =
caosdb_connection_connection_create_transaction(&connection, &transaction);
EXPECT_EQ(return_code, 0);
// COUNT query with an empty result
return_code = caosdb_transaction_transaction_query(&transaction,
"COUNT ENTITY WITH id=-1");
EXPECT_EQ(return_code, caosdb::StatusCode::GO_ON);
return_code = caosdb_transaction_transaction_execute(&transaction);
EXPECT_EQ(return_code, 0);
long dummy(-1); // NOLINT
long *count = &dummy; // NOLINT
return_code =
caosdb_transaction_transaction_get_count_result(&transaction, count);
EXPECT_EQ(return_code, 0);
EXPECT_EQ(*count, 0);
return_code = caosdb_transaction_delete_transaction(&transaction);
EXPECT_EQ(return_code, 0);
}
TEST(test_ccaosdb, test_query) {
caosdb_connection_connection connection;
int return_code(
caosdb_connection_connection_manager_get_default_connection(&connection));
EXPECT_EQ(return_code, 0);
caosdb_transaction_transaction transaction;
return_code =
caosdb_connection_connection_create_transaction(&connection, &transaction);
EXPECT_EQ(return_code, 0);
return_code = caosdb_transaction_transaction_query(&transaction,
"FIND ENTITY WITH id=-1");
EXPECT_EQ(return_code, caosdb::StatusCode::GO_ON);
return_code = caosdb_transaction_transaction_execute(&transaction);
EXPECT_EQ(return_code, 0);
caosdb_transaction_result_set result_set;
return_code =
caosdb_transaction_transaction_get_result_set(&transaction, &result_set);
EXPECT_EQ(return_code, 0);
int dummy(-1);
int *count = &dummy; // NOLINT
return_code = caosdb_transaction_result_set_size(&result_set, count);
EXPECT_EQ(return_code, 0);
EXPECT_EQ(*count, 0);
return_code = caosdb_transaction_delete_transaction(&transaction);
EXPECT_EQ(return_code, 0);
}
TEST(test_ccaosdb, test_single_id_retrieve) {
caosdb_connection_connection connection;
int return_code(
caosdb_connection_connection_manager_get_default_connection(&connection));
EXPECT_EQ(return_code, 0);
caosdb_transaction_transaction transaction;
return_code =
caosdb_connection_connection_create_transaction(&connection, &transaction);
EXPECT_EQ(return_code, 0);
return_code =
caosdb_transaction_transaction_retrieve_by_id(&transaction, "21");
EXPECT_EQ(return_code, caosdb::StatusCode::GO_ON);
return_code = caosdb_transaction_transaction_execute(&transaction);
EXPECT_EQ(return_code, 0);
caosdb_transaction_result_set result_set;
return_code =
caosdb_transaction_transaction_get_result_set(&transaction, &result_set);
EXPECT_EQ(return_code, 0);
int dummy(-1);
int *count = &dummy; // NOLINT
return_code = caosdb_transaction_result_set_size(&result_set, count);
EXPECT_EQ(return_code, 0);
EXPECT_EQ(*count, 1);
caosdb_entity_entity entity;
return_code = caosdb_transaction_result_set_at(&result_set, &entity, 0);
EXPECT_EQ(return_code, 0);
return_code = caosdb_entity_entity_get_errors_size(&entity, count);
EXPECT_EQ(return_code, 0);
EXPECT_EQ(*count, 0);
char out[255] = {"255"}; // NOLINT
return_code = caosdb_entity_entity_get_name(&entity, out);
EXPECT_EQ(return_code, 0);
EXPECT_STREQ(out, "unit");
return_code = caosdb_transaction_delete_transaction(&transaction);
EXPECT_EQ(return_code, 0);
}
TEST(test_ccaosdb, test_multi_id_retrieve) {
caosdb_connection_connection connection;
int return_code(
caosdb_connection_connection_manager_get_default_connection(&connection));
EXPECT_EQ(return_code, 0);
caosdb_transaction_transaction transaction;
return_code =
caosdb_connection_connection_create_transaction(&connection, &transaction);
EXPECT_EQ(return_code, 0);
const char *ids[] = {"20", "21", "22"}; // NOLINT
return_code =
caosdb_transaction_transaction_retrieve_by_ids(&transaction, ids, 3);
EXPECT_EQ(return_code, caosdb::StatusCode::GO_ON);
return_code = caosdb_transaction_transaction_execute(&transaction);
// Should have an error since entity 22 doesn't exist
EXPECT_TRUE((return_code > 0));
caosdb_transaction_result_set result_set;
return_code =
caosdb_transaction_transaction_get_result_set(&transaction, &result_set);
EXPECT_EQ(return_code, 0);
int dummy(-1);
int *count = &dummy; // NOLINT
return_code = caosdb_transaction_result_set_size(&result_set, count);
EXPECT_EQ(return_code, 0);
EXPECT_EQ(*count, 3);
caosdb_entity_entity entity;
return_code = caosdb_transaction_result_set_at(&result_set, &entity, 1);
EXPECT_EQ(return_code, 0);
return_code = caosdb_entity_entity_get_errors_size(&entity, count);
EXPECT_EQ(return_code, 0);
EXPECT_EQ(*count, 0);
char out[255] = {"255"}; // NOLINT
return_code = caosdb_entity_entity_get_name(&entity, out);
EXPECT_EQ(return_code, 0);
EXPECT_STREQ(out, "unit");
caosdb_entity_entity other_entity;
return_code = caosdb_transaction_result_set_at(&result_set, &other_entity, 2);
EXPECT_EQ(return_code, 0);
return_code = caosdb_entity_entity_get_errors_size(&other_entity, count);
EXPECT_EQ(return_code, 0);
EXPECT_EQ(*count, 1);
caosdb_entity_message err_msg;
return_code = caosdb_entity_entity_get_error(&other_entity, &err_msg, 0);
EXPECT_EQ(return_code, 0);
return_code = caosdb_entity_message_get_code(&err_msg, count);
EXPECT_EQ(return_code, 0);
EXPECT_EQ(*count, caosdb::entity::MessageCode::ENTITY_DOES_NOT_EXIST);
return_code = caosdb_transaction_delete_transaction(&transaction);
EXPECT_EQ(return_code, 0);
}
test/test_connection.cpp
+ 16
− 16
View file @ 06968a09
...@@ -28,9 +28,9 @@ ...@@ -28,9 +28,9 @@
#include "caosdb/info.h" // for VersionInfo #include "caosdb/info.h" // for VersionInfo
#include "caosdb/utility.h" // for get_env_var #include "caosdb/utility.h" // for get_env_var
#include "caosdb_test_utility.h" // for EXPECT_THROW_MESSAGE #include "caosdb_test_utility.h" // for EXPECT_THROW_MESSAGE
#include <gtest/gtest-message.h> // for Message #include <gtest/gtest-message.h> // NOLINT TODO ?? for Message
#include <gtest/gtest-test-part.h> // for SuiteApiResolver, TestPartR... #include <gtest/gtest-test-part.h> // NOLINT for SuiteApiResolver, TestPartR...
#include "gtest/gtest_pred_impl.h" // for Test, TestInfo, TEST, EXPEC... #include "gtest/gtest_pred_impl.h" // NOLINT for Test, TestInfo, TEST, EXPEC...
#include <memory> // for allocator, unique_ptr, __sh... #include <memory> // for allocator, unique_ptr, __sh...
#include <string> // for stoi, string #include <string> // for stoi, string
...@@ -62,10 +62,10 @@ TEST(test_connection, connect_somehost_42347_fails) { ...@@ -62,10 +62,10 @@ TEST(test_connection, connect_somehost_42347_fails) {
TEST(test_connection, connection_insecure_authentication_error_anonymous) { TEST(test_connection, connection_insecure_authentication_error_anonymous) {
const auto *port_str = const auto *port_str =
caosdb::utility::get_env_var("CAOSDB_SERVER_GRPC_PORT_HTTP", "8080"); caosdb::utility::get_env_var("CAOSDB_SERVER_GRPC_PORT_HTTP", "8080");
auto port = std::stoi(port_str); auto port = std::stoi(port_str);
const auto *const host = const auto *const host =
caosdb::utility::get_env_var("CAOSDB_SERVER_HOST", "localhost"); caosdb::utility::get_env_var("CAOSDB_SERVER_HOST", "localhost");
auto config = InsecureConnectionConfiguration(host, port); auto config = InsecureConnectionConfiguration(host, port);
auto connection = Connection(config); auto connection = Connection(config);
...@@ -75,12 +75,12 @@ TEST(test_connection, connection_insecure_authentication_error_anonymous) { ...@@ -75,12 +75,12 @@ TEST(test_connection, connection_insecure_authentication_error_anonymous) {
TEST(test_connection, connection_ssl_authentication_error_anonymous) { TEST(test_connection, connection_ssl_authentication_error_anonymous) {
const auto *port_str = const auto *port_str =
caosdb::utility::get_env_var("CAOSDB_SERVER_GRPC_PORT_HTTPS", "8443"); caosdb::utility::get_env_var("CAOSDB_SERVER_GRPC_PORT_HTTPS", "8443");
auto port = std::stoi(port_str); auto port = std::stoi(port_str);
const auto *const host = const auto *const host =
caosdb::utility::get_env_var("CAOSDB_SERVER_HOST", "localhost"); caosdb::utility::get_env_var("CAOSDB_SERVER_HOST", "localhost");
const auto path = const auto path =
caosdb::utility::get_env_var("CAOSDB_SERVER_CERT", std::string()); caosdb::utility::get_env_var("CAOSDB_SERVER_CERT", std::string());
auto cert = PemFileCertificateProvider(path); auto cert = PemFileCertificateProvider(path);
auto config = TlsConnectionConfiguration(host, port, cert); auto config = TlsConnectionConfiguration(host, port, cert);
...@@ -94,12 +94,12 @@ TEST(test_connection, connection_ssl_authentication_error_anonymous) { ...@@ -94,12 +94,12 @@ TEST(test_connection, connection_ssl_authentication_error_anonymous) {
TEST(test_connection, connection_ssl_authentication_error_wrong_credentials) { TEST(test_connection, connection_ssl_authentication_error_wrong_credentials) {
const auto *port_str = const auto *port_str =
caosdb::utility::get_env_var("CAOSDB_SERVER_GRPC_PORT_HTTPS", "8443"); caosdb::utility::get_env_var("CAOSDB_SERVER_GRPC_PORT_HTTPS", "8443");
auto port = std::stoi(port_str); auto port = std::stoi(port_str);
const auto *const host = const auto *const host =
caosdb::utility::get_env_var("CAOSDB_SERVER_HOST", "localhost"); caosdb::utility::get_env_var("CAOSDB_SERVER_HOST", "localhost");
const auto path = const auto path =
caosdb::utility::get_env_var("CAOSDB_SERVER_CERT", std::string()); caosdb::utility::get_env_var("CAOSDB_SERVER_CERT", std::string());
const auto *const user = "not-a-user-23461237"; const auto *const user = "not-a-user-23461237";
const auto *const password = "very-c-cred"; const auto *const password = "very-c-cred";
...@@ -109,10 +109,10 @@ TEST(test_connection, connection_ssl_authentication_error_wrong_credentials) { ...@@ -109,10 +109,10 @@ TEST(test_connection, connection_ssl_authentication_error_wrong_credentials) {
auto connection = Connection(config); auto connection = Connection(config);
EXPECT_THROW_MESSAGE( EXPECT_THROW_MESSAGE(
connection.RetrieveVersionInfo(), AuthenticationError, connection.RetrieveVersionInfo(), AuthenticationError,
"The attempt to execute this transaction has not been executed at all " "The attempt to execute this transaction has not been executed at all "
"because the authentication did not succeed. Original error: " "because the authentication did not succeed. Original error: "
"Authentication failed. Username or password wrong."); "Authentication failed. Username or password wrong.");
} }
TEST(test_connection, connection_ssl_authentication_success) { TEST(test_connection, connection_ssl_authentication_success) {
...@@ -121,7 +121,7 @@ TEST(test_connection, connection_ssl_authentication_success) { ...@@ -121,7 +121,7 @@ TEST(test_connection, connection_ssl_authentication_success) {
auto major = caosdb::COMPATIBLE_SERVER_VERSION_MAJOR; auto major = caosdb::COMPATIBLE_SERVER_VERSION_MAJOR;
auto minor = caosdb::COMPATIBLE_SERVER_VERSION_MINOR; auto minor = caosdb::COMPATIBLE_SERVER_VERSION_MINOR;
const auto pre_release = const auto pre_release =
std::string(caosdb::COMPATIBLE_SERVER_VERSION_PRE_RELEASE); std::string(caosdb::COMPATIBLE_SERVER_VERSION_PRE_RELEASE);
const auto &v_info = connection->RetrieveVersionInfo(); const auto &v_info = connection->RetrieveVersionInfo();
EXPECT_EQ(major, v_info.GetMajor()); EXPECT_EQ(major, v_info.GetMajor());
......
test/test_file_transmission.cpp 0 → 100644
+ 164
− 0
View file @ 06968a09
/*
* 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 "caosdb/connection.h" // for Connection, ConnectionManager
#include "caosdb/entity.h"
#include "caosdb/entity/v1alpha1/main.pb.h"
#include "caosdb/transaction.h" // for Transaction, UniqueRe...
#include "gtest/gtest-message.h" // for Message
#include "gtest/gtest-test-part.h" // for TestPartResult, SuiteApiResolver
#include "gtest/gtest_pred_impl.h" // for Test, EXPECT_EQ, AssertionResult
#include <memory> // for unique_ptr, allocator, __shar...
#include <string> // for string
namespace caosdb::transaction {
using caosdb::entity::Entity;
using caosdb::entity::v1alpha1::FileDownloadResponse;
using caosdb::entity::v1alpha1::FileUploadResponse;
using caosdb::entity::v1alpha1::RegisterFileDownloadResponse;
using caosdb::entity::v1alpha1::RegisterFileUploadResponse;
using caosdb::entity::v1alpha1::RegistrationStatus;
using caosdb::entity::v1alpha1::TransmissionStatus;
// TODO(tf) this file is currently not used (see CMakeLists.txt)
// Is it still necessary or is it obsolete due to test_transaction.cpp?
// RegisterFileDownloadResponse is currently not defined by proto or the h
// file.
class test_file_transmission : public ::testing::Test {
protected:
void SetUp() override {}
void TearDown() override {
// TODO(tf): delete all created entities
}
};
TEST_F(test_file_transmission, register_file_upload) {
const auto &connection =
caosdb::connection::ConnectionManager::GetDefaultConnection();
auto transaction(connection->CreateTransaction());
RegisterFileUploadResponse response;
EXPECT_EQ(response.status(),
RegistrationStatus::REGISTRATION_STATUS_UNSPECIFIED);
transaction->RegisterUploadFile(&response);
EXPECT_EQ(response.status(),
RegistrationStatus::REGISTRATION_STATUS_ACCEPTED);
EXPECT_FALSE(response.registration_id().empty());
}
TEST_F(test_file_transmission, file_upload) {
const auto &connection =
caosdb::connection::ConnectionManager::GetDefaultConnection();
auto transaction(connection->CreateTransaction());
RegisterFileUploadResponse registration_response;
transaction->RegisterUploadFile(&registration_response);
ASSERT_EQ(registration_response.status(),
RegistrationStatus::REGISTRATION_STATUS_ACCEPTED);
auto registration_id = registration_response.registration_id();
FileUploadResponse upload_response;
transaction->UploadFile(&upload_response, registration_id);
EXPECT_EQ(upload_response.status(),
TransmissionStatus::TRANSMISSION_STATUS_GO_ON);
}
TEST_F(test_file_transmission, file_insertion) {
const auto &connection =
caosdb::connection::ConnectionManager::GetDefaultConnection();
auto transaction(connection->CreateTransaction());
RegisterFileUploadResponse registration_response;
transaction->RegisterUploadFile(&registration_response);
ASSERT_EQ(registration_response.status(),
RegistrationStatus::REGISTRATION_STATUS_ACCEPTED);
auto registration_id = registration_response.registration_id();
FileUploadResponse upload_response;
transaction->UploadFile(&upload_response, registration_id);
Entity file_entity;
file_entity.SetRole("File");
file_entity.SetFileTransmissionId(registration_id, "test.txt");
file_entity.SetFilePath("test.txt");
transaction->InsertEntity(&file_entity);
transaction->Execute();
auto cleanup_transaction(connection->CreateTransaction());
cleanup_transaction->DeleteById(transaction->GetResultSet().At(0).GetId());
cleanup_transaction->Execute();
}
TEST_F(test_file_transmission, file_download) {
const auto &connection =
caosdb::connection::ConnectionManager::GetDefaultConnection();
auto upload_transaction(connection->CreateTransaction());
RegisterFileUploadResponse upload_registration_response;
upload_transaction->RegisterUploadFile(&upload_registration_response);
ASSERT_EQ(upload_registration_response.status(),
RegistrationStatus::REGISTRATION_STATUS_ACCEPTED);
auto registration_id = upload_registration_response.registration_id();
FileUploadResponse upload_response;
upload_transaction->UploadFile(&upload_response, registration_id);
Entity file_entity;
file_entity.SetRole("File");
file_entity.SetFileTransmissionId(registration_id, "test.txt");
file_entity.SetFilePath("test.txt");
upload_transaction->InsertEntity(&file_entity);
upload_transaction->Execute();
// Download by entity_id
auto download_transaction(connection->CreateTransaction());
RegisterFileDownloadResponse download_registration_response;
RegisterFileDownloadRequest download_registration_request;
download_registration_request.add_files()->set_entity_id(
upload_transaction->GetResultSet().At(0).GetId());
download_transaction->RegisterDownloadFile(download_registration_request,
&download_registration_response);
ASSERT_EQ(download_registration_response.status(),
RegistrationStatus::REGISTRATION_STATUS_ACCEPTED);
FileDownloadResponse download_response;
download_transaction->DownloadFile(&download_response,
download_registration_response);
EXPECT_EQ(download_response.chunk().data(), "this is some data");
// CLEANUP
auto cleanup_transaction(connection->CreateTransaction());
cleanup_transaction->DeleteById(
upload_transaction->GetResultSet().At(0).GetId());
cleanup_transaction->Execute();
}
} // namespace caosdb::transaction
test/test_transaction.cpp
+ 200
− 111
View file @ 06968a09
...@@ -18,18 +18,26 @@ ...@@ -18,18 +18,26 @@
* along with this program. If not, see <https://www.gnu.org/licenses/>. * along with this program. If not, see <https://www.gnu.org/licenses/>.
* *
*/ */
#include "caosdb/connection.h" // for Connection, ConnectionManager #include "caosdb/connection.h" // for Connection, ConnectionManager
#include "caosdb/entity.h" // for Entity, Messages, Message #include "caosdb/entity.h" // for Entity, Messages, Message
#include "caosdb/file_transmission/file_reader.h" // for FileReader
#include "caosdb/file_transmission/file_writer.h" // for FileWriter
#include "caosdb/message_code.h" // for ENTITY_DOES_NOT_EXIST, Messag... #include "caosdb/message_code.h" // for ENTITY_DOES_NOT_EXIST, Messag...
#include "caosdb/status_code.h" // for SUCCESS, StatusCode #include "caosdb/status_code.h" // for SUCCESS, StatusCode
#include "caosdb/transaction.h" // for Entity, Transaction, UniqueRe... #include "caosdb/transaction.h" // for Entity, Transaction,...
#include "caosdb/transaction_status.h" // for TransactionStatus, StatusCode #include "caosdb/transaction_status.h" // for TransactionStatus, StatusCode
#include "gtest/gtest-message.h" // for Message #include <boost/filesystem/operations.hpp> // for remove
#include "gtest/gtest-test-part.h" // for TestPartResult, SuiteApiResolver #include <boost/filesystem/path.hpp> // for path
#include "gtest/gtest_pred_impl.h" // for Test, EXPECT_EQ, AssertionResult #include <boost/filesystem/path_traits.hpp> // for filesystem
#include <memory> // for unique_ptr, allocator, __shar... #include <gtest/gtest-message.h> // for Message
#include <string> // for string #include <gtest/gtest-test-part.h> // for TestPartResult, SuiteApiResolver
#include <vector> // for vector #include <gtest/gtest_pred_impl.h> // for Test, EXPECT_EQ, AssertionResult
#include <iostream>
#include <memory> // for unique_ptr, allocator, __shar...
#include <string> // for string
#include <vector> // for vector
namespace fs = boost::filesystem;
namespace caosdb::transaction { namespace caosdb::transaction {
using caosdb::entity::Entity; using caosdb::entity::Entity;
using caosdb::entity::MessageCode; using caosdb::entity::MessageCode;
...@@ -38,15 +46,30 @@ using caosdb::entity::Property; ...@@ -38,15 +46,30 @@ using caosdb::entity::Property;
class test_transaction : public ::testing::Test { class test_transaction : public ::testing::Test {
protected: protected:
void SetUp() override {} fs::path test_upload_file_1;
fs::path test_download_file_1;
void SetUp() override {
test_upload_file_1 = fs::path("test_upload_file_1_delete_me.dat");
test_download_file_1 = fs::path("test_download_file_1_delete_me.dat");
FileWriter writer(test_upload_file_1);
std::string buffer(1024, 'c');
for (int i = 0; i < 8; i++) {
writer.write(buffer);
}
}
void TearDown() override { void TearDown() override {
fs::remove(test_upload_file_1);
fs::remove(test_download_file_1);
const auto &connection = const auto &connection =
caosdb::connection::ConnectionManager::GetDefaultConnection(); caosdb::connection::ConnectionManager::GetDefaultConnection();
auto query_transaction(connection->CreateTransaction()); auto query_transaction(connection->CreateTransaction());
query_transaction->Query("FIND ENTITY WITH id > 99"); query_transaction->Query("FIND ENTITY WITH id > 99");
query_transaction->Execute(); query_transaction->Execute();
if (query_transaction->GetResultSet().Size() > 0) { if (query_transaction->GetResultSet().size() > 0) {
auto delete_transaction(connection->CreateTransaction()); auto delete_transaction(connection->CreateTransaction());
for (const Entity &entity : query_transaction->GetResultSet()) { for (const Entity &entity : query_transaction->GetResultSet()) {
delete_transaction->DeleteById(entity.GetId()); delete_transaction->DeleteById(entity.GetId());
...@@ -58,7 +81,7 @@ protected: ...@@ -58,7 +81,7 @@ protected:
TEST_F(test_transaction, DISABLED_retrieve_manufacturer_by_id) { TEST_F(test_transaction, DISABLED_retrieve_manufacturer_by_id) {
const auto &connection = const auto &connection =
caosdb::connection::ConnectionManager::GetDefaultConnection(); caosdb::connection::ConnectionManager::GetDefaultConnection();
const auto *id = "107"; const auto *id = "107";
const auto *role = "RecordType"; const auto *role = "RecordType";
...@@ -70,10 +93,9 @@ TEST_F(test_transaction, DISABLED_retrieve_manufacturer_by_id) { ...@@ -70,10 +93,9 @@ TEST_F(test_transaction, DISABLED_retrieve_manufacturer_by_id) {
transaction->RetrieveById(id); transaction->RetrieveById(id);
transaction->Execute(); transaction->Execute();
const auto &result_set = const auto &result_set = transaction->GetResultSet();
dynamic_cast<const UniqueResult &>(transaction->GetResultSet());
const auto &entity = result_set.GetEntity(); const auto &entity = result_set.at(0);
EXPECT_EQ(id, entity.GetId()); EXPECT_EQ(id, entity.GetId());
EXPECT_EQ(name, entity.GetName()); EXPECT_EQ(name, entity.GetName());
EXPECT_EQ(role, entity.GetRole()); EXPECT_EQ(role, entity.GetRole());
...@@ -83,7 +105,7 @@ TEST_F(test_transaction, DISABLED_retrieve_manufacturer_by_id) { ...@@ -83,7 +105,7 @@ TEST_F(test_transaction, DISABLED_retrieve_manufacturer_by_id) {
TEST_F(test_transaction, retrieve_non_existing) { TEST_F(test_transaction, retrieve_non_existing) {
const auto &connection = const auto &connection =
caosdb::connection::ConnectionManager::GetDefaultConnection(); caosdb::connection::ConnectionManager::GetDefaultConnection();
auto transaction(connection->CreateTransaction()); auto transaction(connection->CreateTransaction());
...@@ -95,20 +117,19 @@ TEST_F(test_transaction, retrieve_non_existing) { ...@@ -95,20 +117,19 @@ TEST_F(test_transaction, retrieve_non_existing) {
EXPECT_EQ(status.GetCode(), TransactionStatus::TRANSACTION_ERROR().GetCode()); EXPECT_EQ(status.GetCode(), TransactionStatus::TRANSACTION_ERROR().GetCode());
ASSERT_EQ(status.GetCode(), StatusCode::GENERIC_TRANSACTION_ERROR); ASSERT_EQ(status.GetCode(), StatusCode::GENERIC_TRANSACTION_ERROR);
const auto &result_set = const auto &result_set = transaction->GetResultSet();
dynamic_cast<const UniqueResult &>(transaction->GetResultSet());
const auto &entity = result_set.GetEntity(); const auto &entity = result_set.at(0);
EXPECT_EQ(id, entity.GetId()); EXPECT_EQ(id, entity.GetId());
EXPECT_TRUE(entity.HasErrors()); EXPECT_TRUE(entity.HasErrors());
ASSERT_EQ(entity.GetErrors().Size(), 1); ASSERT_EQ(entity.GetErrors().size(), 1);
EXPECT_EQ(entity.GetErrors().At(0).GetCode(), EXPECT_EQ(entity.GetErrors().at(0).GetCode(),
MessageCode::ENTITY_DOES_NOT_EXIST); MessageCode::ENTITY_DOES_NOT_EXIST);
} }
TEST_F(test_transaction, insert_without_delete) { TEST_F(test_transaction, insert_without_delete) {
const auto &connection = const auto &connection =
caosdb::connection::ConnectionManager::GetDefaultConnection(); caosdb::connection::ConnectionManager::GetDefaultConnection();
auto insert_transaction(connection->CreateTransaction()); auto insert_transaction(connection->CreateTransaction());
...@@ -123,22 +144,21 @@ TEST_F(test_transaction, insert_without_delete) { ...@@ -123,22 +144,21 @@ TEST_F(test_transaction, insert_without_delete) {
ASSERT_TRUE(insert_status.IsTerminated()); ASSERT_TRUE(insert_status.IsTerminated());
ASSERT_FALSE(insert_status.IsError()); ASSERT_FALSE(insert_status.IsError());
const auto &insert_result_set = const auto &insert_result_set = insert_transaction->GetResultSet();
dynamic_cast<const UniqueResult &>(insert_transaction->GetResultSet());
const auto &new_entity = insert_result_set.GetEntity(); const auto &new_entity = insert_result_set.at(0);
EXPECT_FALSE(new_entity.GetId().empty()); EXPECT_FALSE(new_entity.GetId().empty());
EXPECT_FALSE(new_entity.HasErrors()); EXPECT_FALSE(new_entity.HasErrors());
// Should have a warning since it has no properties // Should have a warning since it has no properties
EXPECT_TRUE(new_entity.HasWarnings()); EXPECT_TRUE(new_entity.HasWarnings());
EXPECT_EQ(new_entity.GetWarnings().Size(), 1); EXPECT_EQ(new_entity.GetWarnings().size(), 1);
EXPECT_EQ(new_entity.GetWarnings().At(0).GetCode(), EXPECT_EQ(new_entity.GetWarnings().at(0).GetCode(),
MessageCode::ENTITY_HAS_NO_PROPERTIES); MessageCode::ENTITY_HAS_NO_PROPERTIES);
} }
TEST_F(test_transaction, insert_delete) { TEST_F(test_transaction, insert_delete) {
const auto &connection = const auto &connection =
caosdb::connection::ConnectionManager::GetDefaultConnection(); caosdb::connection::ConnectionManager::GetDefaultConnection();
auto insert_transaction(connection->CreateTransaction()); auto insert_transaction(connection->CreateTransaction());
...@@ -153,16 +173,15 @@ TEST_F(test_transaction, insert_delete) { ...@@ -153,16 +173,15 @@ TEST_F(test_transaction, insert_delete) {
ASSERT_TRUE(insert_status.IsTerminated()); ASSERT_TRUE(insert_status.IsTerminated());
ASSERT_FALSE(insert_status.IsError()); ASSERT_FALSE(insert_status.IsError());
const auto &insert_result_set = const auto &insert_result_set = insert_transaction->GetResultSet();
dynamic_cast<const UniqueResult &>(insert_transaction->GetResultSet());
const auto &new_entity = insert_result_set.GetEntity(); const auto &new_entity = insert_result_set.at(0);
EXPECT_FALSE(new_entity.GetId().empty()); EXPECT_FALSE(new_entity.GetId().empty());
EXPECT_FALSE(new_entity.HasErrors()); EXPECT_FALSE(new_entity.HasErrors());
// Should have a warning since it has no properties // Should have a warning since it has no properties
EXPECT_TRUE(new_entity.HasWarnings()); EXPECT_TRUE(new_entity.HasWarnings());
EXPECT_EQ(new_entity.GetWarnings().Size(), 1); EXPECT_EQ(new_entity.GetWarnings().size(), 1);
EXPECT_EQ(new_entity.GetWarnings().At(0).GetCode(), EXPECT_EQ(new_entity.GetWarnings().at(0).GetCode(),
MessageCode::ENTITY_HAS_NO_PROPERTIES); MessageCode::ENTITY_HAS_NO_PROPERTIES);
auto delete_transaction(connection->CreateTransaction()); auto delete_transaction(connection->CreateTransaction());
...@@ -175,17 +194,16 @@ TEST_F(test_transaction, insert_delete) { ...@@ -175,17 +194,16 @@ TEST_F(test_transaction, insert_delete) {
ASSERT_TRUE(delete_status.IsTerminated()); ASSERT_TRUE(delete_status.IsTerminated());
ASSERT_FALSE(delete_status.IsError()); ASSERT_FALSE(delete_status.IsError());
const auto &delete_result_set = const auto &delete_result_set = delete_transaction->GetResultSet();
dynamic_cast<const UniqueResult &>(delete_transaction->GetResultSet());
const auto &deleted_entity = delete_result_set.GetEntity(); const auto &deleted_entity = delete_result_set.at(0);
EXPECT_EQ(deleted_entity.GetId(), new_entity.GetId()); EXPECT_EQ(deleted_entity.GetId(), new_entity.GetId());
EXPECT_FALSE(deleted_entity.HasErrors()); EXPECT_FALSE(deleted_entity.HasErrors());
} }
TEST_F(test_transaction, insert_delete_with_parent) { TEST_F(test_transaction, insert_delete_with_parent) {
const auto &connection = const auto &connection =
caosdb::connection::ConnectionManager::GetDefaultConnection(); caosdb::connection::ConnectionManager::GetDefaultConnection();
auto insert_transaction(connection->CreateTransaction()); auto insert_transaction(connection->CreateTransaction());
...@@ -200,10 +218,9 @@ TEST_F(test_transaction, insert_delete_with_parent) { ...@@ -200,10 +218,9 @@ TEST_F(test_transaction, insert_delete_with_parent) {
ASSERT_TRUE(insert_status.IsTerminated()); ASSERT_TRUE(insert_status.IsTerminated());
ASSERT_FALSE(insert_status.IsError()); ASSERT_FALSE(insert_status.IsError());
const auto &insert_result_set = const auto &insert_result_set = insert_transaction->GetResultSet();
dynamic_cast<const UniqueResult &>(insert_transaction->GetResultSet());
const auto &inserted_rt = insert_result_set.GetEntity(); const auto &inserted_rt = insert_result_set.at(0);
Entity rec; Entity rec;
rec.SetRole("Record"); rec.SetRole("Record");
...@@ -223,10 +240,9 @@ TEST_F(test_transaction, insert_delete_with_parent) { ...@@ -223,10 +240,9 @@ TEST_F(test_transaction, insert_delete_with_parent) {
ASSERT_TRUE(rec_insert_status.IsTerminated()); ASSERT_TRUE(rec_insert_status.IsTerminated());
ASSERT_FALSE(rec_insert_status.IsError()); ASSERT_FALSE(rec_insert_status.IsError());
const auto &rec_result_set = const auto &rec_result_set = rec_transaction->GetResultSet();
dynamic_cast<const UniqueResult &>(rec_transaction->GetResultSet());
const auto &inserted_rec = rec_result_set.GetEntity(); const auto &inserted_rec = rec_result_set.at(0);
EXPECT_FALSE(inserted_rec.GetId().empty()); EXPECT_FALSE(inserted_rec.GetId().empty());
...@@ -240,14 +256,13 @@ TEST_F(test_transaction, insert_delete_with_parent) { ...@@ -240,14 +256,13 @@ TEST_F(test_transaction, insert_delete_with_parent) {
ASSERT_TRUE(rec_retrieve_status.IsTerminated()); ASSERT_TRUE(rec_retrieve_status.IsTerminated());
ASSERT_FALSE(rec_retrieve_status.IsError()); ASSERT_FALSE(rec_retrieve_status.IsError());
const auto &retrieve_result_set = const auto &retrieve_result_set = retrieve_transaction->GetResultSet();
dynamic_cast<const UniqueResult &>(retrieve_transaction->GetResultSet()); const auto &retrieved_rec = retrieve_result_set.at(0);
const auto &retrieved_rec = retrieve_result_set.GetEntity();
EXPECT_EQ(retrieved_rec.GetName(), rec.GetName()); EXPECT_EQ(retrieved_rec.GetName(), rec.GetName());
EXPECT_EQ(retrieved_rec.GetParents().Size(), 1); EXPECT_EQ(retrieved_rec.GetParents().size(), 1);
EXPECT_EQ(retrieved_rec.GetParents().At(0).GetId(), inserted_rt.GetId()); EXPECT_EQ(retrieved_rec.GetParents().at(0).GetId(), inserted_rt.GetId());
EXPECT_EQ(retrieved_rec.GetParents().At(0).GetName(), rt.GetName()); EXPECT_EQ(retrieved_rec.GetParents().at(0).GetName(), rt.GetName());
auto rec_deletion(connection->CreateTransaction()); auto rec_deletion(connection->CreateTransaction());
...@@ -272,7 +287,7 @@ TEST_F(test_transaction, insert_delete_with_parent) { ...@@ -272,7 +287,7 @@ TEST_F(test_transaction, insert_delete_with_parent) {
TEST_F(test_transaction, insert_delete_with_property) { TEST_F(test_transaction, insert_delete_with_property) {
const auto &connection = const auto &connection =
caosdb::connection::ConnectionManager::GetDefaultConnection(); caosdb::connection::ConnectionManager::GetDefaultConnection();
// Create and insert property // Create and insert property
Entity prop_ent; Entity prop_ent;
...@@ -289,10 +304,9 @@ TEST_F(test_transaction, insert_delete_with_property) { ...@@ -289,10 +304,9 @@ TEST_F(test_transaction, insert_delete_with_property) {
ASSERT_TRUE(prop_insert_status.IsTerminated()); ASSERT_TRUE(prop_insert_status.IsTerminated());
ASSERT_FALSE(prop_insert_status.IsError()); ASSERT_FALSE(prop_insert_status.IsError());
const auto &prop_result_set = const auto &prop_result_set = prop_insertion->GetResultSet();
dynamic_cast<const UniqueResult &>(prop_insertion->GetResultSet());
const auto &inserted_prop = prop_result_set.GetEntity(); const auto &inserted_prop = prop_result_set.at(0);
EXPECT_FALSE(inserted_prop.GetId().empty()); EXPECT_FALSE(inserted_prop.GetId().empty());
// create and insert record type with the above property // create and insert record type with the above property
...@@ -304,6 +318,7 @@ TEST_F(test_transaction, insert_delete_with_property) { ...@@ -304,6 +318,7 @@ TEST_F(test_transaction, insert_delete_with_property) {
Entity rt; Entity rt;
rt.SetRole("RecordType"); rt.SetRole("RecordType");
rt.SetName("TestRT"); rt.SetName("TestRT");
rt.SetDescription("Some description");
rt.AppendProperty(prop_rt); rt.AppendProperty(prop_rt);
auto rt_insertion(connection->CreateTransaction()); auto rt_insertion(connection->CreateTransaction());
...@@ -315,10 +330,9 @@ TEST_F(test_transaction, insert_delete_with_property) { ...@@ -315,10 +330,9 @@ TEST_F(test_transaction, insert_delete_with_property) {
ASSERT_TRUE(rt_insert_status.IsTerminated()); ASSERT_TRUE(rt_insert_status.IsTerminated());
ASSERT_FALSE(rt_insert_status.IsError()); ASSERT_FALSE(rt_insert_status.IsError());
const auto &rt_result_set = const auto &rt_result_set = rt_insertion->GetResultSet();
dynamic_cast<const UniqueResult &>(rt_insertion->GetResultSet());
const auto &inserted_rt = rt_result_set.GetEntity(); const auto &inserted_rt = rt_result_set.at(0);
EXPECT_FALSE(inserted_rt.GetId().empty()); EXPECT_FALSE(inserted_rt.GetId().empty());
// retrieve inserted rt for testing // retrieve inserted rt for testing
...@@ -330,15 +344,15 @@ TEST_F(test_transaction, insert_delete_with_property) { ...@@ -330,15 +344,15 @@ TEST_F(test_transaction, insert_delete_with_property) {
ASSERT_TRUE(rt_retrieve_status.IsTerminated()); ASSERT_TRUE(rt_retrieve_status.IsTerminated());
ASSERT_FALSE(rt_retrieve_status.IsError()); ASSERT_FALSE(rt_retrieve_status.IsError());
const auto &rt_retrieve_results = const auto &rt_retrieve_results = rt_retrieval->GetResultSet();
dynamic_cast<const UniqueResult &>(rt_retrieval->GetResultSet());
const auto &retrieved_rt = rt_retrieve_results.GetEntity(); const auto &retrieved_rt = rt_retrieve_results.at(0);
EXPECT_EQ(inserted_rt.GetId(), retrieved_rt.GetId()); EXPECT_EQ(inserted_rt.GetId(), retrieved_rt.GetId());
EXPECT_EQ(rt.GetName(), retrieved_rt.GetName()); EXPECT_EQ(rt.GetName(), retrieved_rt.GetName());
EXPECT_EQ(retrieved_rt.GetProperties().Size(), 1); EXPECT_EQ(rt.GetDescription(), retrieved_rt.GetDescription());
EXPECT_EQ(retrieved_rt.GetProperties().size(), 1);
const auto &retrieved_prop_rt = retrieved_rt.GetProperties().At(0); const auto &retrieved_prop_rt = retrieved_rt.GetProperties().at(0);
EXPECT_EQ(retrieved_prop_rt.GetName(), prop_ent.GetName()); EXPECT_EQ(retrieved_prop_rt.GetName(), prop_ent.GetName());
EXPECT_EQ(retrieved_prop_rt.GetId(), inserted_prop.GetId()); EXPECT_EQ(retrieved_prop_rt.GetId(), inserted_prop.GetId());
EXPECT_EQ(retrieved_prop_rt.GetDatatype(), prop_ent.GetDatatype()); EXPECT_EQ(retrieved_prop_rt.GetDatatype(), prop_ent.GetDatatype());
...@@ -370,10 +384,9 @@ TEST_F(test_transaction, insert_delete_with_property) { ...@@ -370,10 +384,9 @@ TEST_F(test_transaction, insert_delete_with_property) {
ASSERT_TRUE(rec_insert_status.IsTerminated()); ASSERT_TRUE(rec_insert_status.IsTerminated());
ASSERT_FALSE(rec_insert_status.IsError()); ASSERT_FALSE(rec_insert_status.IsError());
const auto &rec_result_set = const auto &rec_result_set = rec_insertion->GetResultSet();
dynamic_cast<const UniqueResult &>(rec_insertion->GetResultSet());
const auto &inserted_rec = rec_result_set.GetEntity(); const auto &inserted_rec = rec_result_set.at(0);
EXPECT_FALSE(inserted_rec.GetId().empty()); EXPECT_FALSE(inserted_rec.GetId().empty());
// Retrieve the record and verify paretn and property // Retrieve the record and verify paretn and property
...@@ -385,20 +398,20 @@ TEST_F(test_transaction, insert_delete_with_property) { ...@@ -385,20 +398,20 @@ TEST_F(test_transaction, insert_delete_with_property) {
ASSERT_TRUE(rec_retrieve_status.IsTerminated()); ASSERT_TRUE(rec_retrieve_status.IsTerminated());
ASSERT_FALSE(rec_retrieve_status.IsError()); ASSERT_FALSE(rec_retrieve_status.IsError());
const auto &rec_retrieve_results = const auto &rec_retrieve_results = rec_retrieval->GetResultSet();
dynamic_cast<const UniqueResult &>(rec_retrieval->GetResultSet());
const auto &retrieved_rec = rec_retrieve_results.GetEntity(); const auto &retrieved_rec = rec_retrieve_results.at(0);
EXPECT_EQ(rec.GetName(), retrieved_rec.GetName()); EXPECT_EQ(rec.GetName(), retrieved_rec.GetName());
EXPECT_EQ(inserted_rec.GetId(), retrieved_rec.GetId()); EXPECT_EQ(inserted_rec.GetId(), retrieved_rec.GetId());
EXPECT_EQ(retrieved_rec.GetParents().Size(), 1); EXPECT_EQ(retrieved_rec.GetParents().size(), 1);
EXPECT_EQ(retrieved_rec.GetProperties().Size(), 1); EXPECT_EQ(retrieved_rec.GetProperties().size(), 1);
const auto &retrieved_parent_rec = retrieved_rec.GetParents().At(0); const auto &retrieved_parent_rec = retrieved_rec.GetParents().at(0);
EXPECT_EQ(retrieved_parent_rec.GetName(), rt.GetName()); EXPECT_EQ(retrieved_parent_rec.GetName(), rt.GetName());
EXPECT_EQ(retrieved_parent_rec.GetId(), inserted_rt.GetId()); EXPECT_EQ(retrieved_parent_rec.GetId(), inserted_rt.GetId());
EXPECT_EQ(retrieved_parent_rec.GetDescription(), rt.GetDescription());
const auto &retrieved_prop_rec = retrieved_rec.GetProperties().At(0); const auto &retrieved_prop_rec = retrieved_rec.GetProperties().at(0);
EXPECT_EQ(retrieved_prop_rec.GetName(), prop_ent.GetName()); EXPECT_EQ(retrieved_prop_rec.GetName(), prop_ent.GetName());
EXPECT_EQ(retrieved_prop_rec.GetId(), inserted_prop.GetId()); EXPECT_EQ(retrieved_prop_rec.GetId(), inserted_prop.GetId());
EXPECT_EQ(retrieved_prop_rec.GetDatatype(), prop_ent.GetDatatype()); EXPECT_EQ(retrieved_prop_rec.GetDatatype(), prop_ent.GetDatatype());
...@@ -407,7 +420,7 @@ TEST_F(test_transaction, insert_delete_with_property) { ...@@ -407,7 +420,7 @@ TEST_F(test_transaction, insert_delete_with_property) {
TEST_F(test_transaction, test_multi_retrieve) { TEST_F(test_transaction, test_multi_retrieve) {
const auto &connection = const auto &connection =
caosdb::connection::ConnectionManager::GetDefaultConnection(); caosdb::connection::ConnectionManager::GetDefaultConnection();
auto transaction(connection->CreateTransaction()); auto transaction(connection->CreateTransaction());
...@@ -418,24 +431,27 @@ TEST_F(test_transaction, test_multi_retrieve) { ...@@ -418,24 +431,27 @@ TEST_F(test_transaction, test_multi_retrieve) {
auto status = transaction->WaitForIt(); auto status = transaction->WaitForIt();
ASSERT_TRUE(status.IsTerminated()); ASSERT_TRUE(status.IsTerminated());
ASSERT_FALSE(status.IsError()); // Should have an error since entity 22 doesn't exist
ASSERT_TRUE(status.IsError());
const auto &result_set = transaction->GetResultSet(); const auto &result_set = transaction->GetResultSet();
EXPECT_EQ(result_set.Size(), 3); EXPECT_EQ(result_set.size(), 3);
EXPECT_EQ(result_set.At(1).GetId(), "21"); EXPECT_EQ(result_set.at(1).GetId(), "21");
EXPECT_EQ(result_set.At(1).GetName(), "unit"); EXPECT_EQ(result_set.at(1).GetName(), "unit");
EXPECT_FALSE(result_set.At(1).HasErrors()); // Exists so should be fine ...
EXPECT_FALSE(result_set.at(1).HasErrors());
EXPECT_EQ(result_set.At(2).GetId(), "22"); // ... but this does not
EXPECT_TRUE(result_set.At(2).HasErrors()); EXPECT_EQ(result_set.at(2).GetId(), "22");
EXPECT_EQ(result_set.At(2).GetErrors().At(0).GetCode(), EXPECT_TRUE(result_set.at(2).HasErrors());
EXPECT_EQ(result_set.at(2).GetErrors().at(0).GetCode(),
MessageCode::ENTITY_DOES_NOT_EXIST); MessageCode::ENTITY_DOES_NOT_EXIST);
} }
TEST_F(test_transaction, insert_update_delete) { TEST_F(test_transaction, insert_update_delete) {
const auto &connection = const auto &connection =
caosdb::connection::ConnectionManager::GetDefaultConnection(); caosdb::connection::ConnectionManager::GetDefaultConnection();
// INSERT // INSERT
auto insert_transaction(connection->CreateTransaction()); auto insert_transaction(connection->CreateTransaction());
...@@ -451,10 +467,9 @@ TEST_F(test_transaction, insert_update_delete) { ...@@ -451,10 +467,9 @@ TEST_F(test_transaction, insert_update_delete) {
ASSERT_TRUE(insert_status.IsTerminated()); ASSERT_TRUE(insert_status.IsTerminated());
ASSERT_FALSE(insert_status.IsError()); ASSERT_FALSE(insert_status.IsError());
const auto &insert_result_set = const auto &insert_result_set = insert_transaction->GetResultSet();
dynamic_cast<const UniqueResult &>(insert_transaction->GetResultSet());
const auto &new_entity = insert_result_set.GetEntity(); const auto &new_entity = insert_result_set.at(0);
EXPECT_FALSE(new_entity.GetId().empty()); EXPECT_FALSE(new_entity.GetId().empty());
EXPECT_FALSE(new_entity.HasErrors()); EXPECT_FALSE(new_entity.HasErrors());
...@@ -465,7 +480,7 @@ TEST_F(test_transaction, insert_update_delete) { ...@@ -465,7 +480,7 @@ TEST_F(test_transaction, insert_update_delete) {
// UPDATE // UPDATE
auto update_transaction(connection->CreateTransaction()); auto update_transaction(connection->CreateTransaction());
auto update_entity(retrieve_transaction->GetResultSet().At(0)); auto update_entity(retrieve_transaction->GetResultSet().at(0));
update_entity.SetName("RT1-Update"); update_entity.SetName("RT1-Update");
update_transaction->UpdateEntity(&update_entity); update_transaction->UpdateEntity(&update_entity);
...@@ -475,8 +490,8 @@ TEST_F(test_transaction, insert_update_delete) { ...@@ -475,8 +490,8 @@ TEST_F(test_transaction, insert_update_delete) {
ASSERT_TRUE(update_status.IsTerminated()); ASSERT_TRUE(update_status.IsTerminated());
ASSERT_FALSE(update_status.IsError()); ASSERT_FALSE(update_status.IsError());
EXPECT_EQ(update_transaction->GetResultSet().Size(), 1); EXPECT_EQ(update_transaction->GetResultSet().size(), 1);
const auto &updated_entity = update_transaction->GetResultSet().At(0); const auto &updated_entity = update_transaction->GetResultSet().at(0);
EXPECT_EQ(updated_entity.GetId(), new_entity.GetId()); EXPECT_EQ(updated_entity.GetId(), new_entity.GetId());
EXPECT_FALSE(updated_entity.HasErrors()); EXPECT_FALSE(updated_entity.HasErrors());
...@@ -492,17 +507,16 @@ TEST_F(test_transaction, insert_update_delete) { ...@@ -492,17 +507,16 @@ TEST_F(test_transaction, insert_update_delete) {
ASSERT_TRUE(delete_status.IsTerminated()); ASSERT_TRUE(delete_status.IsTerminated());
ASSERT_FALSE(delete_status.IsError()); ASSERT_FALSE(delete_status.IsError());
const auto &delete_result_set = const auto &delete_result_set = delete_transaction->GetResultSet();
dynamic_cast<const UniqueResult &>(delete_transaction->GetResultSet());
const auto &deleted_entity = delete_result_set.GetEntity(); const auto &deleted_entity = delete_result_set.at(0);
EXPECT_EQ(deleted_entity.GetId(), new_entity.GetId()); EXPECT_EQ(deleted_entity.GetId(), new_entity.GetId());
EXPECT_FALSE(deleted_entity.HasErrors()); EXPECT_FALSE(deleted_entity.HasErrors());
} }
TEST_F(test_transaction, test_query) { TEST_F(test_transaction, test_query) {
const auto &connection = const auto &connection =
caosdb::connection::ConnectionManager::GetDefaultConnection(); caosdb::connection::ConnectionManager::GetDefaultConnection();
auto insert_transaction(connection->CreateTransaction()); auto insert_transaction(connection->CreateTransaction());
...@@ -517,33 +531,34 @@ TEST_F(test_transaction, test_query) { ...@@ -517,33 +531,34 @@ TEST_F(test_transaction, test_query) {
ASSERT_TRUE(insert_status.IsTerminated()); ASSERT_TRUE(insert_status.IsTerminated());
ASSERT_FALSE(insert_status.IsError()); ASSERT_FALSE(insert_status.IsError());
const auto &insert_result_set = const auto &insert_result_set = insert_transaction->GetResultSet();
dynamic_cast<const UniqueResult &>(insert_transaction->GetResultSet());
const auto &new_entity = insert_result_set.GetEntity(); const auto &new_entity = insert_result_set.at(0);
EXPECT_FALSE(new_entity.GetId().empty()); EXPECT_FALSE(new_entity.GetId().empty());
EXPECT_FALSE(new_entity.HasErrors()); EXPECT_FALSE(new_entity.HasErrors());
auto query_transaction(connection->CreateTransaction()); auto query_transaction(connection->CreateTransaction());
query_transaction->Query("FIND ENTITY WITH id = " + new_entity.GetId()); query_transaction->Query("FIND ENTITY WITH id = " + new_entity.GetId());
query_transaction->Execute(); query_transaction->Execute();
EXPECT_EQ(query_transaction->GetResultSet().Size(), 1); EXPECT_EQ(query_transaction->GetResultSet().size(), 1);
EXPECT_EQ(query_transaction->GetResultSet().At(0).GetId(), EXPECT_EQ(query_transaction->GetResultSet().at(0).GetId(),
new_entity.GetId()); new_entity.GetId());
// No count query, so no count result should be present // No count query, so no count result should be present
EXPECT_TRUE((query_transaction->GetCountResult() < 0)); EXPECT_TRUE((query_transaction->GetCountResult() < 0));
auto count_query_trans(connection->CreateTransaction()); auto count_query_trans(connection->CreateTransaction());
std::cout << "Creating count query ..." << std::endl;
count_query_trans->Query("COUNT ENTITY WITH id = " + new_entity.GetId()); count_query_trans->Query("COUNT ENTITY WITH id = " + new_entity.GetId());
std::cout << "Executing count query ..." << std::endl;
count_query_trans->Execute(); count_query_trans->Execute();
// No result set in a count query // No result set in a count query
EXPECT_EQ(count_query_trans->GetResultSet().Size(), 0); EXPECT_EQ(count_query_trans->GetResultSet().size(), 0);
EXPECT_EQ(count_query_trans->GetCountResult(), 1); EXPECT_EQ(count_query_trans->GetCountResult(), 1);
} }
TEST_F(test_transaction, test_query_with_retrieve) { TEST_F(test_transaction, test_query_with_retrieve) {
const auto &connection = const auto &connection =
caosdb::connection::ConnectionManager::GetDefaultConnection(); caosdb::connection::ConnectionManager::GetDefaultConnection();
// rt1 // rt1
Entity rt1; Entity rt1;
...@@ -559,10 +574,9 @@ TEST_F(test_transaction, test_query_with_retrieve) { ...@@ -559,10 +574,9 @@ TEST_F(test_transaction, test_query_with_retrieve) {
ASSERT_TRUE(insert_rt1_status.IsTerminated()); ASSERT_TRUE(insert_rt1_status.IsTerminated());
ASSERT_FALSE(insert_rt1_status.IsError()); ASSERT_FALSE(insert_rt1_status.IsError());
const auto &insert_rt1_results = const auto &insert_rt1_results = insert_rt1_transaction->GetResultSet();
dynamic_cast<const UniqueResult &>(insert_rt1_transaction->GetResultSet());
const auto &inserted_rt1 = insert_rt1_results.GetEntity(); const auto &inserted_rt1 = insert_rt1_results.at(0);
EXPECT_FALSE(inserted_rt1.GetId().empty()); EXPECT_FALSE(inserted_rt1.GetId().empty());
EXPECT_FALSE(inserted_rt1.HasErrors()); EXPECT_FALSE(inserted_rt1.HasErrors());
...@@ -580,10 +594,9 @@ TEST_F(test_transaction, test_query_with_retrieve) { ...@@ -580,10 +594,9 @@ TEST_F(test_transaction, test_query_with_retrieve) {
ASSERT_TRUE(insert_rt2_status.IsTerminated()); ASSERT_TRUE(insert_rt2_status.IsTerminated());
ASSERT_FALSE(insert_rt2_status.IsError()); ASSERT_FALSE(insert_rt2_status.IsError());
const auto &insert_rt2_results = const auto &insert_rt2_results = insert_rt2_transaction->GetResultSet();
dynamic_cast<const UniqueResult &>(insert_rt2_transaction->GetResultSet());
const auto &inserted_rt2 = insert_rt2_results.GetEntity(); const auto &inserted_rt2 = insert_rt2_results.at(0);
EXPECT_FALSE(inserted_rt2.GetId().empty()); EXPECT_FALSE(inserted_rt2.GetId().empty());
EXPECT_FALSE(inserted_rt2.HasErrors()); EXPECT_FALSE(inserted_rt2.HasErrors());
...@@ -601,10 +614,9 @@ TEST_F(test_transaction, test_query_with_retrieve) { ...@@ -601,10 +614,9 @@ TEST_F(test_transaction, test_query_with_retrieve) {
ASSERT_TRUE(insert_rt3_status.IsTerminated()); ASSERT_TRUE(insert_rt3_status.IsTerminated());
ASSERT_FALSE(insert_rt3_status.IsError()); ASSERT_FALSE(insert_rt3_status.IsError());
const auto &insert_rt3_results = const auto &insert_rt3_results = insert_rt3_transaction->GetResultSet();
dynamic_cast<const UniqueResult &>(insert_rt3_transaction->GetResultSet());
const auto &inserted_rt3 = insert_rt3_results.GetEntity(); const auto &inserted_rt3 = insert_rt3_results.at(0);
EXPECT_FALSE(inserted_rt3.GetId().empty()); EXPECT_FALSE(inserted_rt3.GetId().empty());
EXPECT_FALSE(inserted_rt3.HasErrors()); EXPECT_FALSE(inserted_rt3.HasErrors());
...@@ -619,7 +631,7 @@ TEST_F(test_transaction, test_query_with_retrieve) { ...@@ -619,7 +631,7 @@ TEST_F(test_transaction, test_query_with_retrieve) {
ASSERT_FALSE(find_status.IsError()); ASSERT_FALSE(find_status.IsError());
const auto &find_results = find_transaction->GetResultSet(); const auto &find_results = find_transaction->GetResultSet();
EXPECT_EQ(find_results.Size(), 3); EXPECT_EQ(find_results.size(), 3);
// only retrieve rt1 and rt2 by id // only retrieve rt1 and rt2 by id
const std::vector<std::string> ids = {inserted_rt1.GetId(), const std::vector<std::string> ids = {inserted_rt1.GetId(),
...@@ -637,7 +649,7 @@ TEST_F(test_transaction, test_query_with_retrieve) { ...@@ -637,7 +649,7 @@ TEST_F(test_transaction, test_query_with_retrieve) {
ASSERT_FALSE(find_and_retrieve_status.IsError()); ASSERT_FALSE(find_and_retrieve_status.IsError());
const auto &result_set_a = find_and_retrieve->GetResultSet(); const auto &result_set_a = find_and_retrieve->GetResultSet();
EXPECT_EQ(result_set_a.Size(), 3); EXPECT_EQ(result_set_a.size(), 3);
// retrieve rt1 and rt2 by ID and count all TestRTs // retrieve rt1 and rt2 by ID and count all TestRTs
auto count_and_retrieve(connection->CreateTransaction()); auto count_and_retrieve(connection->CreateTransaction());
...@@ -652,8 +664,85 @@ TEST_F(test_transaction, test_query_with_retrieve) { ...@@ -652,8 +664,85 @@ TEST_F(test_transaction, test_query_with_retrieve) {
const auto &result_set_b = count_and_retrieve->GetResultSet(); const auto &result_set_b = count_and_retrieve->GetResultSet();
// TODO(fspreck) Re-enable once we implemented this // TODO(fspreck) Re-enable once we implemented this
// EXPECT_EQ(result_set_b.Size(), 2); // EXPECT_EQ(result_set_b.size(), 2);
EXPECT_EQ(count_and_retrieve->GetCountResult(), 3); EXPECT_EQ(count_and_retrieve->GetCountResult(), 3);
} }
TEST_F(test_transaction, test_file_upload) {
const auto &connection =
caosdb::connection::ConnectionManager::GetDefaultConnection();
Entity file;
file.SetRole("File");
file.SetFilePath("test.txt");
file.SetLocalPath(test_upload_file_1);
auto insert_transaction(connection->CreateTransaction());
insert_transaction->InsertEntity(&file);
insert_transaction->ExecuteAsynchronously();
auto insert_status = insert_transaction->WaitForIt();
ASSERT_TRUE(insert_status.IsTerminated());
EXPECT_EQ(insert_status.GetCode(), StatusCode::SUCCESS);
const auto &insert_results = insert_transaction->GetResultSet();
const auto &inserted_file = insert_results.at(0);
EXPECT_FALSE(inserted_file.GetId().empty());
EXPECT_FALSE(inserted_file.HasErrors());
// Check file once more
auto count_query(connection->CreateTransaction());
count_query->Query("COUNT FILE WHICH IS STORED AT 'test.txt'");
count_query->ExecuteAsynchronously();
auto count_and_retrieve_status = count_query->WaitForIt();
ASSERT_TRUE(count_and_retrieve_status.IsTerminated());
ASSERT_FALSE(count_and_retrieve_status.IsError());
EXPECT_EQ(count_query->GetCountResult(), 1);
}
TEST_F(test_transaction, test_file_up_n_download) {
const auto &connection =
caosdb::connection::ConnectionManager::GetDefaultConnection();
Entity file;
file.SetRole("File");
file.SetFilePath("test.txt");
file.SetLocalPath(test_upload_file_1);
auto insert_transaction(connection->CreateTransaction());
insert_transaction->InsertEntity(&file);
insert_transaction->Execute();
const auto &insert_results = insert_transaction->GetResultSet();
const auto &inserted_file = insert_results.at(0);
ASSERT_FALSE(inserted_file.GetId().empty());
ASSERT_FALSE(inserted_file.HasErrors());
auto download_transaction(connection->CreateTransaction());
download_transaction->RetrieveAndDownloadFilesById(
inserted_file.GetId(), test_download_file_1.string());
download_transaction->ExecuteAsynchronously();
ASSERT_EQ(download_transaction->WaitForIt().GetCode(), StatusCode::SUCCESS);
const auto &download_results = download_transaction->GetResultSet();
ASSERT_EQ(download_results.size(), 1);
const auto &downloaded_file = download_results.at(0);
ASSERT_FALSE(downloaded_file.GetId().empty());
ASSERT_FALSE(downloaded_file.HasErrors());
EXPECT_EQ(downloaded_file.GetLocalPath().string(),
test_download_file_1.string());
FileReader reader_remote(test_upload_file_1);
std::string buffer_local(1024, 'c');
std::string buffer_remote(1024, 'c');
for (int i = 0; i < 8; i++) {
reader_remote.read(buffer_remote);
EXPECT_EQ(buffer_remote, buffer_local);
}
}
} // namespace caosdb::transaction } // namespace caosdb::transaction
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment