Unit-tests for removing words from the dictionary

chrome/browser/spellchecker/spellcheck_custom_dictionary_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include <vector>
 
 #include "chrome/browser/spellchecker/spellcheck_custom_dictionary.h"
 #include "chrome/browser/spellchecker/spellcheck_factory.h"
 #include "chrome/browser/spellchecker/spellcheck_service.h"
 #include "chrome/common/spellcheck_common.h"
@@ skipping 27 matching lines @@
     MessageLoop::current()->RunUntilIdle();
   }
 
   MessageLoop message_loop_;
   content::TestBrowserThread ui_thread_;
   content::TestBrowserThread file_thread_;
 
   scoped_ptr<TestingProfile> profile_;
 };
 
-// TODO(rlp/rouslan): Shift some of these to a cutsom dictionary test suite.
 TEST_F(SpellcheckCustomDictionaryTest, SpellcheckSetCustomWordList) {
   SpellcheckService* spellcheck_service =
       SpellcheckServiceFactory::GetForProfile(profile_.get());
 
   WordList loaded_custom_words;
   loaded_custom_words.push_back("foo");
   loaded_custom_words.push_back("bar");
   WordList expected(loaded_custom_words);
   SpellcheckCustomDictionary* custom_dictionary =
       spellcheck_service->GetCustomDictionary();
   custom_dictionary->SetCustomWordList(&loaded_custom_words);
   EXPECT_EQ(custom_dictionary->GetWords(), expected);
 }
 
-TEST_F(SpellcheckCustomDictionaryTest, CustomWordAddedLocally) {
+TEST_F(SpellcheckCustomDictionaryTest, CustomWordAddedAndRemovedLocally) {
   SpellcheckService* spellcheck_service =
       SpellcheckServiceFactory::GetForProfile(profile_.get());
 
   WordList loaded_custom_words;
   SpellcheckCustomDictionary* custom_dictionary =
       spellcheck_service->GetCustomDictionary();
-  custom_dictionary->Load();
   WordList expected;
   EXPECT_EQ(custom_dictionary->GetWords(), expected);
   custom_dictionary->CustomWordAddedLocally("foo");
   expected.push_back("foo");
   EXPECT_EQ(custom_dictionary->GetWords(), expected);
   custom_dictionary->CustomWordAddedLocally("bar");
   expected.push_back("bar");
   EXPECT_EQ(custom_dictionary->GetWords(), expected);
+
+  custom_dictionary->CustomWordRemovedLocally("foo");
+  custom_dictionary->CustomWordRemovedLocally("bar");
+  expected.clear();
+  EXPECT_EQ(custom_dictionary->GetWords(), expected);
 }
 
 TEST_F(SpellcheckCustomDictionaryTest, SaveAndLoad) {
   SpellcheckService* spellcheck_service =
       SpellcheckServiceFactory::GetForProfile(profile_.get());
   SpellcheckCustomDictionary* custom_dictionary =
       spellcheck_service->GetCustomDictionary();
 
   WordList loaded_custom_words;
   custom_dictionary->LoadDictionaryIntoCustomWordList(&loaded_custom_words);
@@ skipping 12 matching lines @@
   custom_dictionary->LoadDictionaryIntoCustomWordList(&loaded_custom_words);
   EXPECT_EQ(loaded_custom_words, expected);
 
   // Load in another instance of SpellCheckService.
   // The result should be the same.
   SpellcheckService spellcheck_service2(profile_.get());
   WordList loaded_custom_words2;
   spellcheck_service2.GetCustomDictionary()->
       LoadDictionaryIntoCustomWordList(&loaded_custom_words2);
   EXPECT_EQ(loaded_custom_words2, expected);
+
+  custom_dictionary->EraseWordFromCustomDictionary("foo");
+  custom_dictionary->EraseWordFromCustomDictionary("bar");
+  custom_dictionary->LoadDictionaryIntoCustomWordList(&loaded_custom_words);
+  expected.clear();
+  EXPECT_EQ(loaded_custom_words, expected);
+
   // Flush the loop now to prevent service init tasks from being run during
   // TearDown();
   MessageLoop::current()->RunUntilIdle();
 }
 
 TEST_F(SpellcheckCustomDictionaryTest, MultiProfile) {
   SpellcheckService* spellcheck_service =
       SpellcheckServiceFactory::GetForProfile(profile_.get());
   SpellcheckCustomDictionary* custom_dictionary =
       spellcheck_service->GetCustomDictionary();
@@ skipping 23 matching lines @@
   EXPECT_EQ(actual1, expected1);
 
   WordList actual2;
   custom_dictionary2->LoadDictionaryIntoCustomWordList(&actual2);
   EXPECT_EQ(actual2, expected2);
 
   // Flush the loop now to prevent service init tasks from being run during
   // TearDown();
   MessageLoop::current()->RunUntilIdle();
 }