1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41

base / test / test_listener_ios.mm [blame]

// 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