Added tests for checking the PHOTO and LOGO TYPEs.

[xestiaab/.git] / source / tests / xestiaab_contactload.h

diff --git a/source/tests/xestiaab_contactload.h b/source/tests/xestiaab_contactload.h
index 3c72c72..bc3f62f 100644 (file)
--- a/source/tests/xestiaab_contactload.h
+++ b/source/tests/xestiaab_contactload.h
@@ -17,7 +17,7 @@
 // with Xestia Address Book. If not, see <http://www.gnu.org/licenses/>
 
 #include <gtest/gtest.h>
-#include "../contacteditor/ContactDataObject.h"
+#include "../contacteditor/cdo/ContactDataObject.h"
 
 TEST(ContactLoad, ContactLoadTests){
 
@@ -2578,6 +2578,11 @@ TEST(ContactLoad, PhotoTests){
        ASSERT_NE(TestFile.PicturesListPID.end(), TestFileIter);
        ASSERT_EQ("71", TestFileIter->second);
        
+       // Check the TYPE section.
+       
+       TestFileIter = TestFile.PicturesListType.find(0);
+       ASSERT_EQ(TestFile.PicturesListType.end(), TestFileIter);
+       
        // Check the MEDIATYPE section.
        
        TestFileIter = TestFile.PicturesListMediatype.find(0);
@@ -2627,6 +2632,12 @@ TEST(ContactLoad, PhotoTests){
        ASSERT_NE(TestFile.PicturesListPID.end(), TestFileIter);
        ASSERT_EQ("21", TestFileIter->second);
        
+       // Check the TYPE section.
+       
+       TestFileIter = TestFile.PicturesListType.find(1);
+       ASSERT_NE(TestFile.PicturesListType.end(), TestFileIter);
+       ASSERT_EQ("home", TestFileIter->second);
+       
        // Check the MEDIATYPE section.
        
        TestFileIter = TestFile.PicturesListMediatype.find(1);
@@ -2676,6 +2687,12 @@ TEST(ContactLoad, PhotoTests){
        ASSERT_NE(TestFile.PicturesListPID.end(), TestFileIter);
        ASSERT_EQ("91", TestFileIter->second);
        
+       // Check the TYPE section.
+       
+       TestFileIter = TestFile.PicturesListType.find(2);
+       ASSERT_NE(TestFile.PicturesListType.end(), TestFileIter);
+       ASSERT_EQ("work", TestFileIter->second);
+       
        // Check the MEDIATYPE section.
        
        TestFileIter = TestFile.PicturesListMediatype.find(2);
@@ -2739,6 +2756,8 @@ TEST(ContactLoad, LogoTests){
        ASSERT_NE(TestFile.LogosListPID.end(), TestFileIter);
        ASSERT_EQ("71", TestFileIter->second);
        
+       // Check the TYPE section.
+               
        // Check the MEDIATYPE section.
        
        TestFileIter = TestFile.LogosListMediatype.find(0);
@@ -2787,6 +2806,12 @@ TEST(ContactLoad, LogoTests){
        TestFileIter = TestFile.LogosListPID.find(1);
        ASSERT_NE(TestFile.LogosListPID.end(), TestFileIter);
        ASSERT_EQ("21", TestFileIter->second);
+
+       // Check the TYPE section.
+       
+       TestFileIter = TestFile.LogosListType.find(1);
+       ASSERT_NE(TestFile.LogosListType.end(), TestFileIter);
+       ASSERT_EQ("home", TestFileIter->second);
        
        // Check the MEDIATYPE section.
        
@@ -2837,6 +2862,12 @@ TEST(ContactLoad, LogoTests){
        ASSERT_NE(TestFile.LogosListPID.end(), TestFileIter);
        ASSERT_EQ("91", TestFileIter->second);
        
+       // Check the TYPE section.
+       
+       TestFileIter = TestFile.LogosListType.find(2);
+       ASSERT_NE(TestFile.LogosListType.end(), TestFileIter);
+       ASSERT_EQ("work", TestFileIter->second);
+       
        // Check the MEDIATYPE section.
        
        TestFileIter = TestFile.LogosListMediatype.find(2);