/*
This file is part of cpp-ethereum.

cpp-ethereum is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

cpp-ethereum 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 General Public License for more details.

You should have received a copy of the GNU General Public License
along with cpp-ethereum.  If not, see <http://www.gnu.org/licenses/>.
*/
/** @file whisperMessage.cpp
* @author Vladislav Gluhovsky <vlad@ethdev.com>
* @date May 2015
*/

#include <boost/test/unit_test.hpp>
#include <libwhisper/Message.h>

using namespace std;
using namespace dev;
using namespace dev::shh;

struct VerbosityHolder
{
	VerbosityHolder(int _temporaryValue) : oldLogVerbosity(g_logVerbosity) { g_logVerbosity = _temporaryValue; }
	~VerbosityHolder() { g_logVerbosity = oldLogVerbosity; }

	int oldLogVerbosity;
};

Topics createRandomTopics(unsigned int i)
{
	Topics ret;
	h256 t(i);

	for (int j = 0; j < 8; ++j)
	{
		t = sha3(t);
		ret.push_back(t);
	}

	return move(ret);
}

bytes createRandomPayload(unsigned int i)
{
	bytes ret;
	srand(i);
	int const sz = rand() % 1024;
	for (int j = 0; j < sz; ++j)
		ret.push_back(rand() % 256);

	return move(ret);
}

void comparePayloads(Message const& m1, Message const& m2)
{
	bytes const& p1 = m1.payload();
	bytes const& p2 = m2.payload();
	BOOST_REQUIRE_EQUAL(p1.size(), p2.size());

	for (size_t i = 0; i < p1.size(); ++i)
		BOOST_REQUIRE_EQUAL(p1[i], p2[i]);
}

void sealAndOpenSingleMessage(unsigned int i)
{
	Secret zero;
	Topics topics = createRandomTopics(i);
	bytes const payload = createRandomPayload(i);
	Message m1(payload);
	Envelope e = m1.seal(zero, topics, 1, 1);

	for (auto const& t: topics)
	{
		Topics singleTopic;
		singleTopic.push_back(t);
		Message m2(e, singleTopic, zero);
		comparePayloads(m1, m2);
	}
}

BOOST_AUTO_TEST_SUITE(whisperMessage)

BOOST_AUTO_TEST_CASE(seal)
{
	VerbosityHolder setTemporaryLevel(10);
	cnote << "Testing Envelope encryption...";

	for (unsigned int i = 1; i < 10; ++i)
		sealAndOpenSingleMessage(i);
}

BOOST_AUTO_TEST_SUITE_END()