update SDLIAPSession unit tests

1 files changed, 7 insertions, 10 deletions

diff --git a/SmartDeviceLinkTests/TransportSpecs/iAP/SDLIAPSessionSpec.m b/SmartDeviceLinkTests/TransportSpecs/iAP/SDLIAPSessionSpec.m
index 8b638afb0..8e6f119ed 100644
--- a/SmartDeviceLinkTests/TransportSpecs/iAP/SDLIAPSessionSpec.m
+++ b/SmartDeviceLinkTests/TransportSpecs/iAP/SDLIAPSessionSpec.m
@@ -10,34 +10,30 @@
 #import <Nimble/Nimble.h>
 #import <OCMock/OCMock.h>
 
-#import "EAAccessory+OCMock.h"
+#import "EAAccessory+OCMock.m"
 #import "SDLIAPSession.h"
 #import "SDLIAPConstants.h"
 
-
 QuickSpecBegin(SDLIAPSessionSpec)
 
 describe(@"SDLIAPSession", ^{
     __block EAAccessory *mockAccessory = nil;
+    __block id<SDLIAPSessionDelegate> mockDelegate = nil;
 
     describe(@"Initialization", ^{
         __block SDLIAPSession *testSession = nil;
 
         beforeEach(^{
             mockAccessory = [EAAccessory.class sdlCoreMock];
-            testSession = [[SDLIAPSession alloc] initWithAccessory:mockAccessory forProtocol:ControlProtocolString];
+            mockDelegate = OCMProtocolMock(@protocol(SDLIAPSessionDelegate));
+            testSession = [[SDLIAPSession alloc] initWithAccessory:mockAccessory forProtocol:ControlProtocolString iAPSessionDelegate:mockDelegate];
         });
 
         it(@"should init correctly", ^{
             expect(testSession.accessory).to(equal(mockAccessory));
             expect(testSession.protocolString).to(equal(ControlProtocolString));
-            expect(testSession.isInputStreamOpen).to(beFalse());
-            expect(testSession.isOutputStreamOpen).to(beFalse());
-        });
-
-        it(@"should get correctly", ^{
-            expect(testSession.isStopped).to(beTrue());
-            expect(testSession.connectionID).to(equal(0));
+            expect(testSession.bothStreamsOpen).to(beFalse());
+            expect(testSession.hasSpaceAvailable).to(beFalse());
             expect(testSession.isSessionInProgress).to(beFalse());
         });
     });
@@ -45,3 +41,4 @@ describe(@"SDLIAPSession", ^{
 
 QuickSpecEnd
 
+