37 lines
667 B
C++
37 lines
667 B
C++
|
|
||
|
|
#include "catch.hpp"
|
||
|
|
|
||
|
|
#include "variant.hpp"
|
||
|
|
#include "variant_io.hpp"
|
||
|
|
|
||
|
|
#include <string>
|
||
|
|
|
||
|
|
template <typename T>
|
||
|
|
struct mutating_visitor
|
||
|
|
{
|
||
|
|
mutating_visitor(T& val)
|
||
|
|
: val_(val) {}
|
||
|
|
|
||
|
|
void operator()(T& val) const
|
||
|
|
{
|
||
|
|
val = val_;
|
||
|
|
}
|
||
|
|
|
||
|
|
template <typename T1>
|
||
|
|
void operator()(T1&) const
|
||
|
|
{
|
||
|
|
} // no-op
|
||
|
|
|
||
|
|
T& val_;
|
||
|
|
};
|
||
|
|
|
||
|
|
TEST_CASE("variant visitation", "[visitor][unary visitor]")
|
||
|
|
{
|
||
|
|
mapbox::util::variant<int, double, std::string> var(123);
|
||
|
|
REQUIRE(var.get<int>() == 123);
|
||
|
|
int val = 456;
|
||
|
|
const mutating_visitor<int> visitor(val);
|
||
|
|
mapbox::util::apply_visitor(visitor, var);
|
||
|
|
REQUIRE(var.get<int>() == 456);
|
||
|
|
}
|