Makes the tab restore service contain any windows that were open at...

chrome/test/testing_profile.h

 // Copyright (c) 2006-2008 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_TEST_TESTING_PROFILE_H__
 #define CHROME_TEST_TESTING_PROFILE_H__
 
 #include "base/base_paths.h"
 #include "base/path_service.h"
 #include "base/file_util.h"
 #include "chrome/browser/bookmarks/bookmark_model.h"
 #include "chrome/browser/browser_prefs.h"
 #include "chrome/browser/history/history.h"
 #include "chrome/browser/profile.h"
+#include "chrome/browser/sessions/session_service.h"
 #include "chrome/browser/template_url_model.h"
 #include "chrome/common/pref_service.h"
 
 class TestingProfile : public Profile {
  public:
   TestingProfile();
   // Creates a new profile by adding |count| to the end of the path. Use this
   // when you need to have more than one TestingProfile running at the same
   // time.
   explicit TestingProfile(int count);
@@ skipping 75 matching lines @@
   }
   virtual DownloadManager* GetDownloadManager() {
     return NULL;
   }
   virtual bool HasCreatedDownloadManager() const {
     return false;
   }
   virtual URLRequestContext* GetRequestContext() {
     return NULL;
   }
+  void set_session_service(SessionService* session_service) {
+    session_service_ = session_service;
+  }
   virtual SessionService* GetSessionService() {
-    return NULL;
+    return session_service_.get();
   }
   virtual void ShutdownSessionService() {
   }
   virtual bool HasSessionService() const {
-    return false;
+    return (session_service_.get() != NULL);
   }
   virtual std::wstring GetName() {
     return std::wstring();
   }
   virtual void SetName(const std::wstring& name) {
   }
   virtual std::wstring GetID() {
     return id_;
   }
   virtual void SetID(const std::wstring& id) {
     id_ = id;
   }
+  void set_last_session_exited_cleanly(bool value) {
+    last_session_exited_cleanly_ = value;
+  }
   virtual bool DidLastSessionExitCleanly() {
-    return true;
+    return last_session_exited_cleanly_;
   }
   virtual void MergeResourceString(int message_id,
                                    std::wstring* output_string) {
   }
   virtual void MergeResourceInteger(int message_id, int* output_value) {
   }
   virtual void MergeResourceBoolean(int message_id, bool* output_value) {
   }
   virtual BookmarkModel* GetBookmarkModel() {
     return bookmark_bar_model_.get();
@@ skipping 37 matching lines @@
 
   // The history service. Only created if CreateHistoryService is invoked.
   scoped_refptr<HistoryService> history_service_;
 
   // The BookmarkModel. Only created if CreateBookmarkModel is invoked.
   scoped_ptr<BookmarkModel> bookmark_bar_model_;
 
   // The TemplateURLFetcher. Only created if CreateTemplateURLModel is invoked.
   scoped_ptr<TemplateURLModel> template_url_model_;
 
+  // The SessionService. Defaults to NULL, but can be set using the setter.
+  scoped_refptr<SessionService> session_service_;
+
   // Do we have a history service? This defaults to the value of
   // history_service, but can be explicitly set.
   bool has_history_service_;
 
   std::wstring id_;
 
   bool off_the_record_;
+
+  // Did the last session exit cleanly? Default is true.
+  bool last_session_exited_cleanly_;
 };
 
 #endif  // CHROME_TEST_TESTING_PROFILE_H__