variant_tests.cpp - Android社区 - https://www.androidos.net.cn/

#include <cstdint>
#include <functional>
#include <memory>
#include <string>
#include <type_traits>

#include <gtest/gtest.h>
#include <pdx/rpc/variant.h>

using namespace android::pdx;
using namespace android::pdx::rpc;

namespace {

struct BaseType {
  BaseType(int value) : value(value) {}
  int value;
};

struct DerivedType : BaseType {
  DerivedType(int value) : BaseType{value} {};
};

template <typename T>
class TestType {
 public:
  TestType(const T& value) : value_(value) {}
  TestType(T&& value) : value_(std::move(value)) {}
  TestType(const TestType&) = default;
  TestType(TestType&&) = default;

TestType& operator=(const TestType&) = default;
  TestType& operator=(TestType&&) = default;

const T& get() const { return value_; }
  T&& take() { return std::move(value_); }

private:
  T value_;
};

template <typename T>
class InstrumentType {
 public:
  InstrumentType(const T& value) : value_(value) { constructor_count_++; }
  InstrumentType(T&& value) : value_(std::move(value)) { constructor_count_++; }
  InstrumentType(const InstrumentType& other) : value_(other.value_) {
    constructor_count_++;
  }
  InstrumentType(InstrumentType&& other) : value_(std::move(other.value_)) {
    constructor_count_++;
  }
  InstrumentType(const TestType<T>& other) : value_(other.get()) {
    constructor_count_++;
  }
  InstrumentType(TestType<T>&& other) : value_(other.take()) {
    constructor_count_++;
  }
  ~InstrumentType() { destructor_count_++; }

InstrumentType& operator=(const InstrumentType& other) {
    copy_assignment_count_++;
    value_ = other.value_;
    return *this;
  }
  InstrumentType& operator=(InstrumentType&& other) {
    move_assignment_count_++;
    value_ = std::move(other.value_);
    return *this;
  }

InstrumentType& operator=(const TestType<T>& other) {
    copy_assignment_count_++;
    value_ = other.get();
    return *this;
  }
  InstrumentType& operator=(TestType<T>&& other) {
    move_assignment_count_++;
    value_ = other.take();
    return *this;
  }

static std::size_t constructor_count() { return constructor_count_; }
  static std::size_t destructor_count() { return destructor_count_; }
  static std::size_t move_assignment_count() { return move_assignment_count_; }
  static std::size_t copy_assignment_count() { return copy_assignment_count_; }

const T& get() const { return value_; }
  T&& take() { return std::move(value_); }

static void clear() {
    constructor_count_ = 0;
    destructor_count_ = 0;
    move_assignment_count_ = 0;
    copy_assignment_count_ = 0;
  }

private:
  T value_;

static std::size_t constructor_count_;
  static std::size_t destructor_count_;
  static std::size_t move_assignment_count_;
  static std::size_t copy_assignment_count_;
};

template <typename T>
std::size_t InstrumentType<T>::constructor_count_ = 0;
template <typename T>
std::size_t InstrumentType<T>::destructor_count_ = 0;
template <typename T>
std::size_t InstrumentType<T>::move_assignment_count_ = 0;
template <typename T>
std::size_t InstrumentType<T>::copy_assignment_count_ = 0;

}  // anonymous namespace

TEST(Variant, Assignment) {
  // Assert basic type properties.
  {
    Variant<int, bool, float> v;
    ASSERT_EQ(-1, v.index());
    ASSERT_FALSE(v.is<int>());
    ASSERT_FALSE(v.is<bool>());
    ASSERT_FALSE(v.is<float>());
  }

{
    Variant<int, bool, float> v;
    v = 10;
    ASSERT_EQ(0, v.index());
    ASSERT_TRUE(v.is<int>());
    ASSERT_FALSE(v.is<bool>());
    ASSERT_FALSE(v.is<float>());
    EXPECT_EQ(10, std::get<int>(v));
  }

{
    Variant<int, bool, float> v;
    v = false;
    ASSERT_EQ(1, v.index());
    ASSERT_FALSE(v.is<int>());
    ASSERT_TRUE(v.is<bool>());
    ASSERT_FALSE(v.is<float>());
    EXPECT_EQ(false, std::get<bool>(v));
  }

{
    Variant<int, bool, float> v;
    v = 1.0f;
    ASSERT_EQ(2, v.index());
    ASSERT_FALSE(v.is<int>());
    ASSERT_FALSE(v.is<bool>());
    ASSERT_TRUE(v.is<float>());
    EXPECT_FLOAT_EQ(1.0f, std::get<float>(v));
  }

{
    Variant<int, bool, float> v;
    // ERROR: More than one type is implicitly convertible from double.
    // v = 1.0;
    v = static_cast<float>(1.0);
  }

{
    Variant<int, bool, float> v;

double x = 1.1;
    v = static_cast<float>(x);
    ASSERT_EQ(2, v.index());
    ASSERT_FALSE(v.is<int>());
    ASSERT_FALSE(v.is<bool>());
    ASSERT_TRUE(v.is<float>());
    EXPECT_FLOAT_EQ(1.1, std::get<float>(v));
  }

{
    Variant<int, std::string> v;
    ASSERT_EQ(-1, v.index());
    ASSERT_FALSE(v.is<int>());
    ASSERT_FALSE(v.is<std::string>());
  }

{
    Variant<int, std::string> v;
    v = 20;
    ASSERT_EQ(0, v.index());
    ASSERT_TRUE(v.is<int>());
    ASSERT_FALSE(v.is<std::string>());
    EXPECT_EQ(20, std::get<int>(v));
  }

{
    Variant<int, std::string> v;
    v = std::string("test");
    ASSERT_EQ(1, v.index());
    ASSERT_FALSE(v.is<int>());
    ASSERT_TRUE(v.is<std::string>());
    EXPECT_EQ("test", std::get<std::string>(v));
  }

{
    Variant<int, std::string> v;
    v = "test";
    ASSERT_EQ(1, v.index());
    ASSERT_FALSE(v.is<int>());
    ASSERT_TRUE(v.is<std::string>());
    EXPECT_EQ("test", std::get<std::string>(v));
  }

{
    Variant<const char*> v1;
    Variant<std::string> v2;

v1 = "test";
    ASSERT_TRUE(v1.is<const char*>());
    v2 = v1;
    ASSERT_TRUE(v2.is<std::string>());
    EXPECT_EQ("test", std::get<std::string>(v2));
  }

{
    Variant<int> a(1);
    Variant<int> b;
    ASSERT_TRUE(!a.empty());
    ASSERT_TRUE(b.empty());

a = b;
    ASSERT_TRUE(a.empty());
    ASSERT_TRUE(b.empty());
  }

{
    Variant<int*, char*> v;

// ERROR: More than one type is implicitly convertible from nullptr.
    // v = nullptr;

v = static_cast<int*>(nullptr);
    EXPECT_TRUE(v.is<int*>());

v = static_cast<char*>(nullptr);
    EXPECT_TRUE(v.is<char*>());
  }

{
    Variant<int*, char*> v;
    int a = 10;
    char b = 20;

v = &b;
    ASSERT_TRUE(v.is<char*>());
    EXPECT_EQ(&b, std::get<char*>(v));
    EXPECT_EQ(b, *std::get<char*>(v));

v = &a;
    ASSERT_TRUE(v.is<int*>());
    EXPECT_EQ(&a, std::get<int*>(v));
    EXPECT_EQ(a, *std::get<int*>(v));
  }

{
    using IntRef = std::reference_wrapper<int>;
    Variant<IntRef> v;
    int a = 10;

v = a;
    ASSERT_TRUE(v.is<IntRef>());
    EXPECT_EQ(a, std::get<IntRef>(v));

a = 20;
    EXPECT_EQ(a, std::get<IntRef>(v));
  }
}

TEST(Variant, MoveAssignment) {
  {
    Variant<std::string> v;
    std::string s = "test";
    v = std::move(s);

EXPECT_TRUE(s.empty());
    ASSERT_TRUE(v.is<std::string>());
    EXPECT_EQ("test", std::get<std::string>(v));
  }

{
    Variant<std::string> v("test");
    std::string s = "fizz";
    s = std::move(std::get<std::string>(v));

ASSERT_TRUE(v.is<std::string>());
    EXPECT_TRUE(std::get<std::string>(v).empty());
    EXPECT_EQ("test", s);
  }

{
    Variant<std::string> a("test");
    Variant<std::string> b;

b = std::move(a);
    ASSERT_TRUE(a.is<std::string>());
    ASSERT_TRUE(b.is<std::string>());
    EXPECT_TRUE(std::get<std::string>(a).empty());
    EXPECT_EQ("test", std::get<std::string>(b));
  }

{
    Variant<std::string> a("test");
    Variant<std::string> b("fizz");

b = std::move(a);
    ASSERT_TRUE(a.is<std::string>());
    ASSERT_TRUE(b.is<std::string>());
    EXPECT_TRUE(std::get<std::string>(a).empty());
    EXPECT_EQ("test", std::get<std::string>(b));
  }

{
    Variant<int, std::string> a("test");
    Variant<int, std::string> b(10);

b = std::move(a);
    ASSERT_TRUE(a.is<std::string>());
    ASSERT_TRUE(b.is<std::string>());
    EXPECT_TRUE(std::get<std::string>(a).empty());
    EXPECT_EQ("test", std::get<std::string>(b));
  }

{
    Variant<int, std::string> a(10);
    Variant<int, std::string> b("test");

b = std::move(a);
    ASSERT_TRUE(a.is<int>());
    ASSERT_TRUE(b.is<int>());
    EXPECT_EQ(10, std::get<int>(a));
    EXPECT_EQ(10, std::get<int>(b));
  }
}

TEST(Variant, Constructor) {
  {
    Variant<int, bool, float> v(true);
    EXPECT_TRUE(v.is<bool>());
  }

{
    Variant<int, bool, float> v(10);
    EXPECT_TRUE(v.is<int>());
  }

{
    Variant<int, bool, float> v(10.1f);
    EXPECT_TRUE(v.is<float>());
  }

{
    Variant<float, std::string> v(10.);
    EXPECT_TRUE(v.is<float>());
  }

{
    TestType<int> i(1);
    Variant<int, bool, float> v(i.take());
    ASSERT_TRUE(v.is<int>());
    EXPECT_EQ(1, std::get<int>(v));
  }

{
    TestType<int> i(1);
    Variant<int, bool, float> v(i.get());
    ASSERT_TRUE(v.is<int>());
    EXPECT_EQ(1, std::get<int>(v));
  }

{
    TestType<bool> b(true);
    Variant<int, bool, float> v(b.take());
    ASSERT_TRUE(v.is<bool>());
    EXPECT_EQ(true, std::get<bool>(v));
  }

{
    TestType<bool> b(true);
    Variant<int, bool, float> v(b.get());
    ASSERT_TRUE(v.is<bool>());
    EXPECT_EQ(true, std::get<bool>(v));
  }

{
    Variant<const char*> c("test");
    Variant<std::string> s(c);
    ASSERT_TRUE(s.is<std::string>());
    EXPECT_EQ("test", std::get<std::string>(s));
  }

{
    Variant<int, bool, float> a(true);
    Variant<int, bool, float> b(a);

ASSERT_TRUE(b.is<bool>());
  }

{
    using IntRef = std::reference_wrapper<int>;
    int a = 10;
    Variant<IntRef> v(a);
    TestType<IntRef> t(a);

ASSERT_TRUE(v.is<IntRef>());
    EXPECT_EQ(a, std::get<IntRef>(v));
    EXPECT_EQ(a, t.get());

a = 20;
    EXPECT_EQ(a, std::get<IntRef>(v));
    EXPECT_EQ(a, t.get());
  }
}

// Verify correct ctor/dtor and assignment behavior used an instrumented type.
TEST(Variant, CopyMoveConstructAssign) {
  {
    InstrumentType<int>::clear();

// Default construct to empty, no InstrumentType activity.
    Variant<int, InstrumentType<int>> v;
    ASSERT_EQ(0u, InstrumentType<int>::constructor_count());
    ASSERT_EQ(0u, InstrumentType<int>::destructor_count());
    ASSERT_EQ(0u, InstrumentType<int>::move_assignment_count());
    ASSERT_EQ(0u, InstrumentType<int>::copy_assignment_count());
  }