//------------------------------------------------------------------------------
//	CountTester.cpp
//
//------------------------------------------------------------------------------

// Standard Includes -----------------------------------------------------------

// System Includes -------------------------------------------------------------
#include <Application.h>
#include <Clipboard.h>

#define CHK	CPPUNIT_ASSERT

// Project Includes ------------------------------------------------------------

// Local Includes --------------------------------------------------------------
#include "CountTester.h"

// Local Defines ---------------------------------------------------------------

// Globals ---------------------------------------------------------------------

//------------------------------------------------------------------------------

/*
	LocalCount()
	@case 1
	@results		count == 0
 */
void CountTester::LocalCount1()
{
	BApplication app("application/x-vnd.clipboardtest");
	BClipboard clip("LocalCount1");

	CHK(clip.LocalCount() == 0);
}

/*
	LocalCount()
	@case 2
	@results		count == +1
 */
void CountTester::LocalCount2()
{
	BApplication app("application/x-vnd.clipboardtest");
	BClipboard clip("LocalCount2");
	BMessage *data;

	uint32 oldCount = clip.SystemCount();

	if (clip.Lock()) {
		clip.Clear();
		if ((data = clip.Data())) {
			data->AddData("text/plain", B_MIME_TYPE, "LocalCount2", 12);
			clip.Commit();
		}
		clip.Unlock();
	}

	CHK(clip.LocalCount() == oldCount + 1);
}

/*
	LocalCount()
	@case 3
	@results		both counts == +1
 */
void CountTester::LocalCount3()
{
	BApplication app("application/x-vnd.clipboardtest");
	BClipboard clipA("LocalCount3");
	BClipboard clipB("LocalCount3");
	BMessage *data;

	uint32 oldCount = clipA.SystemCount();

	CHK(clipB.Lock());
	clipB.Clear();
	if ((data = clipB.Data())) {
		data->AddData("text/plain", B_MIME_TYPE, "LocalCount3", 12);
		clipB.Commit();
	}
	clipB.Unlock();

	CHK(clipA.Lock());
	clipA.Unlock();

	CHK(clipA.LocalCount() == oldCount + 1);
	CHK(clipB.LocalCount() == oldCount + 1);
}

/*
	LocalCount()
	@case 4
	@results	clipA.LocalCount() == +1
				clipB.LocalCount() == +2
 */
void CountTester::LocalCount4()
{
	BApplication app("application/x-vnd.clipboardtest");
	BClipboard clipA("LocalCount4");
	BClipboard clipB("LocalCount4");
	BMessage *data;

	uint32 oldCount = clipA.SystemCount();

	if (clipB.Lock()) {
		clipB.Clear();
		if ((data = clipB.Data())) {
			data->AddData("text/plain", B_MIME_TYPE, "LocalCount4", 12);
			clipB.Commit();
		}
		clipB.Unlock();
	}

	if (clipA.Lock())
		clipA.Unlock();

	if (clipB.Lock()) {
		clipB.Clear();
		if ((data = clipB.Data())) {
			data->AddData("text/plain",B_MIME_TYPE,"LocalCount4",12);
			clipB.Commit();
		}
		clipB.Unlock();
	}

	CHK(clipA.LocalCount() == oldCount + 1);
	CHK(clipB.LocalCount() == oldCount + 2);
}

/*
	LocalCount()
	@case 5
	@results	clipA.LocalCount() == 1
				clipB.LocalCount() == 2
 */
void CountTester::LocalCount5()
{
	BApplication app("application/x-vnd.clipboardtest");
	BClipboard clipA("LocalCount5");
	BClipboard clipB("LocalCount5");
	BMessage *data;

	uint32 oldCount = clipA.SystemCount();

	if (clipA.Lock()) {
		clipA.Clear();
		if ((data = clipA.Data())) {
			data->AddData("text/plain", B_MIME_TYPE, "LocalCount5", 12);
			clipA.Commit();
		}
		clipA.Unlock();
	}

	if (clipB.Lock()) {
		clipB.Clear();
		if ((data = clipB.Data())) {
			data->AddData("text/plain", B_MIME_TYPE, "LocalCount5", 12);
			clipB.Commit();
		}
		clipB.Unlock();
	}

	CHK(clipA.LocalCount() == oldCount + 1);
	CHK(clipB.LocalCount() == oldCount + 2);
}

/*
	LocalCount()
	@case 6
	@results	clipA.LocalCount() == +1
				clipB.LocalCount() == 0
 */
void CountTester::LocalCount6()
{
	BApplication app("application/x-vnd.clipboardtest");
	BClipboard clipA("LocalCount6A");
	BClipboard clipB("LocalCount6B");
	BMessage *data;

	uint32 oldCount = clipA.SystemCount();

	if (clipA.Lock()) {
		clipA.Clear();
		if ((data = clipA.Data())) {
			data->AddData("text/plain", B_MIME_TYPE, "LocalCount6" ,12);
			clipA.Commit();
		}
		clipA.Unlock();
	}

	if (clipB.Lock())
		clipB.Unlock();

	CHK(clipA.LocalCount() == oldCount + 1);
	CHK(clipB.LocalCount() == 0);
}

/*
	SystemCount()
	@case 1
	@results		count == 0
 */
void CountTester::SystemCount1()
{
	BApplication app("application/x-vnd.clipboardtest");
	BClipboard clip("SystemCount1");

	CHK(clip.SystemCount() == 0);
}

/*
	SystemCount()
	@case 2
	@results	clipA.SystemCount() == +1
				clipB.SystemCount() == +1
 */
void CountTester::SystemCount2()
{
	BApplication app("application/x-vnd.clipboardtest");
	BClipboard clipA("SystemCount2");
	BClipboard clipB("SystemCount2");
	BMessage *data;

	uint32 oldCount = clipA.SystemCount();

	if (clipA.Lock()) {
		clipA.Clear();
		if ((data = clipA.Data())) {
			data->AddData("text/plain", B_MIME_TYPE, "SystemCount2", 12);
			clipA.Commit();
		}
		clipA.Unlock();
	}

	CHK(clipA.SystemCount() == oldCount + 1);
	CHK(clipB.SystemCount() == oldCount + 1);
}

/*
	SystemCount()
	@case 3
	@results	clipA.SystemCount() == +1
				clipB.SystemCount() == 0
 */
void CountTester::SystemCount3()
{
	BApplication app("application/x-vnd.clipboardtest");
	BClipboard clipA("SystemCount3A");
	BClipboard clipB("SystemCount3B");
	BMessage *data;

	uint32 oldCount = clipA.SystemCount();

	if (clipA.Lock()) {
		clipA.Clear();
		if ((data = clipA.Data())) {
			data->AddData("text/plain", B_MIME_TYPE, "SystemCount3", 12);
			clipA.Commit();
		}
		clipA.Unlock();
	}

	CHK(clipA.SystemCount() == oldCount + 1);
	CHK(clipB.SystemCount() == 0);
}

Test* CountTester::Suite()
{
	TestSuite* SuiteOfTests = new TestSuite;

	ADD_TEST4(BClipboard, SuiteOfTests, CountTester, LocalCount1);
	ADD_TEST4(BClipboard, SuiteOfTests, CountTester, LocalCount2);
	ADD_TEST4(BClipboard, SuiteOfTests, CountTester, LocalCount3);
	ADD_TEST4(BClipboard, SuiteOfTests, CountTester, LocalCount4);
	ADD_TEST4(BClipboard, SuiteOfTests, CountTester, LocalCount5);
	ADD_TEST4(BClipboard, SuiteOfTests, CountTester, LocalCount6);
	ADD_TEST4(BClipboard, SuiteOfTests, CountTester, SystemCount1);
	ADD_TEST4(BClipboard, SuiteOfTests, CountTester, SystemCount2);
	ADD_TEST4(BClipboard, SuiteOfTests, CountTester, SystemCount3);

	return SuiteOfTests;
}