1 files changed, 11 insertions, 3 deletions
diff --git a/chrome/renderer/spellchecker/custom_dictionary_engine_unittest.cc b/chrome/renderer/spellchecker/custom_dictionary_engine_unittest.cc
index 440d110..3a6ea8a 100644
--- a/chrome/renderer/spellchecker/custom_dictionary_engine_unittest.cc
+++ b/chrome/renderer/spellchecker/custom_dictionary_engine_unittest.cc
@@ -1,15 +1,23 @@
 // Copyright (c) 2013 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 "chrome/renderer/spellchecker/custom_dictionary_engine.h"
 
+#include "base/strings/utf_string_conversions.h"
+#include "chrome/renderer/spellchecker/custom_dictionary_engine.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 TEST(CustomDictionaryTest, HandlesEmptyWordWithInvalidSubstring) {
   CustomDictionaryEngine engine;
-  std::vector<std::string> custom_words;
-
+  std::set<std::string> custom_words;
   engine.Init(custom_words);
   EXPECT_FALSE(engine.SpellCheckWord(string16().c_str(), 15, 23));
 }
 
+TEST(CustomDictionaryTest, Basic) {
+  CustomDictionaryEngine engine;
+  EXPECT_FALSE(engine.SpellCheckWord(ASCIIToUTF16("helllo").c_str(), 0, 6));
+  std::set<std::string> custom_words;
+  custom_words.insert("helllo");
+  engine.Init(custom_words);
+  EXPECT_TRUE(engine.SpellCheckWord(ASCIIToUTF16("helllo").c_str(), 0, 6));
+}