src/chrome/browser/sessions/session_service_test_helper.h

   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 #ifndef CHROME_BROWSER_SESSIONS_SESSION_SERVICE_TEST_HELPER_H_
   6 #define CHROME_BROWSER_SESSIONS_SESSION_SERVICE_TEST_HELPER_H_
   7
   8 #include <string>
   9 #include <vector>
  10
  11 #include "base/basictypes.h"
  12 #include "base/memory/scoped_ptr.h"
  13 #include "chrome/browser/sessions/session_id.h"
  14
  15 class SessionBackend;
  16 class SessionCommand;
  17 class SessionService;
  18 struct SessionTab;
  19 struct SessionWindow;
  20
  21 namespace sessions {
  22 class SerializedNavigationEntry;
  23 }
  24
  25 // A simple class that makes writing SessionService related tests easier.
  26
  27 class SessionServiceTestHelper {
  28  public:
  29   explicit SessionServiceTestHelper();
  30   explicit SessionServiceTestHelper(SessionService* service);
  31   ~SessionServiceTestHelper();
  32
  33   void PrepareTabInWindow(const SessionID& window_id,
  34                           const SessionID& tab_id,
  35                           int visual_index,
  36                           bool select);
  37
  38   void SetTabExtensionAppID(const SessionID& window_id,
  39                             const SessionID& tab_id,
  40                             const std::string& extension_app_id);
  41
  42   void SetTabUserAgentOverride(const SessionID& window_id,
  43                                const SessionID& tab_id,
  44                                const std::string& user_agent_override);
  45
  46   void SetForceBrowserNotAliveWithNoWindows(
  47       bool force_browser_not_alive_with_no_windows);
  48
  49   // Reads the contents of the last session.
  50   void ReadWindows(std::vector<SessionWindow*>* windows,
  51                    SessionID::id_type* active_window_id);
  52
  53   void AssertTabEquals(SessionID& window_id,
  54                        SessionID& tab_id,
  55                        int visual_index,
  56                        int nav_index,
  57                        size_t nav_count,
  58                        const SessionTab& session_tab);
  59
  60   void AssertTabEquals(int visual_index,
  61                        int nav_index,
  62                        size_t nav_count,
  63                        const SessionTab& session_tab);
  64
  65   void AssertNavigationEquals(
  66       const sessions::SerializedNavigationEntry& expected,
  67       const sessions::SerializedNavigationEntry& actual);
  68
  69   void AssertSingleWindowWithSingleTab(
  70       const std::vector<SessionWindow*>& windows,
  71       size_t nav_count);
  72
  73   void SetService(SessionService* service);
  74   SessionService* ReleaseService() { return service_.release(); }
  75   SessionService* service() { return service_.get(); }
  76
  77   SessionBackend* backend();
  78
  79  private:
  80   scoped_ptr<SessionService> service_;
  81
  82   DISALLOW_COPY_AND_ASSIGN(SessionServiceTestHelper);
  83 };
  84
  85 #endif  // CHROME_BROWSER_SESSIONS_SESSION_SERVICE_TEST_HELPER_H_