// xestiaab_contactload.h - Xestia Address Book Contact Load Unit Tests.
//
// (c) 2012-2015 Xestia Software Development.
//
// This file is part of Xestia Address Book.
//
// Xestia Address Book is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by the
// Free Software Foundation, version 3 of the license.
//
// Xestia Address Book is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License along
// with Xestia Address Book. If not, see <http://www.gnu.org/licenses/>

#include <gtest/gtest.h>
#include "../contacteditor/ContactDataObject.h"

TEST(ContactLoad, ContactLoadTests){

	ContactDataObject TestFile;

	// Check that the file given is not missing.
	
	ASSERT_EQ(CONTACTLOAD_FILEMISSING, TestFile.LoadFile("MissingFile.vcf"));
	
	// Check that the file given does not have an error when attempting to
	// open it. 
	
	ASSERT_EQ(CONTACTLOAD_FILEERROR, TestFile.LoadFile("InvalidPermissions.vcf"));
	
	// Check that the file is a valid vCard 4.0 file.
	
	ASSERT_EQ(CONTACTLOAD_FILEINVALIDFORMAT, TestFile.LoadFile("LoadCheck-Load1-v3.vcf"));

	// Check that the vCard 4.0 file meets the base specification.
	
	ASSERT_EQ(CONTACTLOAD_FILEBASESPECFAIL, TestFile.LoadFile("LoadCheck-Load3.vcf"));

	// Check that the vCard 4.0 file loads OK.

	ASSERT_EQ(CONTACTLOAD_OK, TestFile.LoadFile("LoadCheck-Load4.vcf"));
	
	// Check that the kind status has been set. (KIND).
	
	ASSERT_NE(CONTACTKIND_NONE, TestFile.ContactKind);
	ASSERT_EQ(CONTACTKIND_INDIVIDUAL, TestFile.ContactKind);
	
	// Check that the groups have been read (MEMBER).
	
	std::map<int,wxString>::iterator TestFileIter;
	std::map<int,int>::iterator TestFileIntIter;
	
	TestFileIter = TestFile.GroupsList.find(0);
	
	ASSERT_NE(TestFile.GroupsList.end(), TestFileIter);
	ASSERT_EQ("7a2af44d-6431-4797-a55f-d86d56304fda", TestFileIter->second);

	// Check the ALTID parameter.

	TestFileIter = TestFile.GroupsListAltID.find(0);
	ASSERT_NE(TestFile.GroupsListAltID.end(), TestFileIter);
	ASSERT_EQ("35", TestFileIter->second);

	// Check the PID parameter.
	
	TestFileIter = TestFile.GroupsListPID.find(0);
	ASSERT_NE(TestFile.GroupsListPID.end(), TestFileIter);
	ASSERT_EQ("40", TestFileIter->second);
	
	// Check the PREF parameter.
	
	TestFileIntIter = TestFile.GroupsListPref.find(0);
	ASSERT_NE(TestFile.GroupsListPref.end(), TestFileIntIter);
	ASSERT_EQ(45, TestFileIntIter->second);
	
	// Check the MEDIATYPE parameter.
	
	TestFileIter = TestFile.GroupsListMediaType.find(0);
	ASSERT_NE(TestFile.GroupsListMediaType.end(), TestFileIter);
	ASSERT_EQ("text/plain", TestFileIter->second);
	
	// Check the extra tokens parameter.
	
	TestFileIter = TestFile.GroupsListTokens.find(0);
	ASSERT_NE(TestFile.GroupsListTokens.end(), TestFileIter);
	ASSERT_EQ("EXAMPLE=Meep", TestFileIter->second);
	
	// Check that the formatted name has been read (FN).
	
	TestFileIter = TestFile.FullNamesList.find(0);
	ASSERT_NE(TestFile.FullNamesList.end(), TestFileIter);
	ASSERT_EQ("Test Contact", TestFileIter->second);
	
	// Check the TYPE parameter.

	TestFileIter = TestFile.FullNamesListType.find(0);
	ASSERT_NE(TestFile.FullNamesListType.end(), TestFileIter);
	ASSERT_EQ("work", TestFileIter->second);
	
	// Check the LANGUAGE parameter.
	
	TestFileIter = TestFile.FullNamesListLanguage.find(0);
	ASSERT_NE(TestFile.FullNamesListLanguage.end(), TestFileIter);
	ASSERT_EQ("kw", TestFileIter->second);
	
	// Check the ALTID parameter.
	
	TestFileIter = TestFile.FullNamesListAltID.find(0);
	ASSERT_NE(TestFile.FullNamesListAltID.end(), TestFileIter);
	ASSERT_EQ("500", TestFileIter->second);

	// Check the PID parameter.

	TestFileIter = TestFile.FullNamesListPID.find(0);
	ASSERT_NE(TestFile.FullNamesListPID.end(), TestFileIter);
	ASSERT_EQ("40", TestFileIter->second);
	
	// Check the PREF parameter.
	
	TestFileIntIter = TestFile.FullNamesListPref.find(0);
	ASSERT_NE(TestFile.FullNamesListPref.end(), TestFileIntIter);
	ASSERT_EQ(45, TestFileIntIter->second);
	
	// Check for the extra tokens parameter.
	
	TestFileIter = TestFile.FullNamesListTokens.find(0);
	ASSERT_NE(TestFile.FullNamesListTokens.end(), TestFileIter);
	ASSERT_EQ("EXAMPLE=Baaa", TestFileIter->second);
	
	// Check that the name has been read (N).
	
	ASSERT_EQ("Contact", TestFile.NameSurname);
	ASSERT_EQ("Test", TestFile.NameForename);
	ASSERT_EQ("Example", TestFile.NameOtherNames);
	ASSERT_EQ("Robot", TestFile.NameTitle);
	ASSERT_EQ("LOONY", TestFile.NameSuffix);
	
	// Check the ALTID parameter.
	
	ASSERT_EQ("3", TestFile.NameAltID);
	
	// Check the LANGUAGE parameter. 
	
	ASSERT_EQ("kw", TestFile.NameLanguage);
	
	// Check the SORT-AS parameter.
	 
	ASSERT_EQ("Contact,Test", TestFile.NameDisplayAs);

	// Check the extra tokens parameter.
	
	ASSERT_EQ("TEST=Yes", TestFile.NameTokens);

}

TEST(ContactLoad, NicknameTests){

	ContactDataObject TestFile;

	// Check that the vCard 4.0 file loads OK.

	ASSERT_EQ(CONTACTLOAD_OK, TestFile.LoadFile("LoadCheck-Load4.vcf"));

	std::map<int,wxString>::iterator TestFileIter;
	std::map<int,int>::iterator TestFileIntIter;

	// Check that the nickname has been read (NICKNAME). General.
	
	TestFileIter = TestFile.GeneralNicknamesList.find(0);
	ASSERT_NE(TestFile.GeneralNicknamesList.end(), TestFileIter);	
	ASSERT_EQ("Tester", TestFileIter->second);
	
	// Check the ALTID parameter.
	
	TestFileIter = TestFile.GeneralNicknamesListAltID.find(0);
	ASSERT_NE(TestFile.GeneralNicknamesListAltID.end(), TestFileIter);
	ASSERT_EQ("35", TestFileIter->second);
	
	// Check the PID parameter.
	
	TestFileIter = TestFile.GeneralNicknamesListPID.find(0);
	ASSERT_NE(TestFile.GeneralNicknamesListPID.end(), TestFileIter);
	ASSERT_EQ("50", TestFileIter->second);
	
	// Check the PREF parameter.

	TestFileIntIter = TestFile.GeneralNicknamesListPref.find(0);
	ASSERT_NE(TestFile.GeneralNicknamesListPref.end(), TestFileIntIter);
	ASSERT_EQ(40, TestFileIntIter->second);

	// Check the LANGUAGE parameter.

	TestFileIter = TestFile.GeneralNicknamesListLanguage.find(0);
	ASSERT_NE(TestFile.GeneralNicknamesListLanguage.end(), TestFileIter);
	ASSERT_EQ("kw", TestFileIter->second);
	
	// Check the extra tokens parameter.
	
	TestFileIter = TestFile.GeneralNicknamesListTokens.find(0);
	ASSERT_NE(TestFile.GeneralNicknamesListTokens.end(), TestFileIter);
	ASSERT_EQ("YAY=Nope;Beep=Boop", TestFileIter->second);
	
	// Repeat for the home type.
	
	TestFileIter = TestFile.HomeNicknamesList.find(1);
	ASSERT_NE(TestFile.HomeNicknamesList.end(), TestFileIter);	
	ASSERT_EQ("Testy", TestFileIter->second);
	
	// Check the ALTID parameter.
	
	TestFileIter = TestFile.HomeNicknamesListAltID.find(1);
	ASSERT_NE(TestFile.HomeNicknamesListAltID.end(), TestFileIter);
	ASSERT_EQ("17", TestFileIter->second);
	
	// Check the PID parameter.
	
	TestFileIter = TestFile.HomeNicknamesListPID.find(1);
	ASSERT_NE(TestFile.HomeNicknamesListPID.end(), TestFileIter);
	ASSERT_EQ("39", TestFileIter->second);
	
	// Check the PREF parameter.

	TestFileIntIter = TestFile.HomeNicknamesListPref.find(1);
	ASSERT_NE(TestFile.HomeNicknamesListPref.end(), TestFileIntIter);
	ASSERT_EQ(78, TestFileIntIter->second);

	// Check the LANGUAGE parameter.

	TestFileIter = TestFile.HomeNicknamesListLanguage.find(1);
	ASSERT_NE(TestFile.HomeNicknamesListLanguage.end(), TestFileIter);
	ASSERT_EQ("en", TestFileIter->second);
	
	// Check the extra tokens parameter.
	
	TestFileIter = TestFile.HomeNicknamesListTokens.find(1);
	ASSERT_NE(TestFile.HomeNicknamesListTokens.end(), TestFileIter);
	ASSERT_EQ("YAY=Yep;Beep=Boop", TestFileIter->second);
	
	// Repeat for the work type.
	
	TestFileIter = TestFile.BusinessNicknamesList.find(2);
	ASSERT_NE(TestFile.BusinessNicknamesList.end(), TestFileIter);	
	ASSERT_EQ("The Testing One", TestFileIter->second);
	
	// Check the ALTID parameter.
	
	TestFileIter = TestFile.BusinessNicknamesListAltID.find(2);
	ASSERT_NE(TestFile.BusinessNicknamesListAltID.end(), TestFileIter);
	ASSERT_EQ("99", TestFileIter->second);
	
	// Check the PID parameter.
	
	TestFileIter = TestFile.BusinessNicknamesListPID.find(2);
	ASSERT_NE(TestFile.BusinessNicknamesListPID.end(), TestFileIter);
	ASSERT_EQ("10", TestFileIter->second);
	
	// Check the PREF parameter.

	TestFileIntIter = TestFile.BusinessNicknamesListPref.find(2);
	ASSERT_NE(TestFile.BusinessNicknamesListPref.end(), TestFileIntIter);
	ASSERT_EQ(1, TestFileIntIter->second);

	// Check the LANGUAGE parameter.

	TestFileIter = TestFile.BusinessNicknamesListLanguage.find(2);
	ASSERT_NE(TestFile.BusinessNicknamesListLanguage.end(), TestFileIter);
	ASSERT_EQ("en-GB", TestFileIter->second);
	
	// Check the extra tokens parameter.
	
	TestFileIter = TestFile.BusinessNicknamesListTokens.find(2);
	ASSERT_NE(TestFile.BusinessNicknamesListTokens.end(), TestFileIter);
	ASSERT_EQ("YAY=Maybe;Boop=Boing", TestFileIter->second);

}

TEST(ContactLoad, GenderTests){

	// Check that the gender has been read.

	ContactDataObject TestFile;

	// Check that the vCard 4.0 file loads OK.

	ASSERT_EQ(CONTACTLOAD_OK, TestFile.LoadFile("LoadCheck-Load4.vcf"));

	std::map<int,wxString>::iterator TestFileIter;
	std::map<int,int>::iterator TestFileIntIter;
	
	ASSERT_NE("", TestFile.Gender);
	ASSERT_EQ("F", TestFile.Gender);
	ASSERT_EQ("Example Text", TestFile.GenderDetails);
	ASSERT_EQ("BEEP=Boop", TestFile.GenderTokens);

}

TEST(ContactLoad, BirthdayTests){

	// Check that the birthday has been read.

	// Check that the gender has been read.

	ContactDataObject TestFile;

	// Check that the vCard 4.0 file loads OK.

	ASSERT_EQ(CONTACTLOAD_OK, TestFile.LoadFile("LoadCheck-Load4.vcf"));

	std::map<int,wxString>::iterator TestFileIter;
	std::map<int,int>::iterator TestFileIntIter;
	
	ASSERT_NE("", TestFile.Birthday);
	ASSERT_EQ("20040101", TestFile.Birthday);
	
	// Check the ALTID parameter.
	
	ASSERT_EQ("35", TestFile.BirthdayAltID);
	
	// Check the CALSCALE parameter.
	
	ASSERT_EQ("georgian", TestFile.BirthdayCalScale);
	
	// Check the extra tokens parameter.

	ASSERT_EQ("HAPPY=Days", TestFile.BirthdayTokens);
	
	// Check the VALUE parameter. (Note have to use a different file
	// due to the *1 rule in RFC6350.
	
	ContactDataObject TestFileValue;
	ASSERT_EQ(CONTACTLOAD_OK, TestFile.LoadFile("LoadCheck-Load4-BDayValue.vcf"));
	
	ASSERT_EQ("Circa 2000s", TestFile.Birthday);
	
}

TEST(ContactLoad, AnniversaryTests){

	// Check that the birthday has been read.

	// Check that the gender has been read.

	ContactDataObject TestFile;

	// Check that the vCard 4.0 file loads OK.

	ASSERT_EQ(CONTACTLOAD_OK, TestFile.LoadFile("LoadCheck-Load4.vcf"));

	std::map<int,wxString>::iterator TestFileIter;
	std::map<int,int>::iterator TestFileIntIter;
	
	ASSERT_NE("", TestFile.Anniversary);
	ASSERT_EQ("20200516", TestFile.Anniversary);
	
	// Check the ALTID parameter.
	
	ASSERT_EQ("70", TestFile.AnniversaryAltID);
	
	// Check the CALSCALE parameter.
	
	ASSERT_EQ("georgian", TestFile.AnniversaryCalScale);
	
	// Check the extra tokens parameter.

	ASSERT_EQ("WONDERFUL=Day", TestFile.AnniversaryTokens);
	
	// Check the VALUE parameter. (Note have to use a different file
	// due to the *1 rule in RFC6350.
	
	ContactDataObject TestFileValue;
	ASSERT_EQ(CONTACTLOAD_OK, TestFile.LoadFile("LoadCheck-Load4-BDayValue.vcf"));
	
	ASSERT_EQ("Circa 2020s", TestFile.Anniversary);	

}

// TODO: Add tests for the Contact Loading process.