diff --git a/conanfile.txt b/conanfile.txt index a5a7c5f57a87bc692b51558e8483daa28aeb79e6..ccd05f44e4855c4d4b13dc1fbc7df8554dd4e8ad 100644 --- a/conanfile.txt +++ b/conanfile.txt @@ -1,5 +1,5 @@ [requires] -caosdb/0.0.10 +caosdb/0.0.11 gtest/1.11.0 [generators] diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt index 66ea8a30ae64a7433b01d76a925705303b35b650..62c9504c5dbf750ad5f7165d6261e3e7c6fed3d2 100644 --- a/test/CMakeLists.txt +++ b/test/CMakeLists.txt @@ -82,7 +82,7 @@ if(LINTING) "--warnings-as-errors=*" "--fix") 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,-cppcoreguidelines-pro-type-vararg,-cppcoreguidelines-pro-bounds-array-to-pointer-decay,-hicpp-no-array-decay") + "--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() else() message(STATUS "LINTING is OFF") @@ -105,7 +105,8 @@ foreach (i RANGE "${len_test_cases}") target_link_libraries(${test_case_name} PRIVATE ${CONAN_LIBS_CAOSDB} ${CONAN_LIBS_GTEST} ${CONAN_LIBS_GRPC} ${CONAN_LIBS_ABSEIL} ${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} PUBLIC ${CONAN_INCLUDE_DIRS} ${CMAKE_CURRENT_SOURCE_DIR}) if(LINTING) diff --git a/test/test_file_transmission.cpp b/test/test_file_transmission.cpp new file mode 100644 index 0000000000000000000000000000000000000000..7c6a72d0b4d52ecd76d01d253e377473ce7954cf --- /dev/null +++ b/test/test_file_transmission.cpp @@ -0,0 +1,158 @@ +/* + * 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; + +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(®istration_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(®istration_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 diff --git a/test/test_transaction.cpp b/test/test_transaction.cpp index f57b1fe6a0550f5940a8385f4af0db223f1bf9b6..cb998dcfddbe61f45490bd394099eefbcbdb4a01 100644 --- a/test/test_transaction.cpp +++ b/test/test_transaction.cpp @@ -18,19 +18,25 @@ * along with this program. If not, see <https://www.gnu.org/licenses/>. * */ -#include "caosdb/connection.h" // for Connection, ConnectionManager -#include "caosdb/entity.h" // for Entity, Messages, Message +#include "caosdb/connection.h" // for Connection, ConnectionManager +#include "caosdb/entity.h" // for Entity, Messages, Message +#include "caosdb/file_transmission/file_writer.h" // for FileWriter #include "caosdb/message_code.h" // for ENTITY_DOES_NOT_EXIST, Messag... #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 "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 <boost/filesystem/operations.hpp> // for remove +#include <boost/filesystem/path.hpp> // for path +#include <boost/filesystem/path_traits.hpp> // for filesystem +#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 <iostream> #include <memory> // for unique_ptr, allocator, __shar... #include <string> // for string #include <vector> // for vector + +namespace fs = boost::filesystem; namespace caosdb::transaction { using caosdb::entity::Entity; using caosdb::entity::MessageCode; @@ -39,9 +45,24 @@ using caosdb::entity::Property; class test_transaction : public ::testing::Test { 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 { + fs::remove(test_upload_file_1); + fs::remove(test_download_file_1); + const auto &connection = caosdb::connection::ConnectionManager::GetDefaultConnection(); auto query_transaction(connection->CreateTransaction()); @@ -71,8 +92,7 @@ TEST_F(test_transaction, DISABLED_retrieve_manufacturer_by_id) { transaction->RetrieveById(id); transaction->Execute(); - const auto &result_set = - dynamic_cast<const MultiResultSet &>(transaction->GetResultSet()); + const auto &result_set = transaction->GetResultSet(); const auto &entity = result_set.at(0); EXPECT_EQ(id, entity.GetId()); @@ -96,14 +116,13 @@ TEST_F(test_transaction, retrieve_non_existing) { EXPECT_EQ(status.GetCode(), TransactionStatus::TRANSACTION_ERROR().GetCode()); ASSERT_EQ(status.GetCode(), StatusCode::GENERIC_TRANSACTION_ERROR); - const auto &result_set = - dynamic_cast<const MultiResultSet &>(transaction->GetResultSet()); + const auto &result_set = transaction->GetResultSet(); const auto &entity = result_set.at(0); EXPECT_EQ(id, entity.GetId()); EXPECT_TRUE(entity.HasErrors()); - ASSERT_EQ(entity.GetErrors().Size(), 1); - EXPECT_EQ(entity.GetErrors().At(0).GetCode(), + ASSERT_EQ(entity.GetErrors().size(), 1); + EXPECT_EQ(entity.GetErrors().at(0).GetCode(), MessageCode::ENTITY_DOES_NOT_EXIST); } @@ -124,16 +143,15 @@ TEST_F(test_transaction, insert_without_delete) { ASSERT_TRUE(insert_status.IsTerminated()); ASSERT_FALSE(insert_status.IsError()); - const auto &insert_result_set = - dynamic_cast<const MultiResultSet &>(insert_transaction->GetResultSet()); + const auto &insert_result_set = insert_transaction->GetResultSet(); const auto &new_entity = insert_result_set.at(0); EXPECT_FALSE(new_entity.GetId().empty()); EXPECT_FALSE(new_entity.HasErrors()); // Should have a warning since it has no properties EXPECT_TRUE(new_entity.HasWarnings()); - EXPECT_EQ(new_entity.GetWarnings().Size(), 1); - EXPECT_EQ(new_entity.GetWarnings().At(0).GetCode(), + EXPECT_EQ(new_entity.GetWarnings().size(), 1); + EXPECT_EQ(new_entity.GetWarnings().at(0).GetCode(), MessageCode::ENTITY_HAS_NO_PROPERTIES); } @@ -154,16 +172,15 @@ TEST_F(test_transaction, insert_delete) { ASSERT_TRUE(insert_status.IsTerminated()); ASSERT_FALSE(insert_status.IsError()); - const auto &insert_result_set = - dynamic_cast<const MultiResultSet &>(insert_transaction->GetResultSet()); + const auto &insert_result_set = insert_transaction->GetResultSet(); const auto &new_entity = insert_result_set.at(0); EXPECT_FALSE(new_entity.GetId().empty()); EXPECT_FALSE(new_entity.HasErrors()); // Should have a warning since it has no properties EXPECT_TRUE(new_entity.HasWarnings()); - EXPECT_EQ(new_entity.GetWarnings().Size(), 1); - EXPECT_EQ(new_entity.GetWarnings().At(0).GetCode(), + EXPECT_EQ(new_entity.GetWarnings().size(), 1); + EXPECT_EQ(new_entity.GetWarnings().at(0).GetCode(), MessageCode::ENTITY_HAS_NO_PROPERTIES); auto delete_transaction(connection->CreateTransaction()); @@ -176,8 +193,7 @@ TEST_F(test_transaction, insert_delete) { ASSERT_TRUE(delete_status.IsTerminated()); ASSERT_FALSE(delete_status.IsError()); - const auto &delete_result_set = - dynamic_cast<const MultiResultSet &>(delete_transaction->GetResultSet()); + const auto &delete_result_set = delete_transaction->GetResultSet(); const auto &deleted_entity = delete_result_set.at(0); EXPECT_EQ(deleted_entity.GetId(), new_entity.GetId()); @@ -201,8 +217,7 @@ TEST_F(test_transaction, insert_delete_with_parent) { ASSERT_TRUE(insert_status.IsTerminated()); ASSERT_FALSE(insert_status.IsError()); - const auto &insert_result_set = - dynamic_cast<const MultiResultSet &>(insert_transaction->GetResultSet()); + const auto &insert_result_set = insert_transaction->GetResultSet(); const auto &inserted_rt = insert_result_set.at(0); @@ -224,8 +239,7 @@ TEST_F(test_transaction, insert_delete_with_parent) { ASSERT_TRUE(rec_insert_status.IsTerminated()); ASSERT_FALSE(rec_insert_status.IsError()); - const auto &rec_result_set = - dynamic_cast<const MultiResultSet &>(rec_transaction->GetResultSet()); + const auto &rec_result_set = rec_transaction->GetResultSet(); const auto &inserted_rec = rec_result_set.at(0); @@ -241,14 +255,13 @@ TEST_F(test_transaction, insert_delete_with_parent) { ASSERT_TRUE(rec_retrieve_status.IsTerminated()); ASSERT_FALSE(rec_retrieve_status.IsError()); - const auto &retrieve_result_set = - dynamic_cast<const MultiResultSet &>(retrieve_transaction->GetResultSet()); + const auto &retrieve_result_set = retrieve_transaction->GetResultSet(); const auto &retrieved_rec = retrieve_result_set.at(0); EXPECT_EQ(retrieved_rec.GetName(), rec.GetName()); - 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).GetName(), rt.GetName()); + 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).GetName(), rt.GetName()); auto rec_deletion(connection->CreateTransaction()); @@ -290,8 +303,7 @@ TEST_F(test_transaction, insert_delete_with_property) { ASSERT_TRUE(prop_insert_status.IsTerminated()); ASSERT_FALSE(prop_insert_status.IsError()); - const auto &prop_result_set = - dynamic_cast<const MultiResultSet &>(prop_insertion->GetResultSet()); + const auto &prop_result_set = prop_insertion->GetResultSet(); const auto &inserted_prop = prop_result_set.at(0); EXPECT_FALSE(inserted_prop.GetId().empty()); @@ -317,8 +329,7 @@ TEST_F(test_transaction, insert_delete_with_property) { ASSERT_TRUE(rt_insert_status.IsTerminated()); ASSERT_FALSE(rt_insert_status.IsError()); - const auto &rt_result_set = - dynamic_cast<const MultiResultSet &>(rt_insertion->GetResultSet()); + const auto &rt_result_set = rt_insertion->GetResultSet(); const auto &inserted_rt = rt_result_set.at(0); EXPECT_FALSE(inserted_rt.GetId().empty()); @@ -332,16 +343,15 @@ TEST_F(test_transaction, insert_delete_with_property) { ASSERT_TRUE(rt_retrieve_status.IsTerminated()); ASSERT_FALSE(rt_retrieve_status.IsError()); - const auto &rt_retrieve_results = - dynamic_cast<const MultiResultSet &>(rt_retrieval->GetResultSet()); + const auto &rt_retrieve_results = rt_retrieval->GetResultSet(); const auto &retrieved_rt = rt_retrieve_results.at(0); EXPECT_EQ(inserted_rt.GetId(), retrieved_rt.GetId()); EXPECT_EQ(rt.GetName(), retrieved_rt.GetName()); EXPECT_EQ(rt.GetDescription(), retrieved_rt.GetDescription()); - EXPECT_EQ(retrieved_rt.GetProperties().Size(), 1); + 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.GetId(), inserted_prop.GetId()); EXPECT_EQ(retrieved_prop_rt.GetDatatype(), prop_ent.GetDatatype()); @@ -373,8 +383,7 @@ TEST_F(test_transaction, insert_delete_with_property) { ASSERT_TRUE(rec_insert_status.IsTerminated()); ASSERT_FALSE(rec_insert_status.IsError()); - const auto &rec_result_set = - dynamic_cast<const MultiResultSet &>(rec_insertion->GetResultSet()); + const auto &rec_result_set = rec_insertion->GetResultSet(); const auto &inserted_rec = rec_result_set.at(0); EXPECT_FALSE(inserted_rec.GetId().empty()); @@ -388,21 +397,20 @@ TEST_F(test_transaction, insert_delete_with_property) { ASSERT_TRUE(rec_retrieve_status.IsTerminated()); ASSERT_FALSE(rec_retrieve_status.IsError()); - const auto &rec_retrieve_results = - dynamic_cast<const MultiResultSet &>(rec_retrieval->GetResultSet()); + const auto &rec_retrieve_results = rec_retrieval->GetResultSet(); const auto &retrieved_rec = rec_retrieve_results.at(0); EXPECT_EQ(rec.GetName(), retrieved_rec.GetName()); EXPECT_EQ(inserted_rec.GetId(), retrieved_rec.GetId()); - EXPECT_EQ(retrieved_rec.GetParents().Size(), 1); - EXPECT_EQ(retrieved_rec.GetProperties().Size(), 1); + EXPECT_EQ(retrieved_rec.GetParents().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.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.GetId(), inserted_prop.GetId()); EXPECT_EQ(retrieved_prop_rec.GetDatatype(), prop_ent.GetDatatype()); @@ -433,10 +441,11 @@ TEST_F(test_transaction, test_multi_retrieve) { // Exists so should be fine ... EXPECT_FALSE(result_set.at(1).HasErrors()); + // ... but this does not EXPECT_EQ(result_set.at(2).GetId(), "22"); EXPECT_TRUE(result_set.at(2).HasErrors()); - EXPECT_EQ(result_set.at(2).GetErrors().At(0).GetCode(), + EXPECT_EQ(result_set.at(2).GetErrors().at(0).GetCode(), MessageCode::ENTITY_DOES_NOT_EXIST); } @@ -458,8 +467,7 @@ TEST_F(test_transaction, insert_update_delete) { ASSERT_TRUE(insert_status.IsTerminated()); ASSERT_FALSE(insert_status.IsError()); - const auto &insert_result_set = - dynamic_cast<const MultiResultSet &>(insert_transaction->GetResultSet()); + const auto &insert_result_set = insert_transaction->GetResultSet(); const auto &new_entity = insert_result_set.at(0); EXPECT_FALSE(new_entity.GetId().empty()); @@ -499,8 +507,7 @@ TEST_F(test_transaction, insert_update_delete) { ASSERT_TRUE(delete_status.IsTerminated()); ASSERT_FALSE(delete_status.IsError()); - const auto &delete_result_set = - dynamic_cast<const MultiResultSet &>(delete_transaction->GetResultSet()); + const auto &delete_result_set = delete_transaction->GetResultSet(); const auto &deleted_entity = delete_result_set.at(0); EXPECT_EQ(deleted_entity.GetId(), new_entity.GetId()); @@ -524,8 +531,7 @@ TEST_F(test_transaction, test_query) { ASSERT_TRUE(insert_status.IsTerminated()); ASSERT_FALSE(insert_status.IsError()); - const auto &insert_result_set = - dynamic_cast<const MultiResultSet &>(insert_transaction->GetResultSet()); + const auto &insert_result_set = insert_transaction->GetResultSet(); const auto &new_entity = insert_result_set.at(0); EXPECT_FALSE(new_entity.GetId().empty()); @@ -546,9 +552,7 @@ TEST_F(test_transaction, test_query) { std::cout << "Executing count query ..." << std::endl; count_query_trans->Execute(); // No result set in a count query - std::cout << "Checking count query result set..." << std::endl; EXPECT_EQ(count_query_trans->GetResultSet().size(), 0); - std::cout << "Checking count query result ..." << std::endl; EXPECT_EQ(count_query_trans->GetCountResult(), 1); } @@ -570,8 +574,7 @@ TEST_F(test_transaction, test_query_with_retrieve) { ASSERT_TRUE(insert_rt1_status.IsTerminated()); ASSERT_FALSE(insert_rt1_status.IsError()); - const auto &insert_rt1_results = dynamic_cast<const MultiResultSet &>( - insert_rt1_transaction->GetResultSet()); + const auto &insert_rt1_results = insert_rt1_transaction->GetResultSet(); const auto &inserted_rt1 = insert_rt1_results.at(0); EXPECT_FALSE(inserted_rt1.GetId().empty()); @@ -591,8 +594,7 @@ TEST_F(test_transaction, test_query_with_retrieve) { ASSERT_TRUE(insert_rt2_status.IsTerminated()); ASSERT_FALSE(insert_rt2_status.IsError()); - const auto &insert_rt2_results = dynamic_cast<const MultiResultSet &>( - insert_rt2_transaction->GetResultSet()); + const auto &insert_rt2_results = insert_rt2_transaction->GetResultSet(); const auto &inserted_rt2 = insert_rt2_results.at(0); EXPECT_FALSE(inserted_rt2.GetId().empty()); @@ -612,8 +614,7 @@ TEST_F(test_transaction, test_query_with_retrieve) { ASSERT_TRUE(insert_rt3_status.IsTerminated()); ASSERT_FALSE(insert_rt3_status.IsError()); - const auto &insert_rt3_results = dynamic_cast<const MultiResultSet &>( - insert_rt3_transaction->GetResultSet()); + const auto &insert_rt3_results = insert_rt3_transaction->GetResultSet(); const auto &inserted_rt3 = insert_rt3_results.at(0); EXPECT_FALSE(inserted_rt3.GetId().empty()); @@ -667,4 +668,64 @@ TEST_F(test_transaction, test_query_with_retrieve) { 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()); +} + +TEST_F(test_transaction, test_file_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()); +} + } // namespace caosdb::transaction