Index: ios/chrome/app/deferred_initialization_runner_unittest.mm |
diff --git a/ios/chrome/app/deferred_initialization_runner_unittest.mm b/ios/chrome/app/deferred_initialization_runner_unittest.mm |
new file mode 100644 |
index 0000000000000000000000000000000000000000..0481ea9cf1edfa5c39669d3776ed24fb8fc09dd8 |
--- /dev/null |
+++ b/ios/chrome/app/deferred_initialization_runner_unittest.mm |
@@ -0,0 +1,107 @@ |
+// Copyright 2016 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#import "ios/chrome/app/deferred_initialization_runner.h" |
+ |
+#include "base/time/time.h" |
+#include "testing/platform_test.h" |
+#import "base/test/ios/wait_util.h" |
+ |
+TEST(DeferredInitializationRunnerTest, TestSharedInstance) { |
+ EXPECT_TRUE([DeferredInitializationRunner sharedInstance]); |
+ // Cancelling a non-existing block does nothing. |
+ [[DeferredInitializationRunner sharedInstance] |
+ cancelBlockNamed:@"Invalid Name"]; |
+} |
+ |
+TEST(DeferredInitializationRunnerTest, TestRunBlock) { |
+ // Setup. |
+ __block BOOL quickFlag = NO; |
+ __block BOOL slowFlag = NO; |
+ ProceduralBlock quickBlock = ^() { |
+ EXPECT_FALSE(quickFlag); |
+ quickFlag = YES; |
+ }; |
+ ConditionBlock quickBlockRun = ^BOOL { |
+ return quickFlag; |
+ }; |
+ ProceduralBlock slowBlock = ^() { |
+ EXPECT_FALSE(slowFlag); |
+ slowFlag = YES; |
+ }; |
+ DeferredInitializationRunner* runner = |
+ [DeferredInitializationRunner sharedInstance]; |
+ |
+ // Action. |
+ [runner enqueueBlockNamed:@"quick block" block:quickBlock]; |
+ [runner enqueueBlockNamed:@"slow block" block:slowBlock]; |
+ |
+ // Test. |
+ // Waits until |quickBlock| is run. |
+ base::test::ios::WaitUntilCondition(quickBlockRun); |
+ EXPECT_TRUE(quickFlag); |
+ EXPECT_FALSE(slowFlag); |
+ EXPECT_EQ(1U, [runner numberOfBlocksRemaining]); |
+ [runner runBlockIfNecessary:@"quick block"]; |
+ [runner runBlockIfNecessary:@"slow block"]; |
+ EXPECT_TRUE(quickFlag); |
+ EXPECT_TRUE(slowFlag); |
+ EXPECT_EQ(0U, [runner numberOfBlocksRemaining]); |
+} |
+ |
+TEST(DeferredInitializationRunnerTest, TestCancelBlock) { |
+ // Setup. |
+ __block BOOL blockFinished = NO; |
+ DeferredInitializationRunner* runner = |
+ [DeferredInitializationRunner sharedInstance]; |
+ [runner enqueueBlockNamed:@"cancel me" |
+ block:^() { |
+ blockFinished = YES; |
+ }]; |
+ ASSERT_EQ(1U, [runner numberOfBlocksRemaining]); |
+ |
+ // Action. |
+ [runner cancelBlockNamed:@"cancel me"]; |
+ |
+ // Test. |
+ EXPECT_FALSE(blockFinished); |
+ EXPECT_EQ(0U, [runner numberOfBlocksRemaining]); |
+} |
+ |
+TEST(DeferredInitializationRunnerTest, TestCancelledBlockDoNothing) { |
+ // Setup. |
+ __block BOOL blockFinished = NO; |
+ DeferredInitializationRunner* runner = |
+ [DeferredInitializationRunner sharedInstance]; |
+ [runner enqueueBlockNamed:@"cancel me" |
+ block:^() { |
+ blockFinished = YES; |
+ }]; |
+ |
+ // Action. |
+ [runner cancelBlockNamed:@"cancel me"]; |
+ [runner runBlockIfNecessary:@"cancel me"]; |
+ |
+ // Test. |
+ EXPECT_FALSE(blockFinished); |
+} |
+ |
+TEST(DeferredInitializationRunnerTest, TestSecondBlockInvalidatesFirst) { |
+ // Setup. |
+ __block int blockRunCount = 0; |
+ ProceduralBlock runBlock = ^() { |
+ ++blockRunCount; |
+ }; |
+ DeferredInitializationRunner* runner = |
+ [DeferredInitializationRunner sharedInstance]; |
+ |
+ // Action. |
+ [runner enqueueBlockNamed:@"multiple" block:runBlock]; |
+ [runner enqueueBlockNamed:@"multiple" block:runBlock]; |
+ |
+ // Test. |
+ [runner runBlockIfNecessary:@"multiple"]; |
+ EXPECT_EQ(0U, [runner numberOfBlocksRemaining]); |
+ EXPECT_EQ(1, blockRunCount); |
+} |