Index: ui/snapshot/test/run_all_unittests.cc |
diff --git a/ui/snapshot/test/run_all_unittests.cc b/ui/snapshot/test/run_all_unittests.cc |
index 5bf3d5fda55f7892d387e4c8af35abf636a02c48..abb7563b18396e9b058ca8c879df1a3fc8794474 100644 |
--- a/ui/snapshot/test/run_all_unittests.cc |
+++ b/ui/snapshot/test/run_all_unittests.cc |
@@ -2,13 +2,15 @@ |
// Use of this source code is governed by a BSD-style license that can be |
// found in the LICENSE file. |
+#include "base/bind.h" |
#include "base/test/test_suite.h" |
+#include "base/test/unit_test_launcher.h" |
#include "ui/compositor/test/test_suite.h" |
int main(int argc, char** argv) { |
-#if defined(USE_AURA) |
- return ui::test::CompositorTestSuite(argc, argv).Run(); |
-#else |
- return base::TestSuite(argc, argv).Run(); |
-#endif |
+ ui::test::CompositorTestSuite test_suite(argc, argv); |
+ |
+ return base::LaunchUnitTests( |
+ argc, argv, base::Bind(&ui::test::CompositorTestSuite::Run, |
+ base::Unretained(&test_suite))); |
} |