src/chrome/test/chromedriver/chrome/chrome_impl.cc

   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 "chrome/test/chromedriver/chrome/chrome_impl.h"
   6
   7 #include "chrome/test/chromedriver/chrome/devtools_client.h"
   8 #include "chrome/test/chromedriver/chrome/devtools_event_listener.h"
   9 #include "chrome/test/chromedriver/chrome/devtools_http_client.h"
  10 #include "chrome/test/chromedriver/chrome/status.h"
  11 #include "chrome/test/chromedriver/chrome/version.h"
  12 #include "chrome/test/chromedriver/chrome/web_view_impl.h"
  13 #include "chrome/test/chromedriver/net/port_server.h"
  14
  15 ChromeImpl::~ChromeImpl() {
  16   if (!quit_)
  17     port_reservation_->Leak();
  18 }
  19
  20 ChromeDesktopImpl* ChromeImpl::GetAsDesktop() {
  21   return NULL;
  22 }
  23
  24 const BrowserInfo* ChromeImpl::GetBrowserInfo() {
  25   return devtools_http_client_->browser_info();
  26 }
  27
  28 bool ChromeImpl::HasCrashedWebView() {
  29   for (WebViewList::iterator it = web_views_.begin();
  30        it != web_views_.end(); ++it) {
  31     if ((*it)->WasCrashed())
  32       return true;
  33   }
  34   return false;
  35 }
  36
  37 Status ChromeImpl::GetWebViewIds(std::list<std::string>* web_view_ids) {
  38   WebViewsInfo views_info;
  39   Status status = devtools_http_client_->GetWebViewsInfo(&views_info);
  40   if (status.IsError())
  41     return status;
  42
  43   // Check if some web views are closed.
  44   WebViewList::iterator it = web_views_.begin();
  45   while (it != web_views_.end()) {
  46     if (!views_info.GetForId((*it)->GetId())) {
  47       it = web_views_.erase(it);
  48     } else {
  49       ++it;
  50     }
  51   }
  52
  53   // Check for newly-opened web views.
  54   for (size_t i = 0; i < views_info.GetSize(); ++i) {
  55     const WebViewInfo& view = views_info.Get(i);
  56     if (view.type != WebViewInfo::kPage && view.type != WebViewInfo::kApp)
  57       continue;
  58
  59     // Workaround to ignore generated background pages that are
  60     // being returned as active windows for some builds of Chrome.
  61     // TODO(bustamante): Once Chrome builds < 1755 are no longer
  62     //   supported this check can be removed.
  63     int kBuildNumber = devtools_http_client_->browser_info()->build_no;
  64     if (kBuildNumber < 1755 && view.type == WebViewInfo::kApp &&
  65         view.url.find("_generated_background") != std::string::npos)
  66       continue;
  67
  68     bool found = false;
  69     for (WebViewList::const_iterator web_view_iter = web_views_.begin();
  70          web_view_iter != web_views_.end(); ++web_view_iter) {
  71       if ((*web_view_iter)->GetId() == view.id) {
  72         found = true;
  73         break;
  74       }
  75     }
  76     if (!found) {
  77       scoped_ptr<DevToolsClient> client(
  78           devtools_http_client_->CreateClient(view.id));
  79       for (ScopedVector<DevToolsEventListener>::const_iterator listener =
  80                devtools_event_listeners_.begin();
  81            listener != devtools_event_listeners_.end(); ++listener) {
  82         client->AddListener(*listener);
  83         // OnConnected will fire when DevToolsClient connects later.
  84       }
  85       web_views_.push_back(make_linked_ptr(new WebViewImpl(
  86           view.id, devtools_http_client_->browser_info(), client.Pass())));
  87     }
  88   }
  89
  90   std::list<std::string> web_view_ids_tmp;
  91   for (WebViewList::const_iterator web_view_iter = web_views_.begin();
  92        web_view_iter != web_views_.end(); ++web_view_iter) {
  93     web_view_ids_tmp.push_back((*web_view_iter)->GetId());
  94   }
  95   web_view_ids->swap(web_view_ids_tmp);
  96   return Status(kOk);
  97 }
  98
  99 Status ChromeImpl::GetWebViewById(const std::string& id, WebView** web_view) {
 100   for (WebViewList::iterator it = web_views_.begin();
 101        it != web_views_.end(); ++it) {
 102     if ((*it)->GetId() == id) {
 103       *web_view = (*it).get();
 104       return Status(kOk);
 105     }
 106   }
 107   return Status(kUnknownError, "web view not found");
 108 }
 109
 110 Status ChromeImpl::CloseWebView(const std::string& id) {
 111   Status status = devtools_http_client_->CloseWebView(id);
 112   if (status.IsError())
 113     return status;
 114   for (WebViewList::iterator iter = web_views_.begin();
 115        iter != web_views_.end(); ++iter) {
 116     if ((*iter)->GetId() == id) {
 117       web_views_.erase(iter);
 118       break;
 119     }
 120   }
 121   return Status(kOk);
 122 }
 123
 124 Status ChromeImpl::ActivateWebView(const std::string& id) {
 125   return devtools_http_client_->ActivateWebView(id);
 126 }
 127
 128 Status ChromeImpl::Quit() {
 129   Status status = QuitImpl();
 130   if (status.IsOk())
 131     quit_ = true;
 132   return status;
 133 }
 134
 135 ChromeImpl::ChromeImpl(
 136     scoped_ptr<DevToolsHttpClient> client,
 137     ScopedVector<DevToolsEventListener>& devtools_event_listeners,
 138     scoped_ptr<PortReservation> port_reservation)
 139     : quit_(false),
 140       devtools_http_client_(client.Pass()),
 141       port_reservation_(port_reservation.Pass()) {
 142   devtools_event_listeners_.swap(devtools_event_listeners);
 143 }