src/ui/app_list/search/tokenized_string_char_iterator_unittest.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "ui/app_list/search/tokenized_string_char_iterator.h"
   6
   7 #include <string>
   8 #include <vector>
   9
  10 #include "base/strings/string_util.h"
  11 #include "base/strings/stringprintf.h"
  12 #include "base/strings/utf_string_conversions.h"
  13 #include "testing/gtest/include/gtest/gtest.h"
  14
  15 namespace app_list {
  16 namespace test {
  17
  18 namespace {
  19
  20 // Returns a string represents the current state of |iter|. The state string
  21 // has three fields. The first is the current char. The second is the offset of
  22 // the current char in terms of the original text of the TokenizedString. The
  23 // last one is optional and only shows up when IsFirstCharOfToken returns true.
  24 std::string GetIterateState(const TokenizedStringCharIterator& iter) {
  25   return base::StringPrintf("%s%d%s",
  26                             base::UTF16ToUTF8(
  27                                 base::string16(1, iter.Get())).c_str(),
  28                             iter.GetArrayPos(),
  29                             iter.IsFirstCharOfToken() ? "!" : "");
  30 }
  31
  32 void TestBeyondTheEnd(TokenizedStringCharIterator* iter) {
  33   ASSERT_TRUE(iter->end());
  34   ASSERT_FALSE(iter->NextChar());
  35   ASSERT_FALSE(iter->NextToken());
  36
  37   // Don't care what it returns, but this shouldn't crash.
  38   iter->Get();
  39 }
  40
  41 void TestEveryChar(const std::string& text, const std::string& expects) {
  42   TokenizedString tokens(base::UTF8ToUTF16(text));
  43   TokenizedStringCharIterator iter(tokens);
  44
  45   std::vector<std::string> results;
  46   while (!iter.end()) {
  47     results.push_back(GetIterateState(iter));
  48     iter.NextChar();
  49   }
  50
  51   EXPECT_EQ(expects, JoinString(results, ' '));
  52   TestBeyondTheEnd(&iter);
  53 }
  54
  55 void TestNextToken(const std::string& text, const std::string& expects) {
  56   TokenizedString tokens(base::UTF8ToUTF16(text));
  57   TokenizedStringCharIterator iter(tokens);
  58
  59   std::vector<std::string> results;
  60   while (!iter.end()) {
  61     results.push_back(GetIterateState(iter));
  62     iter.NextToken();
  63   }
  64
  65   EXPECT_EQ(expects, JoinString(results, ' '));
  66   TestBeyondTheEnd(&iter);
  67 }
  68
  69 void TestFirstTwoCharInEveryToken(const std::string& text,
  70                                   const std::string& expects) {
  71   TokenizedString tokens(base::UTF8ToUTF16(text));
  72   TokenizedStringCharIterator iter(tokens);
  73
  74   std::vector<std::string> results;
  75   while (!iter.end()) {
  76     results.push_back(GetIterateState(iter));
  77     if (iter.NextChar())
  78       results.push_back(GetIterateState(iter));
  79
  80     iter.NextToken();
  81   }
  82
  83   EXPECT_EQ(expects, JoinString(results, ' '));
  84   TestBeyondTheEnd(&iter);
  85 }
  86
  87 }  // namespace
  88
  89 TEST(TokenizedStringCharIteratorTest, NoTerms) {
  90   const char* text;
  91
  92   text = "";
  93   TestEveryChar(text, "");
  94   TestNextToken(text, "");
  95   TestFirstTwoCharInEveryToken(text, "");
  96
  97   text = "!@#$%^&*()<<<**>>>";
  98   TestEveryChar(text, "");
  99   TestNextToken(text, "");
 100   TestFirstTwoCharInEveryToken(text, "");
 101 }
 102
 103 TEST(TokenizedStringCharIteratorTest, Basic) {
 104   const char* text;
 105
 106   text = "c";
 107   TestEveryChar(text, "c0!");
 108   TestNextToken(text, "c0!");
 109   TestFirstTwoCharInEveryToken(text, "c0!");
 110
 111   text = "Simple";
 112   TestEveryChar(text, "s0! i1 m2 p3 l4 e5");
 113   TestNextToken(text, "s0!");
 114   TestFirstTwoCharInEveryToken(text, "s0! i1");
 115
 116   text = "ScratchPad";
 117   TestEveryChar(text, "s0! c1 r2 a3 t4 c5 h6 p7! a8 d9");
 118   TestNextToken(text, "s0! p7!");
 119   TestFirstTwoCharInEveryToken(text, "s0! c1 p7! a8");
 120
 121   text = "Chess2.0";
 122   TestEveryChar(text, "c0! h1 e2 s3 s4 25! .6 07");
 123   TestNextToken(text, "c0! 25!");
 124   TestFirstTwoCharInEveryToken(text, "c0! h1 25! .6");
 125
 126   text = "Cut the rope";
 127   TestEveryChar(text, "c0! u1 t2 t4! h5 e6 r8! o9 p10 e11");
 128   TestNextToken(text, "c0! t4! r8!");
 129   TestFirstTwoCharInEveryToken(text, "c0! u1 t4! h5 r8! o9");
 130
 131   text = "AutoCAD WS";
 132   TestEveryChar(text, "a0! u1 t2 o3 c4! a5 d6 w8! s9");
 133   TestNextToken(text, "a0! c4! w8!");
 134   TestFirstTwoCharInEveryToken(text, "a0! u1 c4! a5 w8! s9");
 135
 136   text = "Great TweetDeck";
 137   TestEveryChar(text, "g0! r1 e2 a3 t4 t6! w7 e8 e9 t10 d11! e12 c13 k14");
 138   TestNextToken(text, "g0! t6! d11!");
 139   TestFirstTwoCharInEveryToken(text, "g0! r1 t6! w7 d11! e12");
 140
 141   text = "Draw-It!";
 142   TestEveryChar(text, "d0! r1 a2 w3 i5! t6");
 143   TestNextToken(text, "d0! i5!");
 144   TestFirstTwoCharInEveryToken(text, "d0! r1 i5! t6");
 145
 146   text = "Faxing & Signing";
 147   TestEveryChar(text, "f0! a1 x2 i3 n4 g5 s9! i10 g11 n12 i13 n14 g15");
 148   TestNextToken(text, "f0! s9!");
 149   TestFirstTwoCharInEveryToken(text, "f0! a1 s9! i10");
 150 }
 151
 152 }  // namespace test
 153 }  // namespace app_list