mirror of
				https://github.com/ton-blockchain/ton
				synced 2025-03-09 15:40:10 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			85 lines
		
	
	
	
		
			2.2 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			85 lines
		
	
	
	
		
			2.2 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
/*
 | 
						|
    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-2020 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();
 | 
						|
};
 |