1// Copyright (c) 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#ifndef CHROME_BROWSER_EXTENSIONS_API_OMNIBOX_OMNIBOX_API_TESTBASE_H_
6#define CHROME_BROWSER_EXTENSIONS_API_OMNIBOX_OMNIBOX_API_TESTBASE_H_
7
8#include "base/format_macros.h"
9#include "base/strings/stringprintf.h"
10#include "base/strings/utf_string_conversions.h"
11#include "chrome/browser/chrome_notification_types.h"
12#include "chrome/browser/extensions/extension_apitest.h"
13#include "chrome/browser/ui/browser.h"
14#include "chrome/browser/ui/browser_window.h"
15#include "chrome/browser/ui/location_bar/location_bar.h"
16#include "chrome/browser/ui/omnibox/omnibox_popup_model.h"
17#include "chrome/browser/ui/omnibox/omnibox_view.h"
18#include "components/omnibox/autocomplete_match.h"
19#include "components/omnibox/autocomplete_result.h"
20#include "content/public/test/test_utils.h"
21
22
23class AutocompleteController;
24
25class OmniboxApiTest : public ExtensionApiTest {
26 protected:
27  LocationBar* GetLocationBar(Browser* browser) const {
28    return browser->window()->GetLocationBar();
29  }
30
31  AutocompleteController* GetAutocompleteController(Browser* browser) const {
32    return GetLocationBar(browser)->GetOmniboxView()->model()->popup_model()->
33        autocomplete_controller();
34  }
35
36  // TODO(phajdan.jr): Get rid of this wait-in-a-loop pattern.
37  void WaitForAutocompleteDone(AutocompleteController* controller) {
38    while (!controller->done()) {
39      content::WindowedNotificationObserver ready_observer(
40          chrome::NOTIFICATION_AUTOCOMPLETE_CONTROLLER_RESULT_READY,
41          content::Source<AutocompleteController>(controller));
42      ready_observer.Wait();
43    }
44  }
45
46  static base::string16 AutocompleteResultAsString(
47      const AutocompleteResult& result) {
48    std::string output(base::StringPrintf("{%" PRIuS "} ", result.size()));
49    for (size_t i = 0; i < result.size(); ++i) {
50      AutocompleteMatch match = result.match_at(i);
51      std::string provider_name = match.provider->GetName();
52      output.append(
53          base::StringPrintf("[\"%s\" by \"%s\"] ",
54                             base::UTF16ToUTF8(match.contents).c_str(),
55                             provider_name.c_str()));
56    }
57    return base::UTF8ToUTF16(output);
58  }
59};
60
61#endif  // CHROME_BROWSER_EXTENSIONS_API_OMNIBOX_OMNIBOX_API_TESTBASE_H_
62