jens@0: /*
jens@0:  *  Dictionary_test.cpp
jens@0:  *  Ottoman
jens@0:  *
jens@0:  *  Created by Jens Alfke on 9/4/09.
jens@0:  *  Copyright 2009 Jens Alfke. All rights reserved.
jens@0:  *  BSD-Licensed: See the file "LICENSE.txt" for details.
jens@0:  */
jens@0: 
jens@0: 
jens@0: #include <gtest/gtest.h>
jens@0: #include "TestUtils.h"
jens@0: #include "File.h"
jens@0: #include "Dictionary.h"
jens@0: #include "Hash.h"
jens@0: #include <fcntl.h>
jens@0: #include <stdio.h>
jens@0: 
jens@0: using namespace Mooseyard;
jens@0: 
jens@0: static const HashDictionary& getDict() {
jens@0:     static HashDictionary *sDict;
jens@0:     if (!sDict) {
jens@0:         printf("Building large HashDictionary...\n");
jens@0:         sDict = new HashDictionary();
jens@0:         readWords();
jens@0:         for( int i=0; i<sNWords; i++) {
jens@0:             Blob kv(sWords[i]);
jens@0:             sDict->put(Key(kv),kv);
jens@0:         }
jens@0:     }
jens@0:     return *sDict;
jens@0: }
jens@0: 
jens@0: TEST(Dictionary,GetAll) {
jens@0:     const Dictionary &dict = getDict();
jens@0:     EXPECT_EQ( sNWords ,  dict.count() );
jens@0:     for( int i=0; i<sNWords; i++) {
jens@0:         Key key(sWords[i]);
jens@0:         Blob value = dict.get(key);
jens@0:         EXPECT_TRUE(value);
jens@0:         EXPECT_TRUE( value.equals(key) );
jens@0:     }
jens@0: }
jens@0: 
jens@0: TEST(Dictionary,Iterate) {
jens@0:     const HashDictionary &dict = getDict();
jens@0:     Timer t("Iterating Dictionary", sNWords);
jens@0:     int n=0;
jens@0:     for( HashDictionary::Iterator it(dict); it; ++it) {
jens@0:         n++;
jens@0:         EXPECT_TRUE(it.key().length > 0 && it.key().length < 50);
jens@0:         EXPECT_TRUE(it.key().equals(it.value()));
jens@0:         EXPECT_EQ( 0, ((size_t)it.value().bytes & 0x3) );  // 4-byte aligned
jens@0:     }
jens@0:     EXPECT_EQ(sNWords, n);
jens@0: }
jens@0: 
jens@0: TEST(Dictionary,Overlay) {
jens@0:     const Dictionary &dict = getDict();
jens@0:     OverlayDictionary overlay(&dict);
jens@0:     
jens@0:     EXPECT_EQ( sNWords ,  overlay.count() );
jens@0:     EXPECT_TRUE(overlay.get("animal").equals("animal"));
jens@0:     EXPECT_TRUE(overlay.get("asparagus").equals("asparagus"));
jens@0:     EXPECT_FALSE(overlay.get("growf"));
jens@0:     
jens@0:     overlay.put("animal", "AMINAL");
jens@0:     overlay.put("growf", "growf");
jens@0:     EXPECT_TRUE(overlay.remove("asparagus"));
jens@0:     
jens@0:     EXPECT_EQ( sNWords, overlay.count() );
jens@0:     EXPECT_TRUE(overlay.get("animal").equals("AMINAL"));
jens@0:     EXPECT_TRUE(overlay.get("growf").equals("growf"));
jens@0:     EXPECT_TRUE(overlay.contains("growf"));
jens@0:     EXPECT_FALSE(overlay.get("asparagus"));
jens@0:     EXPECT_FALSE(overlay.contains("asparagus"));
jens@0:     
jens@0:     int n=0;
jens@0:     for( OverlayDictionary::Iterator it(overlay); it; ++it) {
jens@0:         n++;
jens@0:         EXPECT_TRUE(!it.key().equals("asparagus"));
jens@0:     }
jens@0:     EXPECT_EQ(sNWords, n);
jens@0:     
jens@0:     printf("Testing ChangeIterator...\n");
jens@0:     n=0;
jens@0:     int foundAsparagus=0, foundAnimal=0, foundGrowf=0;
jens@0:     for( Dictionary::ChangeIterator it(&overlay); it; ++it) {
jens@0:         n++;
jens@0:         if (it.key().equals("animal")) {
jens@0:             foundAnimal++;
jens@0:             EXPECT_TRUE(it.value().equals("AMINAL"));
jens@0:             EXPECT_TRUE(it.otherValue().equals("animal"));
jens@0:         } else if (it.key().equals("asparagus")) {
jens@0:             foundAsparagus++;
jens@0:             EXPECT_FALSE(it.value());
jens@0:             EXPECT_TRUE(it.otherValue().equals("asparagus"));
jens@0:         } else if (it.key().equals("growf")) {
jens@0:             foundGrowf++;
jens@0:             EXPECT_TRUE(it.value().equals("growf"));
jens@0:             EXPECT_FALSE(it.otherValue());
jens@0:         } else {
jens@0:             EXPECT_TRUE(false);
jens@0:         }
jens@0:     }
jens@0:     EXPECT_EQ(1, foundAnimal);
jens@0:     EXPECT_EQ(1, foundAsparagus);
jens@0:     EXPECT_EQ(1, foundGrowf);
jens@0:     EXPECT_EQ(3, n);
jens@0: }