chrome/browser/session_service_test_helper.h

   1 // Copyright (c) 2006-2008 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_SESSION_SERVICE_TEST_HELPER_H__
   6 #define CHROME_BROWSER_SESSION_SERVICE_TEST_HELPER_H__
   7
   8 #include <vector>
   9
  10 #include "base/ref_counted.h"
  11
  12 class SessionBackend;
  13 class SessionCommand;
  14 class SessionID;
  15 class SessionService;
  16 struct SessionTab;
  17 struct SessionWindow;
  18 struct TabNavigation;
  19
  20 // A simple class that makes writing SessionService related tests easier.
  21
  22 class SessionServiceTestHelper {
  23  public:
  24   explicit SessionServiceTestHelper() {}
  25
  26   explicit SessionServiceTestHelper(SessionService* service)
  27       : service_(service) {}
  28
  29   void RestoreSessionFromCommands(const std::vector<SessionCommand*>& commands,
  30                                   std::vector<SessionWindow*>* valid_windows);
  31
  32   void PrepareTabInWindow(const SessionID& window_id,
  33                           const SessionID& tab_id,
  34                           int visual_index,
  35                           bool select);
  36
  37   // Reads the contents of the last session.
  38   void ReadWindows(std::vector<SessionWindow*>* windows);
  39
  40   void AssertTabEquals(SessionID& window_id,
  41                        SessionID& tab_id,
  42                        int visual_index,
  43                        int nav_index,
  44                        int nav_count,
  45                        const SessionTab& session_tab);
  46
  47   void AssertTabEquals(int visual_index,
  48                        int nav_index,
  49                        int nav_count,
  50                        const SessionTab& session_tab);
  51
  52   void AssertNavigationEquals(const TabNavigation& expected,
  53                               const TabNavigation& actual);
  54
  55   void AssertSingleWindowWithSingleTab(
  56       const std::vector<SessionWindow*>& windows,
  57       int nav_count);
  58
  59   void set_service(SessionService* service) { service_ = service; }
  60   SessionService* service() { return service_.get(); }
  61
  62   SessionBackend* backend();
  63
  64  private:
  65   scoped_refptr<SessionService> service_;
  66
  67   DISALLOW_EVIL_CONSTRUCTORS(SessionServiceTestHelper);
  68 };
  69
  70 #endif  // CHROME_BROWSER_SESSION_SERVICE_TEST_HELPER_H__
  71