src/chrome_frame/test/reliability/reliability_test_suite.h

   1 // Copyright (c) 2012 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_FRAME_TEST_RELIABILITY_RELIABILITY_TEST_SUITE_H_
   6 #define CHROME_FRAME_TEST_RELIABILITY_RELIABILITY_TEST_SUITE_H_
   7
   8 #include "base/win/scoped_com_initializer.h"
   9 #include "chrome_frame/test/reliability/page_load_test.h"
  10 #include "chrome/test/ui/ui_test_suite.h"
  11
  12 class ReliabilityTestSuite : public UITestSuite {
  13  public:
  14   ReliabilityTestSuite(int argc, char** argv) : UITestSuite(argc, argv) {
  15   }
  16
  17  private:
  18   virtual void Initialize() OVERRIDE {
  19     com_initializer_.reset(new base::win::ScopedCOMInitializer(
  20         base::win::ScopedCOMInitializer::kMTA));
  21     SetPageRange(*CommandLine::ForCurrentProcess());
  22     UITestSuite::Initialize();
  23   }
  24
  25   virtual void Shutdown() OVERRIDE {
  26     UITestSuite::Shutdown();
  27     com_initializer_.reset();
  28   }
  29
  30   scoped_ptr<base::win::ScopedCOMInitializer> com_initializer_;
  31
  32   DISALLOW_COPY_AND_ASSIGN(ReliabilityTestSuite);
  33 };
  34
  35 #endif  // CHROME_FRAME_TEST_RELIABILITY_RELIABILITY_TEST_SUITE_H_
  36