#pragma once

#include "fakeit/DefaultFakeit.hpp"
#include "fakeit/EventHandler.hpp"

namespace fakeit {

class TpUnitAdapter: public EventHandler {
	EventFormatter& _formatter;
public:

	class AssertionException: public std::runtime_error {
	public:
		AssertionException(std::string msg)
				: runtime_error(msg) {
		}
	};

	virtual ~TpUnitAdapter() = default;
	TpUnitAdapter(EventFormatter& formatter):_formatter(formatter){}

	virtual void handle(const UnexpectedMethodCallEvent& e) {
		throw AssertionException(_formatter.format(e));
	}

	virtual void handle(const SequenceVerificationEvent& e) {
		throw AssertionException(_formatter.format(e));
	}

	virtual void handle(const NoMoreInvocationsVerificationEvent& e) {
		throw AssertionException(_formatter.format(e));
	}
};

class TpUnitFakeit: public DefaultFakeit {

public:
	virtual ~TpUnitFakeit() = default;

	TpUnitFakeit()
			: _formatter(), _tpunitAdapter(*this) {
	}

	static TpUnitFakeit &getInstance() {
		static TpUnitFakeit instance;
		return instance;
	}

protected:

	fakeit::EventHandler &accessTestingFrameworkAdapter() override {
		return _tpunitAdapter;
	}

	EventFormatter &accessEventFormatter() override {
		return _formatter;
	}

private:

	DefaultEventFormatter _formatter;
	TpUnitAdapter _tpunitAdapter;

};

}