/*
 * Copyright 2017, Andrew Lindesay <apl@lindesay.co.nz>
 * Distributed under the terms of the MIT License.
 */
#include "JsonToMessageTest.h"

#include <Json.h>

#include <cppunit/TestCaller.h>
#include <cppunit/TestSuite.h>

#include "JsonSamples.h"


#define LOOPS_TEST_OBJECT_A_FOR_PERFORMANCE 100000

using namespace BPrivate;


JsonToMessageTest::JsonToMessageTest()
{
}


JsonToMessageTest::~JsonToMessageTest()
{
}


void
JsonToMessageTest::TestArrayA()
{
	BMessage message;
	BMessage subMessage;
	BString stringValue;
	double doubleValue;

	// ----------------------
	BJson::Parse(JSON_SAMPLE_ARRAY_A_IN, message);
	// ----------------------

	CPPUNIT_ASSERT_EQUAL_MESSAGE("!find [0]", B_OK,
		message.FindString("0", &stringValue));
	CPPUNIT_ASSERT_EQUAL_MESSAGE("!eq [0]", BString("1234"), stringValue);

	CPPUNIT_ASSERT_EQUAL_MESSAGE("!find [1]", B_OK,
		message.FindDouble("1", &doubleValue));
	CPPUNIT_ASSERT_EQUAL_MESSAGE("!eq [1]", 4567, doubleValue);

	CPPUNIT_ASSERT_EQUAL_MESSAGE("!find [2]", B_OK,
		message.FindMessage("2", &subMessage));

	CPPUNIT_ASSERT_EQUAL_MESSAGE("!find [2.0]", B_OK,
		subMessage.FindString("0", &stringValue));
	CPPUNIT_ASSERT_EQUAL_MESSAGE("!eq [2.0]", BString("A"), stringValue);

	CPPUNIT_ASSERT_EQUAL_MESSAGE("!find [2.1]", B_OK,
		subMessage.FindString("1", &stringValue));
	CPPUNIT_ASSERT_EQUAL_MESSAGE("!eq [2.1]", BString("b"), stringValue);

	CPPUNIT_ASSERT_EQUAL_MESSAGE("!find [2.2]", B_OK,
		subMessage.FindString("2", &stringValue));
	CPPUNIT_ASSERT_EQUAL_MESSAGE("!eq [2.2]", BString("C\xc3\xa9zanne"),
		stringValue);
}


void
JsonToMessageTest::TestArrayB()
{
	BMessage message;
	BMessage subMessage;
	BMessage subSubMessage;
	BString stringValue;

	// ----------------------
	BJson::Parse(JSON_SAMPLE_ARRAY_B_IN, message);
	// ----------------------

	CPPUNIT_ASSERT_EQUAL_MESSAGE("!find [0]", B_OK,
		message.FindString("0", &stringValue));
	CPPUNIT_ASSERT_EQUAL_MESSAGE("!eq [0]", BString("Whirinaki"), stringValue);

	CPPUNIT_ASSERT_EQUAL_MESSAGE("!find [1]", B_OK,
		message.FindString("1", &stringValue));
	CPPUNIT_ASSERT_EQUAL_MESSAGE("!eq [1]", BString("Wellington"), stringValue);

	CPPUNIT_ASSERT_EQUAL_MESSAGE("!find [2]", B_OK,
		message.FindMessage("2", &subMessage));

	CPPUNIT_ASSERT_EQUAL_MESSAGE("!find [2.0]", B_OK,
		subMessage.FindMessage("key", &subSubMessage));

	CPPUNIT_ASSERT_EQUAL_MESSAGE("!find [2.0.0]", B_OK,
		subSubMessage.FindString("0", &stringValue));
	CPPUNIT_ASSERT_EQUAL_MESSAGE("!eq [2.0.0]", BString("value"), stringValue);
}


void
JsonToMessageTest::TestObjectA()
{
	BMessage message;
	BMessage subMessage;
	BString stringValue;

	// ----------------------
	BJson::Parse(JSON_SAMPLE_OBJECT_A_IN, message);
	// ----------------------

	CPPUNIT_ASSERT_EQUAL(B_OK, message.FindString("weather", &stringValue));
	CPPUNIT_ASSERT_EQUAL(BString("raining"), stringValue);

	CPPUNIT_ASSERT_EQUAL(B_OK, message.FindString("humidity", &stringValue));
	CPPUNIT_ASSERT_EQUAL(BString("too-high"), stringValue);

	CPPUNIT_ASSERT_EQUAL(B_OK, message.FindMessage("daysOfWeek", &subMessage));

	CPPUNIT_ASSERT_EQUAL(B_OK, subMessage.FindString("0", &stringValue));
	CPPUNIT_ASSERT_EQUAL(BString("MON"), stringValue);

	CPPUNIT_ASSERT_EQUAL(B_OK, subMessage.FindString("1", &stringValue));
	CPPUNIT_ASSERT_EQUAL(BString("TUE"), stringValue);

	CPPUNIT_ASSERT_EQUAL(B_OK, subMessage.FindString("2", &stringValue));
	CPPUNIT_ASSERT_EQUAL(BString("WED"), stringValue);

	CPPUNIT_ASSERT_EQUAL(B_OK, subMessage.FindString("3", &stringValue));
	CPPUNIT_ASSERT_EQUAL(BString("THR"), stringValue);

	CPPUNIT_ASSERT_EQUAL(B_OK, subMessage.FindString("4", &stringValue));
	CPPUNIT_ASSERT_EQUAL(BString("FRI"), stringValue);
}


/*! This is not a real test, but is a convenient point at which to implement a
    performance test.
*/

void
JsonToMessageTest::TestObjectAForPerformance()
{
	int i;

	for (i=0; i < LOOPS_TEST_OBJECT_A_FOR_PERFORMANCE; i++)
		TestObjectA();
}


void
JsonToMessageTest::TestObjectB()
{
	BMessage message;
	bool boolValue;
	void *ptrValue; // this is how NULL is represented.

	// ----------------------
	BJson::Parse(JSON_SAMPLE_OBJECT_B_IN, message);
	// ----------------------

	CPPUNIT_ASSERT_EQUAL(B_OK, message.FindBool("testTrue", &boolValue));
	CPPUNIT_ASSERT_EQUAL(true, boolValue);

	CPPUNIT_ASSERT_EQUAL(B_OK, message.FindBool("testFalse", &boolValue));
	CPPUNIT_ASSERT_EQUAL(false, boolValue);

	CPPUNIT_ASSERT_EQUAL(B_OK, message.FindPointer("testNull", &ptrValue));
	CPPUNIT_ASSERT_EQUAL(0, (addr_t)ptrValue);
}


void
JsonToMessageTest::TestObjectC()
{
	BMessage message;
	BMessage subMessage;
	BString stringValue;

	// ----------------------
	BJson::Parse(JSON_SAMPLE_OBJECT_C_IN, message);
	// ----------------------

	CPPUNIT_ASSERT_EQUAL(B_OK, message.FindString("key", &stringValue));
	CPPUNIT_ASSERT_EQUAL(BString("value"), stringValue);
}


void
JsonToMessageTest::TestUnterminatedObject()
{
	BMessage message;

	// ----------------------
	status_t result = BJson::Parse(JSON_SAMPLE_BROKEN_UNTERMINATED_OBJECT,
		message);
	// ----------------------

	CPPUNIT_ASSERT_EQUAL(B_BAD_DATA, result);
}


void
JsonToMessageTest::TestUnterminatedArray()
{
	BMessage message;

	// ----------------------
	status_t result = BJson::Parse(JSON_SAMPLE_BROKEN_UNTERMINATED_ARRAY,
		message);
	// ----------------------

	CPPUNIT_ASSERT_EQUAL(B_BAD_DATA, result);
}


void JsonToMessageTest::TestHaikuDepotFetchBatch()
{
	const unsigned char input[] = JSON_SAMPLE_HDS_FETCH_BATCH_PKGS;
	BMessage message;
	BMessage resultMessage;
	BMessage pkgsMessage;
	BMessage pkgMessage1;
	double modifyTimestampDouble;
	double expectedModifyTimestampDouble = 1488785331631.0;

	// ----------------------
	status_t result = BJson::Parse((char*)input,
		message);
	// ----------------------

	CPPUNIT_ASSERT_EQUAL(B_OK, result);

		// this is quite a large test input so a random sample "thing to
		// check" is chosen to indicate that the parse was successful.

	CPPUNIT_ASSERT_EQUAL_MESSAGE("result", B_OK,
		message.FindMessage("result", &resultMessage));
	CPPUNIT_ASSERT_EQUAL_MESSAGE("pkgs", B_OK,
		resultMessage.FindMessage("pkgs", &pkgsMessage));
	CPPUNIT_ASSERT_EQUAL_MESSAGE("1", B_OK,
		pkgsMessage.FindMessage("1", &pkgMessage1));
	CPPUNIT_ASSERT_EQUAL_MESSAGE("modifyTimestamp", B_OK,
		pkgMessage1.FindDouble("modifyTimestamp", &modifyTimestampDouble));

	CPPUNIT_ASSERT_EQUAL(expectedModifyTimestampDouble, modifyTimestampDouble);
}


/*static*/ void
JsonToMessageTest::AddTests(BTestSuite& parent)
{
	CppUnit::TestSuite& suite = *new CppUnit::TestSuite(
		"JsonToMessageTest");

	suite.addTest(new CppUnit::TestCaller<JsonToMessageTest>(
		"JsonToMessageTest::TestArrayA",
		&JsonToMessageTest::TestArrayA));

	suite.addTest(new CppUnit::TestCaller<JsonToMessageTest>(
		"JsonToMessageTest::TestArrayB",
		&JsonToMessageTest::TestArrayB));

	suite.addTest(new CppUnit::TestCaller<JsonToMessageTest>(
		"JsonToMessageTest::TestObjectA",
		&JsonToMessageTest::TestObjectA));

	suite.addTest(new CppUnit::TestCaller<JsonToMessageTest>(
		"JsonToMessageTest::TestObjectB",
		&JsonToMessageTest::TestObjectB));

	suite.addTest(new CppUnit::TestCaller<JsonToMessageTest>(
		"JsonToMessageTest::TestObjectC",
		&JsonToMessageTest::TestObjectC));

	suite.addTest(new CppUnit::TestCaller<JsonToMessageTest>(
		"JsonToMessageTest::TestUnterminatedObject",
		&JsonToMessageTest::TestUnterminatedObject));

	suite.addTest(new CppUnit::TestCaller<JsonToMessageTest>(
		"JsonToMessageTest::TestUnterminatedArray",
		&JsonToMessageTest::TestUnterminatedArray));

	suite.addTest(new CppUnit::TestCaller<JsonToMessageTest>(
		"JsonToMessageTest::TestHaikuDepotFetchBatch",
		&JsonToMessageTest::TestHaikuDepotFetchBatch));

//	suite.addTest(new CppUnit::TestCaller<JsonToMessageTest>(
//		"JsonToMessageTest::TestObjectAForPerformance",
//		&JsonToMessageTest::TestObjectAForPerformance));

	parent.addTest("JsonToMessageTest", &suite);
}