SpellCheck refactoring: Moved user custom dictionary for spell check to SpellCheckProfile.

chrome/browser/spellchecker/spellcheck_profile_unittest.cc

 // Copyright (c) 2011 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 "base/scoped_temp_dir.h"
 #include "chrome/browser/spellchecker/spellcheck_host.h"
 #include "chrome/browser/spellchecker/spellcheck_profile.h"
 #include "content/browser/browser_thread.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 
 class MockSpellCheckHost : public SpellCheckHost {
  public:
@@ skipping 32 matching lines @@
   bool IsCreatedHostReady() {
     return GetHost() == returning_from_create_.get();
   }
 
   void SetHostToBeCreated(MockSpellCheckHost* host) {
     EXPECT_CALL(*host, UnsetProfile()).Times(1);
     EXPECT_CALL(*host, IsReady()).WillRepeatedly(testing::Return(true));
     returning_from_create_ = host;
   }
 
+  FilePath GetCustomDictionaryDir() {
+    if (!custom_dictionary_dir_.IsValid())
+      EXPECT_TRUE(custom_dictionary_dir_.CreateUniqueTempDir());
+
+    return custom_dictionary_dir_.path();
+  }
+
   size_t create_host_calls_;
   scoped_refptr<SpellCheckHost> returning_from_create_;
+  ScopedTempDir custom_dictionary_dir_;
 };
 
 typedef SpellCheckProfile::ReinitializeResult ResultType;
 }  // namespace
 
 class SpellCheckProfileTest : public testing::Test {
  protected:
   SpellCheckProfileTest()
-      : file_thread_(BrowserThread::FILE) {
+      : ui_thread_(BrowserThread::UI, &message_loop_),
+        file_thread_(BrowserThread::FILE, &message_loop_) {
   }
 
   // SpellCheckHost will be deleted on FILE thread.
+  MessageLoop message_loop_;
+  BrowserThread ui_thread_;
   BrowserThread file_thread_;
 };
 
 TEST_F(SpellCheckProfileTest, ReinitializeEnabled) {
   scoped_refptr<MockSpellCheckHost> host(new MockSpellCheckHost());
-  TestingSpellCheckProfile target;
-  target.SetHostToBeCreated(host.get());
+  scoped_refptr<TestingSpellCheckProfile> target(
+      new TestingSpellCheckProfile());
+  target->SetHostToBeCreated(host.get());
 
   // The first call should create host.
-  ResultType result1 = target.ReinitializeHost(false, true, "", NULL);
-  EXPECT_EQ(target.create_host_calls_, 1U);
+  ResultType result1 = target->ReinitializeHost(false, true, "", NULL);
+  EXPECT_EQ(target->create_host_calls_, 1U);
   EXPECT_EQ(result1, SpellCheckProfile::REINITIALIZE_CREATED_HOST);
 
   // The second call should be ignored.
-  ResultType result2 = target.ReinitializeHost(false, true, "", NULL);
+  ResultType result2 = target->ReinitializeHost(false, true, "", NULL);
   EXPECT_EQ(result2, SpellCheckProfile::REINITIALIZE_DID_NOTHING);
-  EXPECT_EQ(target.create_host_calls_, 1U);
+  EXPECT_EQ(target->create_host_calls_, 1U);
 
   // Host should become ready after the notification.
-  EXPECT_FALSE(target.IsCreatedHostReady());
-  target.SpellCheckHostInitialized(0);
-  EXPECT_TRUE(target.IsCreatedHostReady());
+  EXPECT_FALSE(target->IsCreatedHostReady());
+  target->SpellCheckHostInitialized();
+  EXPECT_TRUE(target->IsCreatedHostReady());
 }
 
 TEST_F(SpellCheckProfileTest, ReinitializeDisabled) {
   scoped_refptr<MockSpellCheckHost> host(new MockSpellCheckHost());
-  TestingSpellCheckProfile target;
-  target.returning_from_create_ = host.get();
+  scoped_refptr<TestingSpellCheckProfile> target(
+      new TestingSpellCheckProfile());
+  target->returning_from_create_ = host.get();
 
   // If enabled is false, nothing should happen
-  ResultType result1 = target.ReinitializeHost(false, false, "", NULL);
-  EXPECT_EQ(target.create_host_calls_, 0U);
+  ResultType result1 = target->ReinitializeHost(false, false, "", NULL);
+  EXPECT_EQ(target->create_host_calls_, 0U);
   EXPECT_EQ(result1, SpellCheckProfile::REINITIALIZE_DID_NOTHING);
 
   // Nothing should happen even if forced.
-  ResultType result2 = target.ReinitializeHost(true, false, "", NULL);
-  EXPECT_EQ(target.create_host_calls_, 0U);
+  ResultType result2 = target->ReinitializeHost(true, false, "", NULL);
+  EXPECT_EQ(target->create_host_calls_, 0U);
   EXPECT_EQ(result2, SpellCheckProfile::REINITIALIZE_DID_NOTHING);
 }
 
 TEST_F(SpellCheckProfileTest, ReinitializeRemove) {
   scoped_refptr<MockSpellCheckHost> host(new MockSpellCheckHost());
-  TestingSpellCheckProfile target;
-  target.SetHostToBeCreated(host.get());
-
+  scoped_refptr<TestingSpellCheckProfile> target(
+      new TestingSpellCheckProfile());
+  target->SetHostToBeCreated(host.get());
 
   // At first, create the host.
-  ResultType result1 = target.ReinitializeHost(false, true, "", NULL);
-  target.SpellCheckHostInitialized(0);
+  ResultType result1 = target->ReinitializeHost(false, true, "", NULL);
+  target->SpellCheckHostInitialized();
   EXPECT_EQ(result1, SpellCheckProfile::REINITIALIZE_CREATED_HOST);
-  EXPECT_TRUE(target.IsCreatedHostReady());
+  EXPECT_TRUE(target->IsCreatedHostReady());
 
   // Then the host should be deleted if it's forced to be disabled.
-  ResultType result2 = target.ReinitializeHost(true, false, "", NULL);
-  target.SpellCheckHostInitialized(0);
+  ResultType result2 = target->ReinitializeHost(true, false, "", NULL);
+  target->SpellCheckHostInitialized();
   EXPECT_EQ(result2, SpellCheckProfile::REINITIALIZE_REMOVED_HOST);
-  EXPECT_FALSE(target.IsCreatedHostReady());
+  EXPECT_FALSE(target->IsCreatedHostReady());
 }
 
 TEST_F(SpellCheckProfileTest, ReinitializeRecreate) {
   scoped_refptr<MockSpellCheckHost> host1(new MockSpellCheckHost());
-  TestingSpellCheckProfile target;
-  target.SetHostToBeCreated(host1.get());
+  scoped_refptr<TestingSpellCheckProfile> target(
+      new TestingSpellCheckProfile());
+  target->SetHostToBeCreated(host1.get());
 
   // At first, create the host.
-  ResultType result1 = target.ReinitializeHost(false, true, "", NULL);
-  target.SpellCheckHostInitialized(0);
-  EXPECT_EQ(target.create_host_calls_, 1U);
+  ResultType result1 = target->ReinitializeHost(false, true, "", NULL);
+  target->SpellCheckHostInitialized();
+  EXPECT_EQ(target->create_host_calls_, 1U);
   EXPECT_EQ(result1, SpellCheckProfile::REINITIALIZE_CREATED_HOST);
-  EXPECT_TRUE(target.IsCreatedHostReady());
+  EXPECT_TRUE(target->IsCreatedHostReady());
 
   // Then the host should be re-created if it's forced to recreate.
   scoped_refptr<MockSpellCheckHost> host2(new MockSpellCheckHost());
-  target.SetHostToBeCreated(host2.get());
+  target->SetHostToBeCreated(host2.get());
 
-  ResultType result2 = target.ReinitializeHost(true, true, "", NULL);
-  target.SpellCheckHostInitialized(0);
-  EXPECT_EQ(target.create_host_calls_, 2U);
+  ResultType result2 = target->ReinitializeHost(true, true, "", NULL);
+  target->SpellCheckHostInitialized();
+  EXPECT_EQ(target->create_host_calls_, 2U);
   EXPECT_EQ(result2, SpellCheckProfile::REINITIALIZE_CREATED_HOST);
-  EXPECT_TRUE(target.IsCreatedHostReady());
+  EXPECT_TRUE(target->IsCreatedHostReady());
 }
 
-TEST_F(SpellCheckProfileTest, SpellCheckHostInitializedWithCustomWords) {
+TEST_F(SpellCheckProfileTest, CustomDictionaryLoaded) {
   scoped_refptr<MockSpellCheckHost> host(new MockSpellCheckHost());
-  TestingSpellCheckProfile target;
-  target.SetHostToBeCreated(host.get());
-  target.ReinitializeHost(false, true, "", NULL);
+  scoped_refptr<TestingSpellCheckProfile> target(
+      new TestingSpellCheckProfile());
+  target->SetHostToBeCreated(host.get());
+  target->ReinitializeHost(false, true, "", NULL);
+
+  scoped_ptr<SpellCheckProfile::CustomWordList> loaded_custom_words
+      (new SpellCheckProfile::CustomWordList());
+  loaded_custom_words->push_back("foo");
+  loaded_custom_words->push_back("bar");
+
+  SpellCheckProfile::CustomWordList expected(*loaded_custom_words);
+  target->CustomDictionaryLoaded(loaded_custom_words.release());
+  EXPECT_EQ(target->GetCustomWords(), expected);
+}
+
+TEST_F(SpellCheckProfileTest, SaveAndLoad) {
+  scoped_refptr<MockSpellCheckHost> host(new MockSpellCheckHost());
+  scoped_refptr<TestingSpellCheckProfile> target(
+      new TestingSpellCheckProfile());
+  target->SetHostToBeCreated(host.get());
+  target->ReinitializeHost(false, true, "", NULL);
 
   scoped_ptr<SpellCheckProfile::CustomWordList> loaded_custom_words
     (new SpellCheckProfile::CustomWordList());
-  loaded_custom_words->push_back("foo");
-  loaded_custom_words->push_back("bar");
-  SpellCheckProfile::CustomWordList expected(*loaded_custom_words);
-  target.SpellCheckHostInitialized(loaded_custom_words.release());
-  EXPECT_EQ(target.GetCustomWords(), expected);
+  target->SpellCheckHostInitialized();
+  SpellCheckProfile::CustomWordList expected;
+  EXPECT_EQ(target->GetCustomWords(), expected);
+  target->CustomWordAddedLocally("foo");
+  expected.push_back("foo");
+  EXPECT_EQ(target->GetCustomWords(), expected);
+  target->CustomWordAddedLocally("bar");
+  expected.push_back("bar");
+  EXPECT_EQ(target->GetCustomWords(), expected);
+
+  file_thread_.message_loop()->RunAllPending();
+
+  // The same dictionary should be loaded again.
+  target->ClearCustomWords();
+  target->LoadCustomDictionary();
+  file_thread_.message_loop()->RunAllPending();
+  ui_thread_.message_loop()->RunAllPending();
+
+  EXPECT_EQ(target->GetCustomWords(), expected);
 }
-
-TEST_F(SpellCheckProfileTest, CustomWordAddedLocally) {
-  scoped_refptr<MockSpellCheckHost> host(new MockSpellCheckHost());
-  TestingSpellCheckProfile target;
-  target.SetHostToBeCreated(host.get());
-  target.ReinitializeHost(false, true, "", NULL);
-
-  scoped_ptr<SpellCheckProfile::CustomWordList> loaded_custom_words
-    (new SpellCheckProfile::CustomWordList());
-  target.SpellCheckHostInitialized(NULL);
-  SpellCheckProfile::CustomWordList expected;
-  EXPECT_EQ(target.GetCustomWords(), expected);
-  target.CustomWordAddedLocally("foo");
-  expected.push_back("foo");
-  EXPECT_EQ(target.GetCustomWords(), expected);
-  target.CustomWordAddedLocally("bar");
-  expected.push_back("bar");
-  EXPECT_EQ(target.GetCustomWords(), expected);
-}