@@ -54,6 +54,7 @@ - (void)testRunCommand {
5454 id mockedDataSource = PFStrictProtocolMock (@protocol (PFInstallationIdentifierStoreProvider));
5555 id mockedSession = PFStrictClassMock ([PFURLSession class ]);
5656 id mockedRequestConstructor = PFStrictClassMock ([PFCommandURLRequestConstructor class ]);
57+ id mockedNotificationCenter = PFStrictClassMock ([NSNotificationCenter class ]);
5758
5859 id mockedCommand = PFStrictClassMock ([PFRESTCommand class ]);
5960 id mockedCommandResult = PFStrictClassMock ([PFCommandResult class ]);
@@ -71,7 +72,8 @@ - (void)testRunCommand {
7172
7273 PFURLSessionCommandRunner *commandRunner = [[PFURLSessionCommandRunner alloc ] initWithDataSource: mockedDataSource
7374 session: mockedSession
74- requestConstructor: mockedRequestConstructor];
75+ requestConstructor: mockedRequestConstructor
76+ notificationCenter: mockedNotificationCenter];
7577
7678 XCTestExpectation *expecatation = [self currentSelectorTestExpectation ];
7779 [[commandRunner runCommandAsync: mockedCommand withOptions: 0 ] continueWithBlock: ^id (BFTask *task) {
@@ -89,14 +91,16 @@ - (void)testRunCommandCancel {
8991 id mockedDataSource = PFStrictProtocolMock (@protocol (PFInstallationIdentifierStoreProvider));
9092 id mockedSession = PFStrictClassMock ([PFURLSession class ]);
9193 id mockedRequestConstructor = PFStrictClassMock ([PFCommandURLRequestConstructor class ]);
94+ id mockedNotificationCenter = PFStrictClassMock ([NSNotificationCenter class ]);
9295
9396 id mockedCommand = PFStrictClassMock ([PFRESTCommand class ]);
9497
9598 OCMStub ([mockedSession invalidateAndCancel ]);
9699
97100 PFURLSessionCommandRunner *commandRunner = [[PFURLSessionCommandRunner alloc ] initWithDataSource: mockedDataSource
98101 session: mockedSession
99- requestConstructor: mockedRequestConstructor];
102+ requestConstructor: mockedRequestConstructor
103+ notificationCenter: mockedNotificationCenter];
100104
101105 BFCancellationTokenSource *cancellationToken = [BFCancellationTokenSource cancellationTokenSource ];
102106 [cancellationToken cancel ];
@@ -117,6 +121,7 @@ - (void)testRunCommandRetry {
117121 id mockedDataSource = PFStrictProtocolMock (@protocol (PFInstallationIdentifierStoreProvider));
118122 id mockedSession = PFStrictClassMock ([PFURLSession class ]);
119123 id mockedRequestConstructor = PFStrictClassMock ([PFCommandURLRequestConstructor class ]);
124+ id mockedNotificationCenter = PFStrictClassMock ([NSNotificationCenter class ]);
120125
121126 id mockedCommand = PFStrictClassMock ([PFRESTCommand class ]);
122127
@@ -140,7 +145,8 @@ - (void)testRunCommandRetry {
140145
141146 PFURLSessionCommandRunner *commandRunner = [[PFURLSessionCommandRunner alloc ] initWithDataSource: mockedDataSource
142147 session: mockedSession
143- requestConstructor: mockedRequestConstructor];
148+ requestConstructor: mockedRequestConstructor
149+ notificationCenter: mockedNotificationCenter];
144150 commandRunner.initialRetryDelay = DBL_MIN; // Lets not needlessly sleep here.
145151
146152 XCTestExpectation *expecatation = [self currentSelectorTestExpectation ];
@@ -162,6 +168,7 @@ - (void)testRunFileUpload {
162168 id mockedDataSource = PFStrictProtocolMock (@protocol (PFInstallationIdentifierStoreProvider));
163169 id mockedSession = PFStrictClassMock ([PFURLSession class ]);
164170 id mockedRequestConstructor = PFStrictClassMock ([PFCommandURLRequestConstructor class ]);
171+ id mockedNotificationCenter = PFStrictClassMock ([NSNotificationCenter class ]);
165172
166173 id mockedCommand = PFStrictClassMock ([PFRESTCommand class ]);
167174 id mockedCommandResult = PFStrictClassMock ([PFCommandResult class ]);
@@ -191,7 +198,8 @@ - (void)testRunFileUpload {
191198
192199 PFURLSessionCommandRunner *commandRunner = [[PFURLSessionCommandRunner alloc ] initWithDataSource: mockedDataSource
193200 session: mockedSession
194- requestConstructor: mockedRequestConstructor];
201+ requestConstructor: mockedRequestConstructor
202+ notificationCenter: mockedNotificationCenter];
195203
196204 XCTestExpectation *expecatation = [self currentSelectorTestExpectation ];
197205 [[commandRunner runFileUploadCommandAsync: mockedCommand
@@ -213,6 +221,7 @@ - (void)testLocalIdResolution {
213221 id mockedDataSource = PFStrictProtocolMock (@protocol (PFInstallationIdentifierStoreProvider));
214222 id mockedSession = PFStrictClassMock ([PFURLSession class ]);
215223 id mockedRequestConstructor = PFStrictClassMock ([PFCommandURLRequestConstructor class ]);
224+ id mockedNotificationCenter = PFStrictClassMock ([NSNotificationCenter class ]);
216225
217226 id mockedCommand = PFStrictClassMock ([PFRESTCommand class ]);
218227 id mockedCommandResult = PFStrictClassMock ([PFCommandResult class ]);
@@ -230,7 +239,8 @@ - (void)testLocalIdResolution {
230239
231240 PFURLSessionCommandRunner *commandRunner = [[PFURLSessionCommandRunner alloc ] initWithDataSource: mockedDataSource
232241 session: mockedSession
233- requestConstructor: mockedRequestConstructor];
242+ requestConstructor: mockedRequestConstructor
243+ notificationCenter: mockedNotificationCenter];
234244
235245 XCTestExpectation *expecatation = [self currentSelectorTestExpectation ];
236246 [[commandRunner runCommandAsync: mockedCommand withOptions: 0 ] continueWithBlock: ^id (BFTask *task) {
0 commit comments