Improve reliability of custom spelling dictionary file.

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 "base/file_util.h"
 #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/chrome_constants.h"
 #include "chrome/common/spellcheck_common.h"
 #include "chrome/test/base/testing_profile.h"
 #include "content/public/test/test_browser_thread.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using content::BrowserThread;
 using chrome::spellcheck_common::WordList;
 
 static ProfileKeyedService* BuildSpellcheckService(Profile* profile) {
@@ skipping 137 matching lines @@
 
   WordList actual2;
   custom_dictionary2->LoadDictionaryIntoCustomWordList(&actual2);
   std::sort(expected2.begin(), expected2.end());
   EXPECT_EQ(actual2, expected2);
 
   // Flush the loop now to prevent service init tasks from being run during
   // TearDown();
   MessageLoop::current()->RunUntilIdle();
 }
+
+// Legacy empty dictionary should be converted to new format empty dicitonary.
+TEST_F(SpellcheckCustomDictionaryTest, LegacyEmptyDictionaryShouldBeConverted) {
+  FilePath dictionary_path(
+      profile_->GetPath().Append(chrome::kCustomDictionaryFileName));
+  SpellcheckService* spellcheck_service =
+      SpellcheckServiceFactory::GetForProfile(profile_.get());
+  SpellcheckCustomDictionary* custom_dictionary =
+      spellcheck_service->GetCustomDictionary();
+  WordList loaded_custom_words;
+
+  std::string content;
+  file_util::WriteFile(dictionary_path, content.c_str(), content.length());
+  custom_dictionary->LoadDictionaryIntoCustomWordList(&loaded_custom_words);
+  EXPECT_TRUE(loaded_custom_words.empty());
+
+  // Flush the loop now to prevent service init tasks from being run during
+  // TearDown();
+  MessageLoop::current()->RunUntilIdle();
+}
+
+// Legacy dictionary with two words should be converted to new format dictionary
+// with two words.
+TEST_F(SpellcheckCustomDictionaryTest,
+       LegacyDictionaryWithTwoWordsShouldBeConverted) {
+  FilePath dictionary_path(
+      profile_->GetPath().Append(chrome::kCustomDictionaryFileName));
+  SpellcheckService* spellcheck_service =
+      SpellcheckServiceFactory::GetForProfile(profile_.get());
+  SpellcheckCustomDictionary* custom_dictionary =
+      spellcheck_service->GetCustomDictionary();
+  WordList loaded_custom_words;
+  WordList expected;
+
+  std::string content = "foo\nbar";
+  file_util::WriteFile(dictionary_path, content.c_str(), content.length());
+  custom_dictionary->LoadDictionaryIntoCustomWordList(&loaded_custom_words);
+  expected.push_back("bar");
+  expected.push_back("foo");
+  EXPECT_EQ(expected, loaded_custom_words);
+
+  // Flush the loop now to prevent service init tasks from being run during
+  // TearDown();
+  MessageLoop::current()->RunUntilIdle();
+}
+
+// Words with spaces are illegal and should be removed.
+TEST_F(SpellcheckCustomDictionaryTest,
+       IllegalWordsShouldBeRemovedFromDictionary) {
+  FilePath dictionary_path(
+      profile_->GetPath().Append(chrome::kCustomDictionaryFileName));
+  SpellcheckService* spellcheck_service =
+      SpellcheckServiceFactory::GetForProfile(profile_.get());
+  SpellcheckCustomDictionary* custom_dictionary =
+      spellcheck_service->GetCustomDictionary();
+  WordList loaded_custom_words;
+  WordList expected;
+
+  std::string content = "foo\nfoo bar\nbar\nfoo bar";
+  file_util::WriteFile(dictionary_path, content.c_str(), content.length());
+  custom_dictionary->LoadDictionaryIntoCustomWordList(&loaded_custom_words);
+  expected.push_back("bar");
+  expected.push_back("foo");
+  EXPECT_EQ(expected, loaded_custom_words);
+
+  // Flush the loop now to prevent service init tasks from being run during
+  // TearDown();
+  MessageLoop::current()->RunUntilIdle();
+}
+
+// Write to dicitonary should backup previous version and write the word to the
+// end of the dictionary. If the dictionary file is corrupted on disk, the
+// previous version should be reloaded.
+TEST_F(SpellcheckCustomDictionaryTest, CorruptedWriteShouldBeRecovered) {
+  FilePath dictionary_path(
+      profile_->GetPath().Append(chrome::kCustomDictionaryFileName));
+  SpellcheckService* spellcheck_service =
+      SpellcheckServiceFactory::GetForProfile(profile_.get());
+  SpellcheckCustomDictionary* custom_dictionary =
+      spellcheck_service->GetCustomDictionary();
+  WordList loaded_custom_words;
+  WordList expected;
+
+  std::string content = "foo\nbar";
+  file_util::WriteFile(dictionary_path, content.c_str(), content.length());
+  custom_dictionary->LoadDictionaryIntoCustomWordList(&loaded_custom_words);
+  expected.push_back("bar");
+  expected.push_back("foo");
+  EXPECT_EQ(expected, loaded_custom_words);
+
+  custom_dictionary->WriteWordToCustomDictionary("baz");
+  content.clear();
+  file_util::ReadFileToString(dictionary_path, &content);
+  content.append("corruption");
+  file_util::WriteFile(dictionary_path, content.c_str(), content.length());
+  custom_dictionary->LoadDictionaryIntoCustomWordList(&loaded_custom_words);
+  EXPECT_EQ(expected, loaded_custom_words);
+
+  // Flush the loop now to prevent service init tasks from being run during
+  // TearDown();
+  MessageLoop::current()->RunUntilIdle();
+}