From 925e343e32f8df10799aad9feee86c0cab3b17ef Mon Sep 17 00:00:00 2001
From: Adlai Holler <guy.on.internet@me.com>
Date: Mon, 17 Oct 2016 11:28:48 -0700
Subject: [PATCH] Make ASNetworkImageNode test wait (#2406)

---
 AsyncDisplayKitTests/ASNetworkImageNodeTests.m | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/AsyncDisplayKitTests/ASNetworkImageNodeTests.m b/AsyncDisplayKitTests/ASNetworkImageNodeTests.m
index c9d807fbf9..b7086e6c85 100644
--- a/AsyncDisplayKitTests/ASNetworkImageNodeTests.m
+++ b/AsyncDisplayKitTests/ASNetworkImageNodeTests.m
@@ -37,9 +37,18 @@
 - (void)testThatProgressBlockIsSetAndClearedCorrectlyOnVisibility
 {
   node.URL = [NSURL URLWithString:@"http://imageA"];
+
+  // Enter preload range, wait for download start.
+  [[[downloader expect] andForwardToRealObject] downloadImageWithURL:[OCMArg isNotNil] callbackQueue:OCMOCK_ANY downloadProgress:OCMOCK_ANY completion:OCMOCK_ANY];
+  [node enterInterfaceState:ASInterfaceStatePreload];
+  [downloader verifyWithDelay:5];
+
+  // Make the node visible.
   [[downloader expect] setProgressImageBlock:[OCMArg isNotNil] callbackQueue:OCMOCK_ANY withDownloadIdentifier:@0];
   [node enterInterfaceState:ASInterfaceStateInHierarchy];
   [downloader verify];
+
+  // Make the node invisible.
   [[downloader expect] setProgressImageBlock:[OCMArg isNil] callbackQueue:OCMOCK_ANY withDownloadIdentifier:@0];
   [node exitInterfaceState:ASInterfaceStateInHierarchy];
   [downloader verify];