-
Notifications
You must be signed in to change notification settings - Fork 11
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
update test dependencies and add option to control downloading (#55)
- Loading branch information
Showing
11 changed files
with
338 additions
and
83 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,216 @@ | ||
| #include "gtest_mpi.hpp" | ||
| #include <gtest/gtest.h> | ||
| #include <iostream> | ||
| #include <memory> | ||
| #include <mpi.h> | ||
| #include <stdexcept> | ||
| #include <vector> | ||
|
|
||
| namespace gtest_mpi { | ||
|
|
||
| namespace { | ||
|
|
||
| class MPIListener : public testing::EmptyTestEventListener { | ||
| public: | ||
| using UnitTest = testing::UnitTest; | ||
| using TestCase = testing::TestCase; | ||
| using TestInfo = testing::TestInfo; | ||
| using TestPartResult = testing::TestPartResult; | ||
| using TestSuite = testing::TestSuite; | ||
|
|
||
| MPIListener(testing::TestEventListener *listener) | ||
| : listener_(listener), comm_(MPI_COMM_WORLD), gather_called_(false) { | ||
| MPI_Comm_dup(MPI_COMM_WORLD, &comm_); | ||
| int rank; | ||
| MPI_Comm_rank(comm_, &rank); | ||
| if (rank != 0) | ||
| listener_.reset(); | ||
| } | ||
|
|
||
| void OnTestProgramStart(const UnitTest &u) override { | ||
| if (listener_) | ||
| listener_->OnTestProgramStart(u); | ||
| } | ||
|
|
||
| void OnTestProgramEnd(const UnitTest &u) override { | ||
| if (listener_) | ||
| listener_->OnTestProgramEnd(u); | ||
| } | ||
|
|
||
| void OnTestStart(const TestInfo &test_info) override { | ||
| gather_called_ = false; | ||
| if (listener_) | ||
| listener_->OnTestStart(test_info); | ||
| } | ||
|
|
||
| void OnTestPartResult(const TestPartResult &test_part_result) override { | ||
| if (listener_) { | ||
| listener_->OnTestPartResult(test_part_result); | ||
| } else if (test_part_result.type() == TestPartResult::Type::kFatalFailure || | ||
| test_part_result.type() == | ||
| TestPartResult::Type::kNonFatalFailure) { | ||
| std::size_t fileIndex = strings_.size(); | ||
| strings_ += test_part_result.file_name(); | ||
| strings_ += '\0'; | ||
|
|
||
| std::size_t messageIndex = strings_.size(); | ||
| strings_ += test_part_result.message(); | ||
| strings_ += '\0'; | ||
|
|
||
| infos_.emplace_back(ResultInfo{test_part_result.type(), fileIndex, | ||
| test_part_result.line_number(), | ||
| messageIndex}); | ||
| } | ||
| } | ||
|
|
||
| void OnTestEnd(const TestInfo &test_info) override { | ||
| if(!gather_called_){ | ||
| std::cerr << "Missing GTEST_MPI_GUARD in test case!" << std::endl; | ||
| throw std::runtime_error("Missing GTEST_MPI_GUARD in test case!"); | ||
| } | ||
|
|
||
| if (listener_) | ||
| listener_->OnTestEnd(test_info); | ||
| } | ||
|
|
||
| void OnTestIterationStart(const UnitTest &u, int it) override { | ||
| if (listener_) | ||
| listener_->OnTestIterationStart(u, it); | ||
| } | ||
|
|
||
| void OnEnvironmentsSetUpStart(const UnitTest &u) override { | ||
| if (listener_) | ||
| listener_->OnEnvironmentsSetUpStart(u); | ||
| } | ||
|
|
||
| void OnEnvironmentsSetUpEnd(const UnitTest &u) override { | ||
| if (listener_) | ||
| listener_->OnEnvironmentsSetUpEnd(u); | ||
| } | ||
|
|
||
| void OnTestSuiteStart(const TestSuite &t) override { | ||
| if (listener_) | ||
| listener_->OnTestSuiteStart(t); | ||
| } | ||
|
|
||
| void OnTestDisabled(const TestInfo &t) override { | ||
| if (listener_) | ||
| listener_->OnTestDisabled(t); | ||
| } | ||
| void OnTestSuiteEnd(const TestSuite &t) override { | ||
| if (listener_) | ||
| listener_->OnTestSuiteEnd(t); | ||
| } | ||
|
|
||
| void OnEnvironmentsTearDownStart(const UnitTest &u) override { | ||
| if (listener_) | ||
| listener_->OnEnvironmentsTearDownStart(u); | ||
| } | ||
|
|
||
| void OnEnvironmentsTearDownEnd(const UnitTest &u) override { | ||
| if (listener_) | ||
| listener_->OnEnvironmentsTearDownEnd(u); | ||
| } | ||
|
|
||
| void OnTestIterationEnd(const UnitTest &u, int it) override { | ||
| if (listener_) | ||
| listener_->OnTestIterationEnd(u, it); | ||
| } | ||
|
|
||
| void GatherPartResults() { | ||
| gather_called_ = true; | ||
| int rank, n_proc; | ||
| MPI_Comm_rank(comm_, &rank); | ||
| MPI_Comm_size(comm_, &n_proc); | ||
|
|
||
| if (rank == 0) { | ||
| decltype(infos_) remoteInfos; | ||
| decltype(strings_) remoteStrings; | ||
| for (int r = 1; r < n_proc; ++r) { | ||
| MPI_Status status; | ||
| int count; | ||
|
|
||
| // Result infos | ||
| MPI_Probe(r, 0, comm_, &status); | ||
| MPI_Get_count(&status, MPI_CHAR, &count); | ||
| auto numResults = static_cast<std::size_t>(count) / | ||
| sizeof(decltype(remoteInfos)::value_type); | ||
| remoteInfos.resize(numResults); | ||
| MPI_Recv(remoteInfos.data(), count, MPI_BYTE, r, 0, comm_, | ||
| MPI_STATUS_IGNORE); | ||
|
|
||
| // Only continue if any results | ||
| if (numResults) { | ||
| // Get strings | ||
| MPI_Probe(r, 0, comm_, &status); | ||
| MPI_Get_count(&status, MPI_CHAR, &count); | ||
| auto stringSize = static_cast<std::size_t>(count) / | ||
| sizeof(decltype(remoteStrings)::value_type); | ||
| remoteStrings.resize(stringSize); | ||
| MPI_Recv(&remoteStrings[0], count, MPI_BYTE, r, 0, comm_, | ||
| MPI_STATUS_IGNORE); | ||
|
|
||
| // Create error for every remote fail | ||
| for (const auto &info : remoteInfos) { | ||
| if (info.type == TestPartResult::Type::kFatalFailure || | ||
| info.type == TestPartResult::Type::kNonFatalFailure) { | ||
| ADD_FAILURE_AT(&remoteStrings[info.fileIndex], info.lineNumber) | ||
| << "Rank " << r << ": " << &remoteStrings[info.messageIndex]; | ||
| } | ||
| } | ||
| } | ||
| } | ||
| } else { | ||
| MPI_Send(infos_.data(), | ||
| infos_.size() * sizeof(decltype(infos_)::value_type), MPI_BYTE, | ||
| 0, 0, comm_); | ||
|
|
||
| // Only send string if results exist | ||
| if (infos_.size()) { | ||
| MPI_Send(strings_.data(), | ||
| strings_.size() * sizeof(decltype(strings_)::value_type), | ||
| MPI_BYTE, 0, 0, comm_); | ||
| } | ||
| } | ||
|
|
||
| infos_.clear(); | ||
| strings_.clear(); | ||
| } | ||
|
|
||
| private: | ||
| struct ResultInfo { | ||
| TestPartResult::Type type; | ||
| std::size_t fileIndex; | ||
| int lineNumber; | ||
| std::size_t messageIndex; | ||
| }; | ||
|
|
||
| std::unique_ptr<testing::TestEventListener> listener_; | ||
| MPI_Comm comm_; | ||
| bool gather_called_; | ||
|
|
||
| std::vector<ResultInfo> infos_; | ||
| std::string strings_; | ||
| }; | ||
|
|
||
| MPIListener *globalMPIListener = nullptr; | ||
|
|
||
| } // namespace | ||
|
|
||
| void InitGoogleTestMPI(int *argc, char **argv) { | ||
|
|
||
| ::testing::InitGoogleTest(argc, argv); | ||
|
|
||
| auto &test_listeners = ::testing::UnitTest::GetInstance()->listeners(); | ||
|
|
||
| globalMPIListener = new MPIListener( | ||
| test_listeners.Release(test_listeners.default_result_printer())); | ||
|
|
||
| test_listeners.Append(globalMPIListener); | ||
| } | ||
|
|
||
| TestGuard CreateTestGuard() { | ||
| return TestGuard{[]() { globalMPIListener->GatherPartResults(); }}; | ||
| } | ||
|
|
||
| } // namespace gtest_mpi |
Oops, something went wrong.