@@ -71,18 +71,11 @@ - (void)testLoadShowRewardedAd:(FLTAdRequest *)request
7171 });
7272 // Stub setting of FullScreenContentDelegate to invoke delegate callbacks.
7373 NSError *error = OCMClassMock ([NSError class ]);
74+ __block id <GADFullScreenContentDelegate> fullScreenContentDelegate;
7475 OCMStub ([rewardedClassMock setFullScreenContentDelegate: [OCMArg any ]])
7576 .andDo (^(NSInvocation *invocation) {
76- id <GADFullScreenContentDelegate> delegate;
77- [invocation getArgument: &delegate atIndex: 2 ];
78- XCTAssertEqual (delegate, ad);
79- [delegate adDidRecordImpression: rewardedClassMock];
80- [delegate adDidRecordClick: rewardedClassMock];
81- [delegate adDidDismissFullScreenContent: rewardedClassMock];
82- [delegate adWillPresentFullScreenContent: rewardedClassMock];
83- [delegate adWillDismissFullScreenContent: rewardedClassMock];
84- [delegate ad: rewardedClassMock
85- didFailToPresentFullScreenContentWithError: error];
77+ [invocation getArgument: &fullScreenContentDelegate atIndex: 2 ];
78+ XCTAssertEqual (fullScreenContentDelegate, ad);
8679 });
8780 GADResponseInfo *responseInfo = OCMClassMock ([GADResponseInfo class ]);
8881 OCMStub ([rewardedClassMock responseInfo ]).andReturn (responseInfo);
@@ -110,6 +103,12 @@ - (void)testLoadShowRewardedAd:(FLTAdRequest *)request
110103 handler (adValue);
111104 return YES ;
112105 }]]);
106+
107+ // Setup mock for UIApplication.sharedInstance
108+ id uiApplicationClassMock = OCMClassMock ([UIApplication class ]);
109+ OCMStub (ClassMethod ([uiApplicationClassMock sharedApplication ]))
110+ .andReturn (uiApplicationClassMock);
111+
113112 // Call load and check expected interactions with mocks.
114113 [ad load ];
115114
@@ -154,12 +153,31 @@ - (void)testLoadShowRewardedAd:(FLTAdRequest *)request
154153 presentFromRootViewController: [OCMArg isEqual: mockRootViewController]
155154 userDidEarnRewardHandler: [OCMArg any ]]);
156155
157- // Verify full screen callbacks.
156+ [fullScreenContentDelegate adWillPresentFullScreenContent: rewardedClassMock];
158157 OCMVerify ([mockManager adWillPresentFullScreenContent: [OCMArg isEqual: ad]]);
159- OCMVerify ([mockManager adDidDismissFullScreenContent: [OCMArg isEqual: ad]]);
160- OCMVerify ([mockManager adWillDismissFullScreenContent: [OCMArg isEqual: ad]]);
158+ // Verify that we hide status bar
159+ #pragma clang diagnostic push
160+ #pragma clang diagnostic ignored "-Wdeprecated-declarations"
161+ OCMVerify ([uiApplicationClassMock setStatusBarHidden: YES ]);
162+ #pragma clang diagnostic pop
163+
164+ [fullScreenContentDelegate adDidRecordImpression: rewardedClassMock];
161165 OCMVerify ([mockManager adDidRecordImpression: [OCMArg isEqual: ad]]);
166+
167+ [fullScreenContentDelegate adDidRecordClick: rewardedClassMock];
162168 OCMVerify ([mockManager adDidRecordClick: [OCMArg isEqual: ad]]);
169+
170+ [fullScreenContentDelegate adDidDismissFullScreenContent: rewardedClassMock];
171+ OCMVerify ([mockManager adDidDismissFullScreenContent: [OCMArg isEqual: ad]]);
172+
173+ [fullScreenContentDelegate adWillDismissFullScreenContent: rewardedClassMock];
174+ OCMVerify ([mockManager adWillDismissFullScreenContent: [OCMArg isEqual: ad]]);
175+ #pragma clang diagnostic push
176+ #pragma clang diagnostic ignored "-Wdeprecated-declarations"
177+ OCMVerify ([uiApplicationClassMock setStatusBarHidden: NO ]);
178+ #pragma clang diagnostic pop
179+
180+ [ad ad: rewardedClassMock didFailToPresentFullScreenContentWithError: error];
163181 OCMVerify ([mockManager
164182 didFailToPresentFullScreenContentWithError: [OCMArg isEqual: ad]
165183 error: [OCMArg isEqual: error]]);
0 commit comments