test_mock_with_web_server.h

// Copyright (c) 2006-2010 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROME_FRAME_TEST_MOCK_WITH_WEB_SERVER_H_
#define CHROME_FRAME_TEST_MOCK_WITH_WEB_SERVER_H_

#include <windows.h>

#include "gmock/gmock.h"
#include "chrome_frame/chrome_frame_automation.h"
#include "chrome_frame/test/chrome_frame_test_utils.h"

#define GMOCK_MUTANT_INCLUDE_LATE_OBJECT_BINDING
#include "testing/gmock_mutant.h"

namespace chrome_frame_test {

using ::testing::Expectation;
using ::testing::ExpectationSet;

// This class provides functionality to add expectations to IE full tab mode
// tests.
class MockWebBrowserEventSink : public chrome_frame_test::WebBrowserEventSink {
 public:
  // Needed to support PostTask.
  static bool ImplementsThreadSafeReferenceCounting() {
    return true;
  }

  MOCK_METHOD7_WITH_CALLTYPE(__stdcall, OnBeforeNavigate2,
                             void (IDispatch* dispatch,  // NOLINT
                                      VARIANT* url,
                                      VARIANT* flags,
                                      VARIANT* target_frame_name,
                                      VARIANT* post_data,
                                      VARIANT* headers,
                                      VARIANT_BOOL* cancel));

  MOCK_METHOD2_WITH_CALLTYPE(__stdcall, OnNavigateComplete2,
                             void (IDispatch* dispatch,  // NOLINT
                                   VARIANT* url));

  MOCK_METHOD5_WITH_CALLTYPE(__stdcall, OnNewWindow3,
                             void (IDispatch** dispatch,  // NOLINT
                                   VARIANT_BOOL* cancel,
                                   DWORD flags,
                                   BSTR url_context,
                                   BSTR url));

  MOCK_METHOD2_WITH_CALLTYPE(__stdcall, OnNewWindow2,
                             void (IDispatch** dispatch,  // NOLINT
                                   VARIANT_BOOL* cancel));

  MOCK_METHOD5_WITH_CALLTYPE(__stdcall, OnNavigateError,
                             void (IDispatch* dispatch,  // NOLINT
                                   VARIANT* url,
                                   VARIANT* frame_name,
                                   VARIANT* status_code,
                                   VARIANT* cancel));

  MOCK_METHOD2_WITH_CALLTYPE(__stdcall, OnFileDownload,
                             void (VARIANT_BOOL active_doc,
                                   VARIANT_BOOL* cancel));

  MOCK_METHOD0_WITH_CALLTYPE(__stdcall, OnQuit,
                             void ());

  MOCK_METHOD1(OnLoad, void (const wchar_t* url));  // NOLINT
  MOCK_METHOD1(OnLoadError, void (const wchar_t* url));  // NOLINT
  MOCK_METHOD3(OnMessage, void (const wchar_t* message,
                                const wchar_t* origin,
                                const wchar_t* source));  // NOLINT
  MOCK_METHOD2(OnNewBrowserWindow, void (IDispatch* dispatch,  // NOLINT
                                         const wchar_t* url));
  MOCK_METHOD2(OnWindowDetected, void (HWND hwnd,  // NOLINT
                                       const std::string& caption));

  // Test expectations
  ExpectationSet ExpectNavigationCardinality(const std::wstring& url,
                                             testing::Cardinality cardinality);
  ExpectationSet ExpectNavigation(const std::wstring& url);
  ExpectationSet ExpectNavigationAndSwitch(const std::wstring& url);
  ExpectationSet ExpectNavigationAndSwitchSequence(const std::wstring& url);
  ExpectationSet ExpectNewWindow(MockWebBrowserEventSink* new_window_mock);

};

}  // namespace chrome_frame_test

#endif  // CHROME_FRAME_TEST_MOCK_WITH_WEB_SERVER_H_