From f510df7b6c0130f86767f565a61d86138cccc759 Mon Sep 17 00:00:00 2001
From: Steve Brokenshire <sbrokenshire@xestia.co.uk>
Date: Sun, 31 Jan 2016 15:12:23 +0000
Subject: [PATCH] Added unit tests for CATEGORIES and COMMENT in
 iCalendarJournal/ObjectDataTests.

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

diff --git a/source/tests/xestiacalendar_icaljournalload.h b/source/tests/xestiacalendar_icaljournalload.h
index 368c922..8ccab78 100644
--- a/source/tests/xestiacalendar_icaljournalload.h
+++ b/source/tests/xestiacalendar_icaljournalload.h
@@ -320,4 +320,128 @@ TEST(iCalendarJournal, ObjectDataTests){
 	ASSERT_EQ("kw", AttendeeDataLanguage);
 	ASSERT_EQ("EXAMPLE=DATA", AttendeeDataTokens);
 	
+	// Get the first CATEGORIES.
+	
+	std::string CategoryData;
+	
+	if (TestJournal.CategoriesList.begin() != TestJournal.CategoriesList.end()){
+		
+		CategoryData = TestJournal.CategoriesList[0];
+		
+	}
+	
+	ASSERT_EQ("CATEGORY ONE, CATEGORY TWO", CategoryData);
+	
+	CategoryData.clear();
+
+	std::string CategoryLanguage;
+	
+	// Get the second CATEGORIES.
+	
+	if (TestJournal.CategoriesList.size() > 1){
+		
+		CategoryData = TestJournal.CategoriesList[1];
+		
+	}
+	
+	if (TestJournal.CategoriesList.size() > 1){
+		
+		CategoryLanguage = TestJournal.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 (TestJournal.CategoriesList.size() > 2){
+		
+		CategoryData = TestJournal.CategoriesList[2];
+		
+	}
+	
+	if (TestJournal.CategoriesList.size() > 2){
+		
+		CategoryLanguage = TestJournal.CategoriesListLanguage[2];
+		
+	}
+	
+	if (TestJournal.CategoriesList.size() > 2){
+		
+		CategoryTokens = TestJournal.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 (TestJournal.CommentList.begin() != TestJournal.CommentList.end()){
+		
+		CommentData = TestJournal.CommentList[0];
+		
+	}
+	
+	ASSERT_EQ("This is the first comment.", CommentData);
+	
+	// Get the second COMMENT.
+	
+	CommentData.clear();
+	
+	std::string CommentDataAltRep;
+	std::string CommentDataLanguage;
+	
+	if (TestJournal.CommentList.size() > 1){
+		
+		CommentData = TestJournal.CommentList[1];
+		
+	}
+	
+	if (TestJournal.CommentList.size() > 1){
+		
+		CommentDataAltRep = TestJournal.CommentListAltRep[1];
+		
+	}
+	
+	if (TestJournal.CommentList.size() > 1){
+		
+		CommentDataLanguage = TestJournal.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 (TestJournal.CommentList.size() > 2){
+		
+		CommentData = TestJournal.CommentList[2];
+		
+	}
+	
+	if (TestJournal.CommentList.size() > 2){
+		
+		CommentDataTokens = TestJournal.CommentListTokens[2];
+		
+	}
+	
+	ASSERT_EQ("This is the third comment.", CommentData);
+	ASSERT_EQ("ZEBRAS=YES", CommentDataTokens);
+	
 }
\ No newline at end of file
-- 
2.39.2