From c3e1459a7c6450a54494a47147b4bae91d277a0f Mon Sep 17 00:00:00 2001
From: Steve Brokenshire <sbrokenshire@xestia.co.uk>
Date: Sun, 31 Jan 2016 13:25:36 +0000
Subject: [PATCH] Added unit tests for CATEGORIES and COMMENT in
 iCalendarTask/ObjectDataTests.

---
 source/tests/xestiacalendar_icaltaskload.h | 124 +++++++++++++++++++++
 1 file changed, 124 insertions(+)

diff --git a/source/tests/xestiacalendar_icaltaskload.h b/source/tests/xestiacalendar_icaltaskload.h
index 208f2d4..318c6b6 100644
--- a/source/tests/xestiacalendar_icaltaskload.h
+++ b/source/tests/xestiacalendar_icaltaskload.h
@@ -396,4 +396,128 @@ TEST(iCalendarTask, ObjectDataTests){
 	ASSERT_EQ("kw", AttendeeDataLanguage);
 	ASSERT_EQ("EXAMPLE=DATA", AttendeeDataTokens);
 	
+	// Get the first CATEGORIES.
+	
+	std::string CategoryData;
+	
+	if (TestTask.CategoriesList.begin() != TestTask.CategoriesList.end()){
+		
+		CategoryData = TestTask.CategoriesList[0];
+		
+	}
+	
+	ASSERT_EQ("CATEGORY ONE, CATEGORY TWO", CategoryData);
+	
+	CategoryData.clear();
+
+	std::string CategoryLanguage;
+	
+	// Get the second CATEGORIES.
+	
+	if (TestTask.CategoriesList.size() > 1){
+		
+		CategoryData = TestTask.CategoriesList[1];
+		
+	}
+	
+	if (TestTask.CategoriesList.size() > 1){
+		
+		CategoryLanguage = TestTask.CategoriesListLanguage[1];
+		
+	}
+		
+	ASSERT_EQ("CATEGORY THREE, CATEGORY FOUR", CategoryData);
+	ASSERT_EQ("en", CategoryLanguage);
+	
+	CategoryData.clear();
+	CategoryLanguage.clear();
+	
+	// Get the third CATEGORIES.
+	
+	std::string CategoryTokens;
+	
+	if (TestTask.CategoriesList.size() > 2){
+		
+		CategoryData = TestTask.CategoriesList[2];
+		
+	}
+	
+	if (TestTask.CategoriesList.size() > 2){
+		
+		CategoryLanguage = TestTask.CategoriesListLanguage[2];
+		
+	}
+	
+	if (TestTask.CategoriesList.size() > 2){
+		
+		CategoryTokens = TestTask.CategoriesListTokens[2];
+		
+	}
+		
+	ASSERT_EQ("CATEGORY FIVE, CATEGORY SIX, CATEGORY SEVEN", CategoryData);
+	ASSERT_EQ("en-GB", CategoryLanguage);
+	ASSERT_EQ("SAMPLE=TOKEN", CategoryTokens);
+	
+	// Get the first COMMENT.
+	
+	std::string CommentData;
+	
+	if (TestTask.CommentList.begin() != TestTask.CommentList.end()){
+		
+		CommentData = TestTask.CommentList[0];
+		
+	}
+	
+	ASSERT_EQ("This is the first comment.", CommentData);
+	
+	// Get the second COMMENT.
+	
+	CommentData.clear();
+	
+	std::string CommentDataAltRep;
+	std::string CommentDataLanguage;
+	
+	if (TestTask.CommentList.size() > 1){
+		
+		CommentData = TestTask.CommentList[1];
+		
+	}
+	
+	if (TestTask.CommentList.size() > 1){
+		
+		CommentDataAltRep = TestTask.CommentListAltRep[1];
+		
+	}
+	
+	if (TestTask.CommentList.size() > 1){
+		
+		CommentDataLanguage = TestTask.CommentListLanguage[1];
+		
+	}
+	
+	ASSERT_EQ("This is the second comment.", CommentData);
+	ASSERT_EQ("null:nodata", CommentDataAltRep);
+	ASSERT_EQ("en", CommentDataLanguage);
+	
+	// Get the third COMMENT.
+	
+	CommentData.clear();
+	
+	std::string CommentDataTokens;
+	
+	if (TestTask.CommentList.size() > 2){
+		
+		CommentData = TestTask.CommentList[2];
+		
+	}
+	
+	if (TestTask.CommentList.size() > 2){
+		
+		CommentDataTokens = TestTask.CommentListTokens[2];
+		
+	}
+	
+	ASSERT_EQ("This is the third comment.", CommentData);
+	ASSERT_EQ("ZEBRAS=YES", CommentDataTokens);
+	
 }
\ No newline at end of file
-- 
2.39.5