//
// WinRegistryTest.cpp
//
// Copyright (c) 2004-2006, Applied Informatics Software Engineering GmbH.
// and Contributors.
//
// SPDX-License-Identifier:	BSL-1.0
//


#include "WinRegistryTest.h"
#include "CppUnit/TestCaller.h"
#include "CppUnit/TestSuite.h"
#include "Poco/Util/WinRegistryKey.h"
#include "Poco/Environment.h"
#include "Poco/Exception.h"
#undef min
#undef max
#include <limits>

#if defined(POCO_HAVE_INT64)
using Poco::Int64;
#endif

using Poco::Util::WinRegistryKey;
using Poco::Environment;



WinRegistryTest::WinRegistryTest(const std::string& name): CppUnit::TestCase(name)
{
}


WinRegistryTest::~WinRegistryTest()
{
}


void WinRegistryTest::testRegistry()
{
	WinRegistryKey regKey("HKEY_CURRENT_USER\\Software\\Applied Informatics\\Test");
	if (regKey.exists()) regKey.deleteKey();
	assertTrue (!regKey.exists());

	regKey.setString("name1", "value1");
	assertTrue (regKey.getString("name1") == "value1");
	regKey.setString("name1", "Value1");
	assertTrue (regKey.getString("name1") == "Value1");
	regKey.setString("name2", "value2");
	assertTrue (regKey.getString("name2") == "value2");
	assertTrue (regKey.exists("name1"));
	assertTrue (regKey.exists("name2"));
	assertTrue (regKey.exists());

	WinRegistryKey regKeyRO("HKEY_CURRENT_USER\\Software\\Applied Informatics\\Test", true);
	assertTrue (regKeyRO.getString("name1") == "Value1");
	try
	{
		regKeyRO.setString("name1", "newValue1");
	}
	catch (Poco::Exception& exc)
	{
		std::string msg = exc.displayText();
	}
	assertTrue (regKey.getString("name1") == "Value1");

	WinRegistryKey::Values vals;
	regKey.values(vals);
	assertTrue (vals.size() == 2);
	assertTrue (vals[0] == "name1" || vals[0] == "name2");
	assertTrue (vals[1] == "name1" || vals[1] == "name2");
	assertTrue (vals[0] != vals[1]);

	Environment::set("FOO", "bar");
	regKey.setStringExpand("name3", "%FOO%");
	assertTrue (regKey.getStringExpand("name3") == "bar");

	regKey.setInt("name4", 42);
	assertTrue (regKey.getInt("name4") == 42);

	assertTrue (regKey.exists("name4"));
	regKey.deleteValue("name4");
	assertTrue (!regKey.exists("name4"));

#if defined(POCO_HAVE_INT64)
	regKey.setInt64("name5", std::numeric_limits<Int64>::max());
	assertTrue (regKey.getInt64("name5") == std::numeric_limits<Int64>::max());

	assertTrue (regKey.exists("name5"));
	regKey.deleteValue("name5");
	assertTrue (!regKey.exists("name5"));
#endif

	const int dataSize = 127;
	std::vector<char> data(dataSize);
	for (int i = 0; i < dataSize; ++i)
		data[i] = rand() % 256;
	regKey.setBinary("binary", data);
	assertTrue (regKey.getBinary("binary") == data);

	assertTrue (regKey.exists("binary"));
	regKey.deleteValue("binary");
	assertTrue (!regKey.exists("binary"));

	regKey.deleteKey();
	assertTrue (!regKey.exists());
}


void WinRegistryTest::setUp()
{
}


void WinRegistryTest::tearDown()
{
}


CppUnit::Test* WinRegistryTest::suite()
{
	CppUnit::TestSuite* pSuite = new CppUnit::TestSuite("WinRegistryTest");

	CppUnit_addTest(pSuite, WinRegistryTest, testRegistry);

	return pSuite;
}
