summaryrefslogtreecommitdiffstats
path: root/chrome/browser/cocoa/browser_test_helper.h
blob: b7e3e64d4d5e32ee05cdb7cb478dbc181e18e5c3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
// Copyright (c) 2009 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_BROWSER_COCOA_BROWSER_TEST_HELPER_H_
#define CHROME_BROWSER_COCOA_BROWSER_TEST_HELPER_H_

#include "chrome/browser/browser.h"
#include "chrome/browser/chrome_thread.h"
#include "chrome/browser/profile.h"
#include "chrome/test/testing_profile.h"

// Base class which contains a valid Browser*.  Lots of boilerplate to
// recycle between unit test classes.
//
// TODO(jrg): move up a level (chrome/browser/cocoa -->
// chrome/browser), and use in non-Mac unit tests such as
// back_forward_menu_model_unittest.cc,
// navigation_controller_unittest.cc, ..
class BrowserTestHelper {
 public:
  BrowserTestHelper()
      : ui_thread_(ChromeThread::UI, &message_loop_),
        file_thread_(ChromeThread::FILE, &message_loop_) {
    profile_.reset(new TestingProfile());
    profile_->CreateBookmarkModel(true);
    profile_->BlockUntilBookmarkModelLoaded();

    // TODO(shess): These are needed in case someone creates a browser
    // window off of browser_.  pkasting indicates that other
    // platforms use a stub |BrowserWindow| and thus don't need to do
    // this.
    // http://crbug.com/39725
    profile_->CreateAutocompleteClassifier();
    profile_->CreateTemplateURLModel();

    browser_.reset(new Browser(Browser::TYPE_NORMAL, profile_.get()));
  }

  ~BrowserTestHelper() {
    // Delete the testing profile on the UI thread. But first release the
    // browser, since it may trigger accesses to the profile upon destruction.
    browser_.reset(NULL);
    message_loop_.DeleteSoon(FROM_HERE, profile_.release());
    message_loop_.RunAllPending();
  }

  TestingProfile* profile() const { return profile_.get(); }
  Browser* browser() const { return browser_.get(); }

  // Creates the browser window. To close this window call |CloseBrowserWindow|.
  // Do NOT call close directly on the window.
  BrowserWindow* CreateBrowserWindow() {
    browser_->CreateBrowserWindow();
    return browser_->window();
  }

  // Closes the window for this browser.
  void CloseBrowserWindow() {
    // Check to make sure a window was actually created.
    DCHECK(browser_->window());
    browser_->CloseAllTabs();
    browser_->CloseWindow();
    browser_.release();
  }

 private:
  scoped_ptr<TestingProfile> profile_;
  scoped_ptr<Browser> browser_;
  MessageLoopForUI message_loop_;
  ChromeThread ui_thread_;
  ChromeThread file_thread_;
};

#endif  // CHROME_BROWSER_COCOA_BROWSER_TEST_HELPER_H_