chrome_frame/test/run_all_unittests.cc

   1 // Copyright (c) 2010 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 #include <atlbase.h>
   6 #include <iostream>
   7
   8 #include "base/path_service.h"
   9 #include "base/platform_thread.h"
  10 #include "base/process_util.h"
  11 #include "base/sys_info.h"
  12 #include "chrome/common/env_vars.h"
  13 #include "base/test/test_suite.h"
  14 #include "base/command_line.h"
  15 #include "chrome/common/chrome_paths.h"
  16
  17 #include "chrome_frame/test/chrome_frame_test_utils.h"
  18 #include "chrome_frame/test/http_server.h"
  19 #include "chrome_frame/test_utils.h"
  20 #include "chrome_frame/utils.h"
  21
  22 // To enable ATL-based code to run in this module
  23 class ChromeFrameUnittestsModule
  24     : public CAtlExeModuleT<ChromeFrameUnittestsModule> {
  25  public:
  26   static HRESULT InitializeCom() {
  27     return CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
  28   }
  29 };
  30
  31 ChromeFrameUnittestsModule _AtlModule;
  32
  33 const wchar_t kNoRegistrationSwitch[] = L"no-registration";
  34
  35 // Causes the test executable to just run the web server and quit when the
  36 // server is killed. Useful for debugging tests.
  37 const wchar_t kRunAsServer[] = L"server";
  38
  39 void PureCall() {
  40   __debugbreak();
  41 }
  42
  43 int main(int argc, char **argv) {
  44   base::EnableTerminationOnHeapCorruption();
  45   PlatformThread::SetName("ChromeFrame tests");
  46
  47   _set_purecall_handler(PureCall);
  48
  49   TestSuite test_suite(argc, argv);
  50
  51   if (CommandLine::ForCurrentProcess()->HasSwitch(kRunAsServer)) {
  52     ChromeFrameHTTPServer server;
  53     server.SetUp();
  54     GURL server_url(server.server()->TestServerPage(""));
  55     std::cout << std::endl
  56               << "Server waiting on " << server_url.spec().c_str()
  57               << std::endl << std::endl
  58               << "Test output will be written to "
  59               << server.server()->GetDataDirectory().value().c_str() << "\\dump"
  60               << std::endl << std::endl
  61               << "Hit Ctrl-C or navigate to "
  62               << server_url.spec().c_str() << "kill to shut down the server."
  63               << std::endl;
  64     server.WaitToFinish(INFINITE);
  65     server.TearDown();
  66     std::cout << "Server stopped.";
  67     return 0;
  68   }
  69
  70   SetConfigBool(kChromeFrameHeadlessMode, true);
  71
  72   base::ProcessHandle crash_service = chrome_frame_test::StartCrashService();
  73   int ret = -1;
  74   // If mini_installer is used to register CF, we use the switch
  75   // --no-registration to avoid repetitive registration.
  76   if (CommandLine::ForCurrentProcess()->HasSwitch(kNoRegistrationSwitch)) {
  77     ret = test_suite.Run();
  78   } else {
  79     // Register paths needed by the ScopedChromeFrameRegistrar.
  80     chrome::RegisterPathProvider();
  81
  82     // This will register the chrome frame in the build directory. It currently
  83     // leaves that chrome frame registered once the tests are done. It must be
  84     // constructed AFTER the TestSuite is created since TestSuites create THE
  85     // AtExitManager.
  86     // TODO(robertshield): Make these tests restore the original registration
  87     // once done.
  88     ScopedChromeFrameRegistrar registrar;
  89     ret = test_suite.Run();
  90   }
  91
  92   DeleteConfigValue(kChromeFrameHeadlessMode);
  93
  94   if (crash_service)
  95     base::KillProcess(crash_service, 0, false);
  96 }