/*
 * Copyright 2014 Haiku, Inc.
 * Distributed under the terms of the MIT License.
 */


#include "CollatorTest.h"

#include <Collator.h>
#include <Locale.h>
#include <LocaleRoster.h>

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


CollatorTest::CollatorTest()
{
}


CollatorTest::~CollatorTest()
{
}


void
CollatorTest::TestSortKeys()
{
	struct Test {
		char* first;
		char* second;
		int sign[3];
	};

	BCollator collator;
	BLocaleRoster::Default()->GetDefaultLocale()->GetCollator(&collator);
	const Test tests[] = {
		{"gehen", "géhen", {0, -1, -1}},
		{"aus", "äUß", {-1, -1, -1}},
		{"auss", "äUß", {0, -1, -1}},
		{"WO", "wÖ", {0, -1, -1}},
		{"SO", "so", {0, 0, 1}},
		{"açñ", "acn", {0, 1, 1}},
		{NULL, NULL, {0, 0, 0}}
	};
	
	for (int32 i = 0; tests[i].first != NULL; i++) {
		NextSubTest();

		for (int32 strength = B_COLLATE_PRIMARY; strength < 4; strength++) {
			BString a, b;
			collator.SetStrength(strength);
			collator.GetSortKey(tests[i].first, &a);
			collator.GetSortKey(tests[i].second, &b);

			int difference = collator.Compare(tests[i].first, tests[i].second);
			CPPUNIT_ASSERT_EQUAL(tests[i].sign[strength - 1], difference);
			int keydiff = strcmp(a.String(), b.String());
			// Check that the keys compare the same as the strings. Either both
			// are 0, or both have the same sign.
			if (difference == 0)
				CPPUNIT_ASSERT_EQUAL(0, keydiff);
			else
				CPPUNIT_ASSERT(keydiff * difference > 0);
		}
	}
}


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

	suite.addTest(new CppUnit::TestCaller<CollatorTest>(
		"CollatorTest::TestSortKeys", &CollatorTest::TestSortKeys));

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