Home | History | Annotate | Download | only in omnibox 
      1 // Copyright (c) 2012 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 "base/values.h"
      6 #include "chrome/browser/extensions/api/omnibox/omnibox_api.h"
      7 #include "chrome/common/extensions/api/omnibox.h"
      8 #include "extensions/common/value_builder.h"
      9 #include "testing/gtest/include/gtest/gtest.h"
     10 #include "testing/platform_test.h"
     11 
     12 namespace extensions {
     13 
     14 namespace omnibox = api::omnibox;
     15 namespace SendSuggestions = omnibox::SendSuggestions;
     16 namespace SetDefaultSuggestion = omnibox::SetDefaultSuggestion;
     17 
     18 namespace {
     19 
     20 const int kNone = ACMatchClassification::NONE;
     21 const int kUrl = ACMatchClassification::URL;
     22 const int kMatch = ACMatchClassification::MATCH;
     23 const int kDim = ACMatchClassification::DIM;
     24 
     25 void CompareClassification(const ACMatchClassifications& expected,
     26                            const ACMatchClassifications& actual) {
     27   EXPECT_EQ(expected.size(), actual.size());
     28   for (size_t i = 0; i < expected.size() && i < actual.size(); ++i) {
     29     EXPECT_EQ(expected[i].offset, actual[i].offset) << "Index:" << i;
     30     EXPECT_EQ(expected[i].style, actual[i].style) << "Index:" << i;
     31   }
     32 }
     33 
     34 }  // namespace
     35 
     36 // Test output key: n = character with no styling, d = dim, m = match, u = url
     37 // u = 1, m = 2, d = 4. u+d = 5, etc.
     38 
     39 //   0123456789
     40 //    mmmm
     41 // +       ddd
     42 // = nmmmmndddn
     43 TEST(ExtensionOmniboxTest, DescriptionStylesSimple) {
     44   scoped_ptr<base::ListValue> list = ListBuilder()
     45       .Append(42)
     46       .Append(ListBuilder()
     47         .Append(DictionaryBuilder()
     48           .Set("content", "content")
     49           .Set("description", "description")
     50           .Set("descriptionStyles", ListBuilder()
     51             .Append(DictionaryBuilder()
     52               .Set("type", "match")
     53               .Set("offset", 1)
     54               .Set("length", 4))
     55             .Append(DictionaryBuilder()
     56               .Set("type", "dim")
     57               .Set("offset", 6)
     58               .Set("length", 3))))).Build();
     59 
     60   ACMatchClassifications styles_expected;
     61   styles_expected.push_back(ACMatchClassification(0, kNone));
     62   styles_expected.push_back(ACMatchClassification(1, kMatch));
     63   styles_expected.push_back(ACMatchClassification(5, kNone));
     64   styles_expected.push_back(ACMatchClassification(6, kDim));
     65   styles_expected.push_back(ACMatchClassification(9, kNone));
     66 
     67   scoped_ptr<SendSuggestions::Params> params(
     68       SendSuggestions::Params::Create(*list));
     69   EXPECT_TRUE(params);
     70   EXPECT_TRUE(params->suggest_results[0].get());
     71   CompareClassification(styles_expected, StyleTypesToACMatchClassifications(
     72       *params->suggest_results[0]));
     73 
     74   // Same input, but swap the order. Ensure it still works.
     75   scoped_ptr<base::ListValue> swap_list = ListBuilder()
     76       .Append(42)
     77       .Append(ListBuilder()
     78         .Append(DictionaryBuilder()
     79           .Set("content", "content")
     80           .Set("description", "description")
     81           .Set("descriptionStyles", ListBuilder()
     82             .Append(DictionaryBuilder()
     83               .Set("type", "dim")
     84               .Set("offset", 6)
     85               .Set("length", 3))
     86             .Append(DictionaryBuilder()
     87               .Set("type", "match")
     88               .Set("offset", 1)
     89               .Set("length", 4))))).Build();
     90 
     91   scoped_ptr<SendSuggestions::Params> swapped_params(
     92       SendSuggestions::Params::Create(*swap_list));
     93   EXPECT_TRUE(swapped_params);
     94   EXPECT_TRUE(swapped_params->suggest_results[0].get());
     95   CompareClassification(styles_expected, StyleTypesToACMatchClassifications(
     96       *swapped_params->suggest_results[0]));
     97 }
     98 
     99 //   0123456789
    100 //   uuuuu
    101 // +          dd
    102 // +          mm
    103 // + mmmm
    104 // +  dd
    105 // = 3773unnnn66
    106 TEST(ExtensionOmniboxTest, DescriptionStylesCombine) {
    107   scoped_ptr<base::ListValue> list = ListBuilder()
    108       .Append(42)
    109       .Append(ListBuilder()
    110         .Append(DictionaryBuilder()
    111           .Set("content", "content")
    112           .Set("description", "description")
    113           .Set("descriptionStyles", ListBuilder()
    114             .Append(DictionaryBuilder()
    115               .Set("type", "url")
    116               .Set("offset", 0)
    117               .Set("length", 5))
    118             .Append(DictionaryBuilder()
    119               .Set("type", "dim")
    120               .Set("offset", 9)
    121               .Set("length", 2))
    122             .Append(DictionaryBuilder()
    123               .Set("type", "match")
    124               .Set("offset", 9)
    125               .Set("length", 2))
    126             .Append(DictionaryBuilder()
    127               .Set("type", "match")
    128               .Set("offset", 0)
    129               .Set("length", 4))
    130             .Append(DictionaryBuilder()
    131               .Set("type", "dim")
    132               .Set("offset", 1)
    133               .Set("length", 2))))).Build();
    134 
    135   ACMatchClassifications styles_expected;
    136   styles_expected.push_back(ACMatchClassification(0, kUrl | kMatch));
    137   styles_expected.push_back(ACMatchClassification(1, kUrl | kMatch | kDim));
    138   styles_expected.push_back(ACMatchClassification(3, kUrl | kMatch));
    139   styles_expected.push_back(ACMatchClassification(4, kUrl));
    140   styles_expected.push_back(ACMatchClassification(5, kNone));
    141   styles_expected.push_back(ACMatchClassification(9, kMatch | kDim));
    142 
    143   scoped_ptr<SendSuggestions::Params> params(
    144       SendSuggestions::Params::Create(*list));
    145   EXPECT_TRUE(params);
    146   EXPECT_TRUE(params->suggest_results[0].get());
    147   CompareClassification(styles_expected, StyleTypesToACMatchClassifications(
    148       *params->suggest_results[0]));
    149 
    150   // Try moving the "dim/match" style pair at offset 9. Output should be the
    151   // same.
    152   scoped_ptr<base::ListValue> moved_list = ListBuilder()
    153       .Append(42)
    154       .Append(ListBuilder()
    155         .Append(DictionaryBuilder()
    156           .Set("content", "content")
    157           .Set("description", "description")
    158           .Set("descriptionStyles", ListBuilder()
    159             .Append(DictionaryBuilder()
    160               .Set("type", "url")
    161               .Set("offset", 0)
    162               .Set("length", 5))
    163             .Append(DictionaryBuilder()
    164               .Set("type", "match")
    165               .Set("offset", 0)
    166               .Set("length", 4))
    167             .Append(DictionaryBuilder()
    168               .Set("type", "dim")
    169               .Set("offset", 9)
    170               .Set("length", 2))
    171             .Append(DictionaryBuilder()
    172               .Set("type", "match")
    173               .Set("offset", 9)
    174               .Set("length", 2))
    175             .Append(DictionaryBuilder()
    176               .Set("type", "dim")
    177               .Set("offset", 1)
    178               .Set("length", 2))))).Build();
    179 
    180   scoped_ptr<SendSuggestions::Params> moved_params(
    181       SendSuggestions::Params::Create(*moved_list));
    182   EXPECT_TRUE(moved_params);
    183   EXPECT_TRUE(moved_params->suggest_results[0].get());
    184   CompareClassification(styles_expected, StyleTypesToACMatchClassifications(
    185       *moved_params->suggest_results[0]));
    186 }
    187 
    188 //   0123456789
    189 //   uuuuu
    190 // + mmmmm
    191 // + mmm
    192 // +   ddd
    193 // + ddd
    194 // = 77777nnnnn
    195 TEST(ExtensionOmniboxTest, DescriptionStylesCombine2) {
    196   scoped_ptr<base::ListValue> list = ListBuilder()
    197       .Append(42)
    198       .Append(ListBuilder()
    199         .Append(DictionaryBuilder()
    200           .Set("content", "content")
    201           .Set("description", "description")
    202           .Set("descriptionStyles", ListBuilder()
    203             .Append(DictionaryBuilder()
    204               .Set("type", "url")
    205               .Set("offset", 0)
    206               .Set("length", 5))
    207             .Append(DictionaryBuilder()
    208               .Set("type", "match")
    209               .Set("offset", 0)
    210               .Set("length", 5))
    211             .Append(DictionaryBuilder()
    212               .Set("type", "match")
    213               .Set("offset", 0)
    214               .Set("length", 3))
    215             .Append(DictionaryBuilder()
    216               .Set("type", "dim")
    217               .Set("offset", 2)
    218               .Set("length", 3))
    219             .Append(DictionaryBuilder()
    220               .Set("type", "dim")
    221               .Set("offset", 0)
    222               .Set("length", 3))))).Build();
    223 
    224   ACMatchClassifications styles_expected;
    225   styles_expected.push_back(ACMatchClassification(0, kUrl | kMatch | kDim));
    226   styles_expected.push_back(ACMatchClassification(5, kNone));
    227 
    228   scoped_ptr<SendSuggestions::Params> params(
    229       SendSuggestions::Params::Create(*list));
    230   EXPECT_TRUE(params);
    231   EXPECT_TRUE(params->suggest_results[0].get());
    232   CompareClassification(styles_expected, StyleTypesToACMatchClassifications(
    233       *params->suggest_results[0]));
    234 }
    235 
    236 //   0123456789
    237 //   uuuuu
    238 // + mmmmm
    239 // + mmm
    240 // +   ddd
    241 // + ddd
    242 // = 77777nnnnn
    243 TEST(ExtensionOmniboxTest, DefaultSuggestResult) {
    244   // Default suggestions should not have a content parameter.
    245   scoped_ptr<base::ListValue> list = ListBuilder()
    246       .Append(DictionaryBuilder()
    247         .Set("description", "description")
    248         .Set("descriptionStyles", ListBuilder()
    249           .Append(DictionaryBuilder()
    250             .Set("type", "url")
    251             .Set("offset", 0)
    252             .Set("length", 5))
    253           .Append(DictionaryBuilder()
    254             .Set("type", "match")
    255             .Set("offset", 0)
    256             .Set("length", 5))
    257           .Append(DictionaryBuilder()
    258             .Set("type", "match")
    259             .Set("offset", 0)
    260             .Set("length", 3))
    261           .Append(DictionaryBuilder()
    262             .Set("type", "dim")
    263             .Set("offset", 2)
    264             .Set("length", 3))
    265           .Append(DictionaryBuilder()
    266             .Set("type", "dim")
    267             .Set("offset", 0)
    268             .Set("length", 3)))).Build();
    269 
    270   scoped_ptr<SetDefaultSuggestion::Params> params(
    271       SetDefaultSuggestion::Params::Create(*list));
    272   EXPECT_TRUE(params);
    273 }
    274 
    275 }  // namespace extensions
    276