| // Copyright 2022 The Chromium Authors |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "base/command_line.h" |
| #include "base/functional/bind.h" |
| #include "base/strings/stringprintf.h" |
| #include "base/test/launcher/unit_test_launcher.h" |
| #include "base/test/test_suite.h" |
| #include "base/test/test_switches.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| #include <iostream> |
| |
| // Update this when adding a new test to rust_test_interop_unittest.rs. |
| int kNumTests = 10; |
| |
| bool is_subprocess() { |
| // The test launching process spawns a subprocess to run tests, and it |
| // includes this flag. |
| return base::CommandLine::ForCurrentProcess()->HasSwitch( |
| base::kGTestFlagfileFlag); |
| } |
| |
| int main(int argc, char** argv) { |
| // Run tests in a single process so we can count the tests. |
| std::string single_process = base::StringPrintf("--test-launcher-jobs=1"); |
| // We verify that the test suite and test name written in the #[gtest] macro |
| // is being propagated to Gtest by using a test filter that matches on the |
| // test suites/names. |
| std::string filter = "--gtest_filter=Test.*:ExactSuite.ExactTest"; |
| |
| int my_argc = argc + 2; |
| char** my_argv = new char*[argc]; |
| for (int i = 0; i < argc; ++i) { |
| my_argv = argv; |
| } |
| my_argv[argc] = single_process.data(); |
| my_argv[argc + 1] = filter.data(); |
| |
| base::TestSuite test_suite(my_argc, my_argv); |
| |
| int result = base::LaunchUnitTests( |
| my_argc, my_argv, |
| base::BindOnce(&base::TestSuite::Run, base::Unretained(&test_suite))); |
| |
| if (is_subprocess()) { |
| // Double-check that we actually ran all the tests. If this fails we'll see |
| // all the tests marked as "fail on exit" since the whole process is |
| // considered a failure. |
| auto succeed = testing::UnitTest::GetInstance()->successful_test_count(); |
| int expected_success = kNumTests; |
| if (succeed != expected_success) { |
| std::cerr << "***ERROR***: Expected " << expected_success |
| << " tests to succeed, but we saw: " << succeed << '\n'; |
| return 1; |
| } else { |
| std::cerr << "***OK***: Ran " << succeed << " tests, yay!\n"; |
| } |
| } |
| |
| return result; |
| } |