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 37