move_only_return_tests.cpp

/*
 * Copyright (c) 2014 Eran Pe'er.
 *
 * This program is made available under the terms of the MIT License.
 *
 * Created on Mar 10, 2014
 */

#include <string>
#include <queue>

#include "tpunit++.hpp"
#include "fakeit.hpp"

using namespace fakeit;

struct MoveOnlyReturnTests: tpunit::TestFixture {

	class MoveOnlyType {
	public:
		int state;
		MoveOnlyType(int value) :
				state(value) {
		}
		MoveOnlyType(const MoveOnlyType&) = delete;
		MoveOnlyType(MoveOnlyType&&) = default;

		bool operator==(const MoveOnlyType& other) const {
			return (other.state == this->state);
		}

	};

	struct MoveOnlyInterface {
		virtual std::unique_ptr<std::string> returnMoveOnlyUniqueString() = 0;
		virtual MoveOnlyType returnMoveOnlyConcreteType() = 0;
		virtual std::vector<MoveOnlyType> returnVectorOfMoveOnly() = 0;
	};

	static std::vector<MoveOnlyType> constructVectorOfMoveOnly(int i) {
		std::vector<MoveOnlyType> vectorOfMoveOnly;
		vectorOfMoveOnly.emplace_back(i);
		return vectorOfMoveOnly;
	}

	MoveOnlyReturnTests() :
			tpunit::TestFixture(
					//
					TEST(MoveOnlyReturnTests::explicitStubbingReturnValuesFromTemporary),
					TEST(MoveOnlyReturnTests::explicitStubbingReturnValuesFromMove)
					//
							) {
	}

	void explicitStubbingReturnValuesFromTemporary() {
		Mock<MoveOnlyInterface> mock;

		When(Method(mock, returnMoveOnlyUniqueString)).Return(std::unique_ptr<std::string>(new std::string("value")));
		When(Method(mock, returnMoveOnlyConcreteType)).Return(MoveOnlyType(10));
		When(Method(mock, returnVectorOfMoveOnly)).Return(constructVectorOfMoveOnly(5));

		MoveOnlyInterface & i = mock.get();

		ASSERT_EQUAL("value", *i.returnMoveOnlyUniqueString());
		ASSERT_EQUAL(MoveOnlyType(10), i.returnMoveOnlyConcreteType());
		ASSERT_EQUAL(constructVectorOfMoveOnly(5), i.returnVectorOfMoveOnly());
	}

	void explicitStubbingReturnValuesFromMove() {
		Mock<MoveOnlyInterface> mock;

		MoveOnlyType c(10);
		std::unique_ptr<std::string> string(new std::string("value"));
		std::vector<MoveOnlyType> vectorOfC = constructVectorOfMoveOnly(5);

		When(Method(mock, returnMoveOnlyUniqueString)).Return(std::move(string));
		When(Method(mock, returnMoveOnlyConcreteType)).Return(std::move(c));
		When(Method(mock, returnVectorOfMoveOnly)).Return(std::move(vectorOfC));

		ASSERT_EQUAL(string, nullptr); // check move did happen
		ASSERT_TRUE(vectorOfC.empty()); // check move did happen

		MoveOnlyInterface& i = mock.get();

		ASSERT_EQUAL(std::string("value"), *i.returnMoveOnlyUniqueString());
		ASSERT_EQUAL(MoveOnlyType(10), i.returnMoveOnlyConcreteType());
		ASSERT_EQUAL(constructVectorOfMoveOnly(5), i.returnVectorOfMoveOnly());
	}
} __MoveOnlyReturnTests;