Skip to content
Snippets Groups Projects

ENH: Support FIND and COUNT queries

Merged Florian Spreckelsen requested to merge f-query into dev
1 unresolved thread
1 file
+ 1
1
Compare changes
  • Side-by-side
  • Inline
+ 36
17
@@ -31,6 +31,7 @@
#include "gtest/gtest_pred_impl.h" // for Test, TestInfo, TEST
#include <memory> // for allocator, unique_ptr
#include <string> // for string, basic_string
#include <utility> // for move
#include <vector> // for vector
namespace caosdb::transaction {
@@ -39,6 +40,7 @@ using caosdb::connection::Connection;
using caosdb::exceptions::ConnectionError;
using caosdb::transaction::UniqueResult;
using ProtoEntity = caosdb::entity::v1alpha1::Entity;
using caosdb::entity::v1alpha1::RetrieveResponse;
TEST(test_transaction, create_transaction) {
const auto *host = "localhost";
@@ -92,21 +94,18 @@ TEST(test_transaction, test_retrieve_by_ids) {
}
TEST(test_transaction, test_multi_result_set_empty) {
MultiTransactionResponse response;
MultiResultSet rs(&response);
std::vector<std::unique_ptr<Entity>> empty;
MultiResultSet rs(std::move(empty));
EXPECT_EQ(rs.Size(), 0);
}
TEST(test_transaction, test_multi_result_iterator) {
MultiTransactionResponse response;
response.add_responses()
->mutable_retrieve_response()
->mutable_entity()
->set_id("100");
std::vector<std::unique_ptr<Entity>> one_elem;
RetrieveResponse response;
response.mutable_entity()->set_id("100");
one_elem.push_back(std::make_unique<Entity>(response.release_entity()));
MultiResultSet rs(&response);
MultiResultSet rs(std::move(one_elem));
EXPECT_EQ(rs.Size(), 1);
for (const Entity &entity : rs) {
@@ -126,18 +125,19 @@ TEST(test_transaction, test_unique_result_iterator) {
}
TEST(test_transaction, test_multi_result_set_one) {
MultiTransactionResponse response;
response.add_responses()
->mutable_retrieve_response()
->mutable_entity()
->set_id("100");
std::vector<std::unique_ptr<Entity>> one_elem;
RetrieveResponse response;
response.mutable_entity()->set_id("100");
one_elem.push_back(std::make_unique<Entity>(response.release_entity()));
MultiResultSet rs(&response);
MultiResultSet rs(std::move(one_elem));
EXPECT_EQ(rs.Size(), 1);
EXPECT_EQ(rs.At(0).GetId(), "100");
}
TEST(test_transaction, test_multi_result_set_three) {
std::vector<std::unique_ptr<Entity>> three_elem;
MultiTransactionResponse response;
response.add_responses()
->mutable_retrieve_response()
@@ -152,7 +152,14 @@ TEST(test_transaction, test_multi_result_set_three) {
->mutable_entity()
->set_id("102");
MultiResultSet rs(&response);
auto *responses = response.mutable_responses();
std::vector<std::unique_ptr<Entity>> entities;
for (auto sub_response : *responses) {
three_elem.push_back(std::make_unique<Entity>(
sub_response.mutable_retrieve_response()->release_entity()));
}
MultiResultSet rs(std::move(three_elem));
EXPECT_EQ(rs.Size(), 3);
EXPECT_TRUE(rs.At(1).HasErrors());
}
@@ -170,4 +177,16 @@ TEST(test_transaction, test_update_entity) {
EXPECT_EQ(error, StatusCode::ORIGINAL_ENTITY_MISSING_ID);
}
TEST(test_transaction, test_multi_deletion) {
const auto *host = "localhost";
auto configuration = InsecureConnectionConfiguration(host, 8000);
Connection connection(configuration);
auto transaction = connection.CreateTransaction();
EXPECT_EQ(transaction->GetStatus().GetCode(), StatusCode::INITIAL);
for (int i = 0; i < 3; i++) {
auto status = transaction->DeleteById("asdf");
EXPECT_EQ(status, StatusCode::GO_ON);
}
}
} // namespace caosdb::transaction
Loading