summaryrefslogtreecommitdiffstats
path: root/chrome/test/unit/chrome_test_suite.h
blob: 79c5e1936e8d2120b72f64e2c495b4ed51ea0738 (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
76
77
78
79
80
81
82
83
84
85
86
87
88
// 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_UNIT_CHROME_TEST_SUITE_H_
#define CHROME_TEST_UNIT_CHROME_TEST_SUITE_H_

#include "build/build_config.h"

#include "base/stats_table.h"
#include "base/file_util.h"
#include "base/path_service.h"
#include "base/scoped_nsautorelease_pool.h"
#include "base/test_suite.h"
#include "chrome/browser/browser_process.h"
#include "chrome/common/chrome_paths.h"
#include "chrome/common/chrome_switches.h"
#if defined(OS_WIN) || defined(OS_LINUX)
// TODO(port): Remove the #ifdef when ResourceBundle is ported.
#include "chrome/common/resource_bundle.h"
#endif
#include "chrome/test/testing_browser_process.h"

class ChromeTestSuite : public TestSuite {
public:
  ChromeTestSuite(int argc, char** argv) : TestSuite(argc, argv) {
  }

protected:

  virtual void Initialize() {
    base::ScopedNSAutoreleasePool autorelease_pool;

    TestSuite::Initialize();

    chrome::RegisterPathProvider();
    g_browser_process = new TestingBrowserProcess;

    // Notice a user data override, and otherwise default to using a custom
    // user data directory that lives alongside the current app.
    // NOTE: The user data directory will be erased before each UI test that
    //       uses it, in order to ensure consistency.
    std::wstring user_data_dir =
        CommandLine::ForCurrentProcess()->GetSwitchValue(
            switches::kUserDataDir);
    if (user_data_dir.empty() &&
      PathService::Get(base::DIR_EXE, &user_data_dir))
      file_util::AppendToPath(&user_data_dir, L"test_user_data");
    if (!user_data_dir.empty())
      PathService::Override(chrome::DIR_USER_DATA, user_data_dir);

#if defined(OS_WIN) || defined(OS_LINUX)
    // TODO(port): Remove the #ifdef when ResourceBundle is ported.
    //
    // Force unittests to run using en-us so if we test against string
    // output, it'll pass regardless of the system language.
    ResourceBundle::InitSharedInstance(L"en-us");
    ResourceBundle::GetSharedInstance().LoadThemeResources();
#endif

    // initialize the global StatsTable for unit_tests
    std::string statsfile = "unit_tests";
    std::string pid_string = StringPrintf("-%d", base::GetCurrentProcId());
    statsfile += pid_string;
    stats_table_ = new StatsTable(statsfile, 20, 200);
    StatsTable::set_current(stats_table_);
  }

  virtual void Shutdown() {
#if defined(OS_WIN) || defined(OS_LINUX)
    // TODO(port): Remove the #ifdef when ResourceBundle is ported.
    ResourceBundle::CleanupSharedInstance();
#endif

    delete g_browser_process;
    g_browser_process = NULL;

    // Tear down shared StatsTable; prevents unit_tests from leaking it.
    StatsTable::set_current(NULL);
    delete stats_table_;
    
    TestSuite::Shutdown();
  }

  StatsTable* stats_table_;
};

#endif // CHROME_TEST_UNIT_CHROME_TEST_SUITE_H_