//------------------------------------------------------------------------------
//	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;

  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() == 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;

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

  CHK(clipA.LocalCount() == 1);
  CHK(clipB.LocalCount() == 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;

  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() == 1);
  CHK(clipB.LocalCount() == 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;

  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() == 1);
  CHK(clipB.LocalCount() == 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;

  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() == 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;

  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() == 1);
  CHK(clipB.SystemCount() == 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;

  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() == 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;
}