diff --git a/test/test_list_properties.cpp b/test/test_list_properties.cpp index b8fdb0d4deaf0788aec6e72deada5106af2dffae..be3b302ca682154320f1c37bee6c985a46ca9b2f 100644 --- a/test/test_list_properties.cpp +++ b/test/test_list_properties.cpp @@ -24,6 +24,7 @@ #include "caosdb/transaction.h" // for Entity, Transaction,... #include "caosdb/transaction_status.h" // for TransactionStatus, StatusCode #include "caosdb/value.h" // for value +#include <cstdint> // for int64_t #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 @@ -113,4 +114,116 @@ TEST_F(test_list_properties, insert_list_of_text) { EXPECT_EQ(value.AsList().at(1).AsString(), "item5"); } +TEST_F(test_list_properties, insert_list_of_int) { + const auto &connection = + caosdb::connection::ConnectionManager::GetDefaultConnection(); + + auto insertion_prop(connection->CreateTransaction()); + + Entity abstract_list_property; + abstract_list_property.SetRole(Role::PROPERTY); + abstract_list_property.SetName("TestProp"); + abstract_list_property.SetDataType(DataType::ListOf(AtomicDataType::INTEGER)); + abstract_list_property.SetValue(std::vector<int64_t>{1, 2, 3}); + + insertion_prop->InsertEntity(&abstract_list_property); + std::cout << "response " << insertion_prop->ResponseToString(); + insertion_prop->Execute(); + EXPECT_TRUE(insertion_prop->GetStatus().IsTerminated()); + + auto insertion_rt(connection->CreateTransaction()); + + Property list_property; + list_property.SetId(insertion_prop->GetResultSet().at(0).GetId()); + list_property.SetValue(std::vector<int64_t>{4, 5, 6}); + + Entity entity; + entity.SetRole(Role::RECORD_TYPE); + entity.SetName("TestRT"); + entity.AppendProperty(list_property); + + insertion_rt->InsertEntity(&entity); + std::cout << "response " << insertion_rt->ResponseToString(); + insertion_rt->Execute(); + EXPECT_TRUE(insertion_rt->GetStatus().IsTerminated()); + EXPECT_FALSE(insertion_rt->GetStatus().IsError()); + + // retrieve and check again + auto retrieval(connection->CreateTransaction()); + retrieval->RetrieveById(insertion_rt->GetResultSet().at(0).GetId()); + retrieval->Execute(); + + EXPECT_TRUE(retrieval->GetStatus().IsTerminated()); + EXPECT_FALSE(retrieval->GetStatus().IsError()); + + const auto &same_entity = retrieval->GetResultSet().at(0); + const auto &data_type = same_entity.GetProperties().at(0).GetDataType(); + const auto &value = same_entity.GetProperties().at(0).GetValue(); + + EXPECT_TRUE(data_type.IsList()); + EXPECT_TRUE(data_type.AsList().IsListOfAtomic()); + EXPECT_EQ(data_type.AsList().GetAtomicDataType(), AtomicDataType::INTEGER); + + EXPECT_TRUE(value.IsList()); + EXPECT_EQ(value.AsList().size(), 3); + EXPECT_TRUE(value.AsList().at(1).IsInteger()); + EXPECT_EQ(value.AsList().at(1).AsInteger(), 5); +} + +TEST_F(test_list_properties, insert_list_of_bool) { + const auto &connection = + caosdb::connection::ConnectionManager::GetDefaultConnection(); + + auto insertion_prop(connection->CreateTransaction()); + + Entity abstract_list_property; + abstract_list_property.SetRole(Role::PROPERTY); + abstract_list_property.SetName("TestProp"); + abstract_list_property.SetDataType(DataType::ListOf(AtomicDataType::INTEGER)); + abstract_list_property.SetValue(std::vector<bool>{true, true, false}); + + insertion_prop->InsertEntity(&abstract_list_property); + std::cout << "response " << insertion_prop->ResponseToString(); + insertion_prop->Execute(); + EXPECT_TRUE(insertion_prop->GetStatus().IsTerminated()); + + auto insertion_rt(connection->CreateTransaction()); + + Property list_property; + list_property.SetId(insertion_prop->GetResultSet().at(0).GetId()); + list_property.SetValue(std::vector<bool>{false, false, true}); + + Entity entity; + entity.SetRole(Role::RECORD_TYPE); + entity.SetName("TestRT"); + entity.AppendProperty(list_property); + + insertion_rt->InsertEntity(&entity); + std::cout << "response " << insertion_rt->ResponseToString(); + insertion_rt->Execute(); + EXPECT_TRUE(insertion_rt->GetStatus().IsTerminated()); + EXPECT_FALSE(insertion_rt->GetStatus().IsError()); + + // retrieve and check again + auto retrieval(connection->CreateTransaction()); + retrieval->RetrieveById(insertion_rt->GetResultSet().at(0).GetId()); + retrieval->Execute(); + + EXPECT_TRUE(retrieval->GetStatus().IsTerminated()); + EXPECT_FALSE(retrieval->GetStatus().IsError()); + + const auto &same_entity = retrieval->GetResultSet().at(0); + const auto &data_type = same_entity.GetProperties().at(0).GetDataType(); + const auto &value = same_entity.GetProperties().at(0).GetValue(); + + EXPECT_TRUE(data_type.IsList()); + EXPECT_TRUE(data_type.AsList().IsListOfAtomic()); + EXPECT_EQ(data_type.AsList().GetAtomicDataType(), AtomicDataType::INTEGER); + + EXPECT_TRUE(value.IsList()); + EXPECT_EQ(value.AsList().size(), 3); + EXPECT_TRUE(value.AsList().at(1).IsBool()); + EXPECT_FALSE(value.AsList().at(1).AsBool()); +} + } // namespace caosdb::entity