initial commit

tdutils/test/variant.cpp

@@ -0,0 +1,85 @@
+/*
+    This file is part of TON Blockchain Library.
+
+    TON Blockchain Library is free software: you can redistribute it and/or modify
+    it under the terms of the GNU Lesser General Public License as published by
+    the Free Software Foundation, either version 2 of the License, or
+    (at your option) any later version.
+
+    TON Blockchain Library 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 Lesser General Public License for more details.
+
+    You should have received a copy of the GNU Lesser General Public License
+    along with TON Blockchain Library.  If not, see <http://www.gnu.org/licenses/>.
+
+    Copyright 2017-2019 Telegram Systems LLP
+*/
+#include "td/utils/Slice.h"
+#include "td/utils/StringBuilder.h"
+#include "td/utils/tests.h"
+#include "td/utils/Variant.h"
+
+REGISTER_TESTS(variant);
+
+static const size_t BUF_SIZE = 1024 * 1024;
+static char buf[BUF_SIZE], buf2[BUF_SIZE];
+static td::StringBuilder sb(td::MutableSlice(buf, BUF_SIZE - 1));
+static td::StringBuilder sb2(td::MutableSlice(buf2, BUF_SIZE - 1));
+
+static td::string move_sb() {
+  auto res = sb.as_cslice().str();
+  sb.clear();
+  return res;
+}
+
+static td::string name(int id) {
+  if (id == 1) {
+    return "A";
+  }
+  if (id == 2) {
+    return "B";
+  }
+  if (id == 3) {
+    return "C";
+  }
+  return "";
+}
+
+template <int id>
+class Class {
+ public:
+  Class() {
+    sb << "+" << name(id);
+  }
+  Class(const Class &) = delete;
+  Class &operator=(const Class &) = delete;
+  Class(Class &&) = delete;
+  Class &operator=(Class &&) = delete;
+  ~Class() {
+    sb << "-" << name(id);
+  }
+};
+
+using A = Class<1>;
+using B = Class<2>;
+using C = Class<3>;
+
+TEST(Variant, simple) {
+  {
+    td::Variant<td::unique_ptr<A>, td::unique_ptr<B>, td::unique_ptr<C>> abc;
+    ASSERT_STREQ("", sb.as_cslice());
+    abc = td::make_unique<A>();
+    ASSERT_STREQ("+A", sb.as_cslice());
+    sb.clear();
+    abc = td::make_unique<B>();
+    ASSERT_STREQ("+B-A", sb.as_cslice());
+    sb.clear();
+    abc = td::make_unique<C>();
+    ASSERT_STREQ("+C-B", sb.as_cslice());
+    sb.clear();
+  }
+  ASSERT_STREQ("-C", move_sb());
+  sb.clear();
+};