| // Copyright 2012 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/test/test_listener_ios.h" |
| |
| #import <Foundation/Foundation.h> |
| |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| // The iOS watchdog timer will kill an app that doesn't spin the main event |
| // loop often enough. This uses a Gtest TestEventListener to spin the current |
| // loop after each test finishes. However, if any individual test takes too |
| // long, it is still possible that the app will get killed. |
| |
| namespace { |
| |
| class IOSRunLoopListener : public testing::EmptyTestEventListener { |
| public: |
| void OnTestEnd(const testing::TestInfo& test_info) override; |
| }; |
| |
| void IOSRunLoopListener::OnTestEnd(const testing::TestInfo& test_info) { |
| @autoreleasepool { |
| // At the end of the test, spin the default loop for a moment. |
| NSDate* stop_date = [NSDate dateWithTimeIntervalSinceNow:0.001]; |
| [NSRunLoop.currentRunLoop runUntilDate:stop_date]; |
| } |
| } |
| |
| } // namespace |
| |
| namespace base::test_listener_ios { |
| |
| void RegisterTestEndListener() { |
| testing::TestEventListeners& listeners = |
| testing::UnitTest::GetInstance()->listeners(); |
| listeners.Append(new IOSRunLoopListener); |
| } |
| |
| } // namespace base::test_listener_ios |