/*************************************************************************/
/*  test_string.cpp                                                      */
/*************************************************************************/
/*                       This file is part of:                           */
/*                           GODOT ENGINE                                */
/*                    http://www.godotengine.org                         */
/*************************************************************************/
/* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur.                 */
/*                                                                       */
/* Permission is hereby granted, free of charge, to any person obtaining */
/* a copy of this software and associated documentation files (the       */
/* "Software"), to deal in the Software without restriction, including   */
/* without limitation the rights to use, copy, modify, merge, publish,   */
/* distribute, sublicense, and/or sell copies of the Software, and to    */
/* permit persons to whom the Software is furnished to do so, subject to */
/* the following conditions:                                             */
/*                                                                       */
/* The above copyright notice and this permission notice shall be        */
/* included in all copies or substantial portions of the Software.       */
/*                                                                       */
/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,       */
/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF    */
/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY  */
/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,  */
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE     */
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.                */
/*************************************************************************/
#include "ustring.h"
#include <wchar.h>
//#include "math_funcs.h"
#include <stdio.h>
#include "os/os.h"
#include "drivers/trex/regex.h"

#include "test_string.h"

namespace TestString {

bool test_1() {
	
	OS::get_singleton()->print("\n\nTest 1: Assign from cstr\n");
	
	String s = "Hello";
	
	OS::get_singleton()->print("\tExpected: Hello\n");
	OS::get_singleton()->print("\tResulted: %ls\n",s.c_str());
	
	return (wcscmp(s.c_str(),L"Hello")==0);
	
}

bool test_2() {
	
	OS::get_singleton()->print("\n\nTest 2: Assign from string (operator=)\n");
	
	String s = "Dolly";
	String t = s;
	
	OS::get_singleton()->print("\tExpected: Dolly\n");
	OS::get_singleton()->print("\tResulted: %ls\n",t.c_str());
	
	return (wcscmp(t.c_str(),L"Dolly")==0);
		
}

bool test_3() {
	
	OS::get_singleton()->print("\n\nTest 3: Assign from c-string (copycon)\n");
	
	String s("Sheep");
	String t(s);
	
	OS::get_singleton()->print("\tExpected: Sheep\n");
	OS::get_singleton()->print("\tResulted: %ls\n",t.c_str());
	
	return (wcscmp(t.c_str(),L"Sheep")==0);
		
}

bool test_4() {
	
	OS::get_singleton()->print("\n\nTest 4: Assign from c-widechar (operator=)\n");
	
	String s(L"Give me");
	
	OS::get_singleton()->print("\tExpected: Give me\n");
	OS::get_singleton()->print("\tResulted: %ls\n",s.c_str());
	
	return (wcscmp(s.c_str(),L"Give me")==0);
		
}

bool test_5() {
	
	OS::get_singleton()->print("\n\nTest 5: Assign from c-widechar (copycon)\n");
	
	String s(L"Wool");
	
	OS::get_singleton()->print("\tExpected: Wool\n");
	OS::get_singleton()->print("\tResulted: %ls\n",s.c_str());
	
	return (wcscmp(s.c_str(),L"Wool")==0);
		
}

bool test_6() {
	
	OS::get_singleton()->print("\n\nTest 6: comparisons (equal)\n");
	
	
	String s="Test Compare";
	
	OS::get_singleton()->print("\tComparing to \"Test Compare\"\n");
	
	if (! ( s=="Test Compare" ) )
		return false;
	
	if (! ( s==L"Test Compare" ) )
		return false;
	
	if (! ( s==String("Test Compare") ) )
		return false;
	
	return true;
		
}

bool test_7() {
	
	OS::get_singleton()->print("\n\nTest 7: comparisons (unequal)\n");
	
	
	String s="Test Compare";
	
	OS::get_singleton()->print("\tComparing to \"Test Compare\"\n");
	
	if (! ( s!="Peanut" ) )
		return false;
	
	if (! ( s!=L"Coconut" ) )
		return false;
	
	if (! ( s!=String("Butter") ) )
		return false;
	
	return true;
		
}

bool test_8() {
	
	OS::get_singleton()->print("\n\nTest 8: comparisons (operator<)\n");
	
	
	String s="Bees";
	
	OS::get_singleton()->print("\tComparing to \"Bees\"\n");
	
	if ( ! (s < "Elephant")  )
		return false;
	
	if ( s < L"Amber"  )
		return false;
	
	if ( s < String("Beatrix") )
		return false;
	
	return true;
		
}

bool test_9() {
	
	OS::get_singleton()->print("\n\nTest 9: Concatenation\n");
	
	
	String s;
	
	s+="Have";
	s+=' ';
	s+='a';
	s+=String(" ");
	s = s + L"Nice";
	s = s + " ";
	s = s + String("Day");
	
	OS::get_singleton()->print("\tComparing to \"Have a Nice Day\"\n");
	
	return (s == "Have a Nice Day");
		
}

bool test_10() {
	
	OS::get_singleton()->print("\n\nTest 10: Misc funcs (size/length/empty/etc)\n");
	
	if (! String("").empty())
		return false;
	
	if (String("Mellon").size() != 7)
		return false;

	if (String("Oranges").length() != 7)
		return false;
	
	return true;
		
}


bool test_11() {
	
	OS::get_singleton()->print("\n\nTest 11: Operator[]\n");
	
	String a="Kugar Sane";
	
	a[0]='S';
	a[6]='C';
	
	if (a != "Sugar Cane")
		return false;
	
	if (a[1]!='u')
		return false;
	
	return true;
		
}

bool test_12() {
	
	OS::get_singleton()->print("\n\nTest 12: case functions\n");
	
	
	String a="MoMoNgA";
	
	if (a.to_upper() != "MOMONGA")
		return false;
	
	if (a.nocasecmp_to("momonga")!=0)
		return false;
	
	return true;
		
}

bool test_13() {
	
	OS::get_singleton()->print("\n\nTest 13: UTF8\n");
	
	/* how can i embed UTF in here? */
	
	static const CharType ustr[] = { 0x304A , 0x360F, 0x3088, 0x3046, 0 };
//	static const wchar_t ustr[] = { 'P', 0xCE, 'p',0xD3, 0 };
	String s=ustr;
	
	OS::get_singleton()->print("\tUnicode: %ls\n",ustr);
	s.parse_utf8( s.utf8().get_data() );
	OS::get_singleton()->print("\tConvert/Parse UTF8: %ls\n",s.c_str());
	
	return (s==ustr);
		
}

bool test_14() {
	
	OS::get_singleton()->print("\n\nTest 14: ASCII\n");
	
	String s = L"Primero Leche";
	OS::get_singleton()->print("\tAscii: %s\n",s.ascii().get_data());
	
	String t=s.ascii().get_data();
	return (s==t);
		
}

bool test_15() {
	
	OS::get_singleton()->print("\n\nTest 15: substr\n");
	
	String s="Killer Baby";
	OS::get_singleton()->print("\tsubstr(3,4) of \"%ls\" is \"%ls\"\n",s.c_str(),s.substr(3,4).c_str());
	
	return (s.substr(3,4)=="ler ");
		
}

bool test_16() {
	
	OS::get_singleton()->print("\n\nTest 16: find\n");
	
	String s="Pretty Woman";
	OS::get_singleton()->print("\tString: %ls\n",s.c_str());
	OS::get_singleton()->print("\t\"tty\" is at %i pos.\n",s.find("tty"));
	OS::get_singleton()->print("\t\"Revenge of the Monster Truck\" is at %i pos.\n",s.find("Revenge of the Monster Truck"));
	
	if (s.find("tty")!=3)
		return false;
	
	if (s.find("Revenge of the Monster Truck")!=-1)
		return false;
	
	return true;
		
}

bool test_17() {
	
	OS::get_singleton()->print("\n\nTest 17: find no case\n");
	
	String s="Pretty Whale";
	OS::get_singleton()->print("\tString: %ls\n",s.c_str());
	OS::get_singleton()->print("\t\"WHA\" is at %i pos.\n",s.findn("WHA"));
	OS::get_singleton()->print("\t\"Revenge of the Monster SawFish\" is at %i pos.\n",s.findn("Revenge of the Monster Truck"));
	
	if (s.findn("WHA")!=7)
		return false;
	
	if (s.findn("Revenge of the Monster SawFish")!=-1)
		return false;
	
	return true;
		
}

bool test_18() {
	
	OS::get_singleton()->print("\n\nTest 18: find no case\n");
	
	String s="Pretty Whale";
	OS::get_singleton()->print("\tString: %ls\n",s.c_str());
	OS::get_singleton()->print("\t\"WHA\" is at %i pos.\n",s.findn("WHA"));
	OS::get_singleton()->print("\t\"Revenge of the Monster SawFish\" is at %i pos.\n",s.findn("Revenge of the Monster Truck"));
	
	if (s.findn("WHA")!=7)
		return false;
	
	if (s.findn("Revenge of the Monster SawFish")!=-1)
		return false;
	
	return true;
		
}

bool test_19() {
	
	OS::get_singleton()->print("\n\nTest 19: Search & replace\n");
	
	String s="Happy Birthday, Anna!";
	OS::get_singleton()->print("\tString: %ls\n",s.c_str());
	
	s=s.replace("Birthday","Halloween");
	OS::get_singleton()->print("\tReplaced Birthday/Halloween: %ls.\n",s.c_str());
	
	return (s=="Happy Halloween, Anna!");
		
}

bool test_20() {
	
	OS::get_singleton()->print("\n\nTest 20: Insertion\n");
	
	String s="Who is Frederic?";
	
	OS::get_singleton()->print("\tString: %ls\n",s.c_str());
	s=s.insert( s.find("?")," Chopin" );
	OS::get_singleton()->print("\tInserted Chopin: %ls.\n",s.c_str());
	
	return (s=="Who is Frederic Chopin?");
		
}

bool test_21() {
	
	OS::get_singleton()->print("\n\nTest 21: Number -> String\n");
	
	OS::get_singleton()->print("\tPi is %f\n",33.141593);
	OS::get_singleton()->print("\tPi String is %ls\n",String::num(3.141593).c_str());
	
	return String::num(3.141593)=="3.141593";
		
}

bool test_22() {
	
	OS::get_singleton()->print("\n\nTest 22: String -> Int\n");
	
	static const char* nums[4]={ "1237461283", "- 22", "0", " - 1123412" };
	static const int num[4]={ 1237461283, -22, 0,  -1123412 };
	
	for (int i=0;i<4;i++) {
		OS::get_singleton()->print("\tString: \"%s\" as Int is %i\n",nums[i],String(nums[i]).to_int());
		
		if (String(nums[i]).to_int()!=num[i])
			return false;
	}
	
	return true;
		
}

bool test_23() {
	
	OS::get_singleton()->print("\n\nTest 23: String -> Float\n");
	
	static const char* nums[4]={ "-12348298412.2", "0.05", "2.0002", " -0.0001" };
	static const double num[4]={ -12348298412.2, 0.05, 2.0002, -0.0001 };
	
	for (int i=0;i<4;i++) {
		OS::get_singleton()->print("\tString: \"%s\" as Float is %f\n",nums[i],String(nums[i]).to_double());
		
		if ( ABS(String(nums[i]).to_double()-num[i])>0.00001)
			return false;
	}
	
	return true;
		
}


bool test_24() {
	
	OS::get_singleton()->print("\n\nTest 24: Slicing\n");
	
	String s="Mars,Jupiter,Saturn,Uranus";
	
	const char*slices[4]={"Mars","Jupiter","Saturn","Uranus"};
	
	OS::get_singleton()->print("\tSlicing \"%ls\" by \"%s\"..\n",s.c_str(),",");
	
	for (int i=0;i<s.get_slice_count(",");i++) {
		
		OS::get_singleton()->print("\t\t%i- %ls\n",i+1,s.get_slice(",",i).c_str());
		
		
		if (s.get_slice(",",i)!=slices[i])
			return false;
	}
	
	return true;
		
}

bool test_25() {
	
	OS::get_singleton()->print("\n\nTest 25: Erasing\n");
	
	String s="Josephine is such a cute girl!";
	
	OS::get_singleton()->print("\tString: %ls\n",s.c_str());
	OS::get_singleton()->print("\tRemoving \"cute\"\n");
	
	s.erase(s.find("cute "),String("cute ").length());
	OS::get_singleton()->print("\tResult: %ls\n",s.c_str());
	
	
	return (s=="Josephine is such a girl!");
		
}

bool test_26() {

	OS::get_singleton()->print("\n\nTest 26: RegEx\n");
	RegEx regexp("(.*):(.*)");
	List<String> captures;

	bool match = regexp.match("name:password", &captures);
	printf("\tmatch: %s\n", match?"true":"false");

	printf("\t%i captures:\n", captures.size());
	List<String>::Element *I = captures.front();
	while (I) {

		printf("%ls\n", I->get().c_str());

		I = I->next();
	};
	return captures.size();
};

struct test_27_data {
	char const * data;
	char const * begin;
	bool expected;
};

bool test_27() {

	OS::get_singleton()->print("\n\nTest 26: begins_with\n");
	test_27_data tc[] = {
		{"res://foobar", "res://", true},
		{"res", "res://", false},
		{"abc", "abc", true}
	};
	size_t count = sizeof(tc) / sizeof(tc[0]);
	bool state = true;
	for (size_t i = 0;state && i < count; ++i) {
		String s = tc[i].data;
		state = s.begins_with(tc[i].begin) == tc[i].expected;
		if (state) {
			String sb = tc[i].begin;
			state = s.begins_with(sb) == tc[i].expected;
		}
		if (!state) {
			OS::get_singleton()->print("\n\t Failure on:\n\t\tstring: ", tc[i].data, "\n\t\tbegin: ", tc[i].begin, "\n\t\texpected: ", tc[i].expected ? "true" : "false", "\n");
		}
	};
	return state;
};

typedef bool (*TestFunc)(void);

TestFunc test_funcs[] = {
	
	test_1,
	test_2,
	test_3,
	test_4,
	test_5,
	test_6,
	test_7,
	test_8,
	test_9,
	test_10,
	test_11,
	test_12,
	test_13,
	test_14,
	test_15,
	test_16,
	test_17,
	test_18,
	test_19,
	test_20,
	test_21,
	test_22,
	test_23,
	test_24,
	test_25,
	test_26,
	test_27,
	0
	
};

MainLoop* test() {
	
	/** A character length != wchar_t may be forced, so the tests wont work */
	
	ERR_FAIL_COND_V( sizeof(CharType) != sizeof(wchar_t), NULL );
	
	int count=0;
	int passed=0;
	
	while(true) {
		if (!test_funcs[count])
			break;
		bool pass=test_funcs[count]();
		if (pass)
			passed++;
		OS::get_singleton()->print("\t%s\n",pass?"PASS":"FAILED");
		
		count++;
	}
	
	OS::get_singleton()->print("\n\n\n");
	OS::get_singleton()->print("*************\n");
	OS::get_singleton()->print("***TOTALS!***\n");
	OS::get_singleton()->print("*************\n");
	
	OS::get_singleton()->print("Passed %i of %i tests\n",count,passed);
	
	return NULL;
}

}