chrome/test/chromedriver/session.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/session.h"
   6
   7 #include <list>
   8
   9 #include "base/lazy_instance.h"
  10 #include "base/threading/thread_local.h"
  11 #include "base/values.h"
  12 #include "chrome/test/chromedriver/chrome/chrome.h"
  13 #include "chrome/test/chromedriver/chrome/status.h"
  14 #include "chrome/test/chromedriver/chrome/version.h"
  15 #include "chrome/test/chromedriver/chrome/web_view.h"
  16 #include "chrome/test/chromedriver/logging.h"
  17
  18 namespace {
  19
  20 base::LazyInstance<base::ThreadLocalPointer<Session> >
  21     lazy_tls_session = LAZY_INSTANCE_INITIALIZER;
  22
  23 }  // namespace
  24
  25 FrameInfo::FrameInfo(const std::string& parent_frame_id,
  26                      const std::string& frame_id,
  27                      const std::string& chromedriver_frame_id)
  28     : parent_frame_id(parent_frame_id),
  29       frame_id(frame_id),
  30       chromedriver_frame_id(chromedriver_frame_id) {}
  31
  32 const base::TimeDelta Session::kDefaultPageLoadTimeout =
  33     base::TimeDelta::FromMinutes(5);
  34
  35 Session::Session(const std::string& id)
  36     : id(id),
  37       quit(false),
  38       detach(false),
  39       force_devtools_screenshot(false),
  40       sticky_modifiers(0),
  41       mouse_position(0, 0),
  42       page_load_timeout(kDefaultPageLoadTimeout),
  43       auto_reporting_enabled(false) {}
  44
  45 Session::Session(const std::string& id, scoped_ptr<Chrome> chrome)
  46     : id(id),
  47       quit(false),
  48       detach(false),
  49       force_devtools_screenshot(false),
  50       chrome(chrome.Pass()),
  51       sticky_modifiers(0),
  52       mouse_position(0, 0),
  53       page_load_timeout(kDefaultPageLoadTimeout),
  54       auto_reporting_enabled(false) {}
  55
  56 Session::~Session() {}
  57
  58 Status Session::GetTargetWindow(WebView** web_view) {
  59   if (!chrome)
  60     return Status(kNoSuchWindow, "no chrome started in this session");
  61
  62   Status status = chrome->GetWebViewById(window, web_view);
  63   if (status.IsError())
  64     status = Status(kNoSuchWindow, "target window already closed", status);
  65   return status;
  66 }
  67
  68 void Session::SwitchToTopFrame() {
  69   frames.clear();
  70 }
  71
  72 void Session::SwitchToParentFrame() {
  73   if (!frames.empty())
  74     frames.pop_back();
  75 }
  76
  77 void Session::SwitchToSubFrame(const std::string& frame_id,
  78                                const std::string& chromedriver_frame_id) {
  79   std::string parent_frame_id;
  80   if (!frames.empty())
  81     parent_frame_id = frames.back().frame_id;
  82   frames.push_back(FrameInfo(parent_frame_id, frame_id, chromedriver_frame_id));
  83 }
  84
  85 std::string Session::GetCurrentFrameId() const {
  86   if (frames.empty())
  87     return std::string();
  88   return frames.back().frame_id;
  89 }
  90
  91 std::vector<WebDriverLog*> Session::GetAllLogs() const {
  92   std::vector<WebDriverLog*> logs;
  93   for (ScopedVector<WebDriverLog>::const_iterator log = devtools_logs.begin();
  94        log != devtools_logs.end();
  95        ++log) {
  96     logs.push_back(*log);
  97   }
  98   if (driver_log)
  99     logs.push_back(driver_log.get());
 100   return logs;
 101 }
 102
 103 std::string Session::GetFirstBrowserError() const {
 104   for (ScopedVector<WebDriverLog>::const_iterator it = devtools_logs.begin();
 105        it != devtools_logs.end();
 106        ++it) {
 107     if ((*it)->type() == WebDriverLog::kBrowserType) {
 108       std::string message = (*it)->GetFirstErrorMessage();
 109       if (!message.empty())
 110         return message;
 111     }
 112   }
 113   return std::string();
 114 }
 115
 116 Session* GetThreadLocalSession() {
 117   return lazy_tls_session.Pointer()->Get();
 118 }
 119
 120 void SetThreadLocalSession(scoped_ptr<Session> session) {
 121   lazy_tls_session.Pointer()->Set(session.release());
 122 }