chrome/test/chromedriver/session_unittest.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 <list>
   6 #include <string>
   7
   8 #include "base/memory/scoped_ptr.h"
   9 #include "chrome/test/chromedriver/chrome/status.h"
  10 #include "chrome/test/chromedriver/chrome/stub_chrome.h"
  11 #include "chrome/test/chromedriver/chrome/stub_web_view.h"
  12 #include "chrome/test/chromedriver/session.h"
  13 #include "testing/gtest/include/gtest/gtest.h"
  14
  15 namespace {
  16
  17 class MockChrome : public StubChrome {
  18  public:
  19   MockChrome() : web_view_("1") {}
  20   virtual ~MockChrome() {}
  21
  22   virtual Status GetWebViewById(const std::string& id,
  23                                 WebView** web_view) OVERRIDE {
  24     if (id == web_view_.GetId()) {
  25       *web_view = &web_view_;
  26       return Status(kOk);
  27     }
  28     return Status(kUnknownError);
  29   }
  30
  31  private:
  32   StubWebView web_view_;
  33 };
  34
  35 }  // namespace
  36
  37 TEST(Session, GetTargetWindowNoChrome) {
  38   Session session("1");
  39   WebView* web_view;
  40   ASSERT_EQ(kNoSuchWindow, session.GetTargetWindow(&web_view).code());
  41 }
  42
  43 TEST(Session, GetTargetWindowTargetWindowClosed) {
  44   scoped_ptr<Chrome> chrome(new MockChrome());
  45   Session session("1", chrome.Pass());
  46   session.window = "2";
  47   WebView* web_view;
  48   ASSERT_EQ(kNoSuchWindow, session.GetTargetWindow(&web_view).code());
  49 }
  50
  51 TEST(Session, GetTargetWindowTargetWindowStillOpen) {
  52   scoped_ptr<Chrome> chrome(new MockChrome());
  53   Session session("1", chrome.Pass());
  54   session.window = "1";
  55   WebView* web_view = NULL;
  56   ASSERT_EQ(kOk, session.GetTargetWindow(&web_view).code());
  57   ASSERT_TRUE(web_view);
  58 }