test_protobuf.cpp

/*
 *
 * 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/data_type.h"               // for DataType, ReferenceDataType
#include "caosdb/entity.h"                  // for Entity
#include "caosdb/entity/v1alpha1/main.pb.h" // for RepeatedPtrField, Message
#include <gtest/gtest-message.h>            // for Message
#include <gtest/gtest-test-part.h>          // for SuiteApiResolver, TestPa...
#include <gtest/gtest_pred_impl.h>          // for Test, TestInfo, TEST
#include <memory>                           // for allocator

namespace caosdb {
using ProtoEntity = caosdb::entity::v1alpha1::Entity;
using caosdb::entity::Entity;
using caosdb::entity::v1alpha1::Message;

TEST(test_protobuf, test_swap_trivial) {
  Message message_source;
  message_source.set_code(1234);
  message_source.set_description("desc");

  Message message_destination;

  EXPECT_EQ(message_source.code(), 1234);
  EXPECT_EQ(message_source.description(), "desc");
  EXPECT_EQ(message_destination.code(), 0);
  EXPECT_EQ(message_destination.description(), "");

  message_source.Swap(&message_destination);

  EXPECT_EQ(message_source.code(), 0);
  EXPECT_EQ(message_source.description(), "");
  EXPECT_EQ(message_destination.code(), 1234);
  EXPECT_EQ(message_destination.description(), "desc");
}

TEST(test_protobuf, test_swap_nested) {
  ProtoEntity entity_source;
  entity_source.set_id("entity_id");
  auto *version_source = entity_source.mutable_version();
  version_source->set_id("version_id");

  ProtoEntity entity_destination;
  auto *version_destination = entity_destination.mutable_version();

  EXPECT_EQ(entity_source.id(), "entity_id");
  EXPECT_EQ(entity_source.version().id(), "version_id");
  EXPECT_EQ(version_source->id(), "version_id");
  EXPECT_EQ(entity_destination.id(), "");
  EXPECT_EQ(entity_destination.version().id(), "");
  EXPECT_EQ(version_destination->id(), "");

  entity_source.Swap(&entity_destination);

  EXPECT_EQ(entity_source.id(), "");
  EXPECT_EQ(entity_source.version().id(), "");
  EXPECT_EQ(entity_destination.id(), "entity_id");
  EXPECT_EQ(entity_destination.version().id(), "version_id");

  // has not been swapped!
  EXPECT_EQ(version_source->id(), "version_id");
  EXPECT_EQ(version_destination->id(), "");

  // Member pointers to nested messages have been swapped
  EXPECT_EQ(entity_source.mutable_version(), version_destination);
  EXPECT_EQ(entity_destination.mutable_version(), version_source);
}

TEST(test_protobuf, test_copy_nested) {
  ProtoEntity entity_source;
  auto *data_type_source = entity_source.mutable_data_type();
  data_type_source->mutable_reference_data_type()->set_name("src_per");

  ProtoEntity entity_destination;
  auto *data_type_destination = entity_destination.mutable_data_type();
  data_type_destination->mutable_reference_data_type()->set_name("dest_per");

  EXPECT_EQ(entity_source.data_type().reference_data_type().name(), "src_per");
  EXPECT_EQ(entity_destination.data_type().reference_data_type().name(), "dest_per");

  entity_destination.CopyFrom(entity_source);

  EXPECT_EQ(entity_source.data_type().reference_data_type().name(), "src_per");
  EXPECT_EQ(entity_destination.data_type().reference_data_type().name(), "src_per");

  Entity entity(&entity_destination);
  EXPECT_EQ(entity.GetDataType().AsReference().GetName(), "src_per");

  const Entity &copy_entity(entity);
  EXPECT_EQ(copy_entity.GetDataType().AsReference().GetName(), "src_per");
}

} // namespace caosdb