Fixed test case

Signed-off-by: NicoleYarroch <nicole@livio.io>

2 files changed, 2 insertions, 3 deletions

diff --git a/SmartDeviceLink/SDLLifecycleProtocolHandler.m b/SmartDeviceLink/SDLLifecycleProtocolHandler.m
index 2094a113f..49a789637 100644
--- a/SmartDeviceLink/SDLLifecycleProtocolHandler.m
+++ b/SmartDeviceLink/SDLLifecycleProtocolHandler.m
@@ -89,7 +89,6 @@ NS_ASSUME_NONNULL_BEGIN
     [self.rpcStartServiceTimeoutTimer start];
 }
 
-
 /// Called when the transport is closed.
 - (void)protocolDidClose:(SDLProtocol *)protocol {
     if (self.protocol != protocol) { return; }
diff --git a/SmartDeviceLinkTests/DevAPISpecs/SDLLifecycleProtocolHandlerSpec.m b/SmartDeviceLinkTests/DevAPISpecs/SDLLifecycleProtocolHandlerSpec.m
index 6cd4665ae..681ec9cc5 100644
--- a/SmartDeviceLinkTests/DevAPISpecs/SDLLifecycleProtocolHandlerSpec.m
+++ b/SmartDeviceLinkTests/DevAPISpecs/SDLLifecycleProtocolHandlerSpec.m
@@ -156,11 +156,11 @@ describe(@"SDLLifecycleProtocolHandler tests", ^{
             });
 
             it(@"should send a transport disconnected notification when the timer elapses", ^{
-                OCMExpect([mockNotificationDispatcher postNotificationName:[OCMArg isEqual:SDLTransportDidDisconnect] infoObject:[OCMArg isNil]]);
+                OCMExpect([mockProtocol stopWithCompletionHandler:[OCMArg any]]);
 
                 [testHandler protocolDidOpen:mockProtocol];
 
-                OCMVerifyAllWithDelay(mockNotificationDispatcher, 11.0);
+                OCMVerifyAllWithDelay(mockProtocol, 11.0);
             });
         });