@@ -32,7 +32,7 @@ public class MockOneSignalClient: NSObject, IOneSignalClient {
3232 let lock = NSLock ( )
3333
3434 var mockResponses : [ String : [ String : Any ] ] = [ : ]
35- var mockFailureResponses : [ String : NSError ] = [ : ]
35+ var mockFailureResponses : [ String : OneSignalClientError ] = [ : ]
3636 public var lastHTTPRequest : OneSignalRequest ?
3737 public var networkRequestCount = 0
3838 public var executedRequests : [ OneSignalRequest ] = [ ]
@@ -90,7 +90,7 @@ public class MockOneSignalClient: NSObject, IOneSignalClient {
9090 remoteParamsOutcomes = [ : ]
9191 }
9292
93- public func execute( _ request: OneSignalRequest , onSuccess successBlock: @escaping OSResultSuccessBlock , onFailure failureBlock: @escaping OSFailureBlock ) {
93+ public func execute( _ request: OneSignalRequest , onSuccess successBlock: @escaping OSResultSuccessBlock , onFailure failureBlock: @escaping OSClientFailureBlock ) {
9494 print ( " 🧪 MockOneSignalClient execute called " )
9595
9696 if executeInstantaneously {
@@ -117,7 +117,7 @@ public class MockOneSignalClient: NSObject, IOneSignalClient {
117117 return stringified
118118 }
119119
120- func finishExecutingRequest( _ request: OneSignalRequest , onSuccess successBlock: OSResultSuccessBlock , onFailure failureBlock: OSFailureBlock ) {
120+ func finishExecutingRequest( _ request: OneSignalRequest , onSuccess successBlock: OSResultSuccessBlock , onFailure failureBlock: OSClientFailureBlock ) {
121121
122122 // TODO: This entire method needs to contained within the equivalent of @synchronized ❗️
123123 print ( " 🧪 completing HTTP request: \( request) " )
@@ -137,8 +137,8 @@ public class MockOneSignalClient: NSObject, IOneSignalClient {
137137 }
138138 if ( mockResponses [ stringifiedRequest] ) != nil {
139139 successBlock ( mockResponses [ stringifiedRequest] )
140- } else if ( mockFailureResponses [ stringifiedRequest] ) != nil {
141- failureBlock ( mockFailureResponses [ stringifiedRequest ] )
140+ } else if let response = mockFailureResponses [ stringifiedRequest] {
141+ failureBlock ( response )
142142 } else if fireSuccessForAllRequests {
143143 allRequestsHandled = false
144144 successBlock ( [ : ] )
@@ -166,7 +166,7 @@ public class MockOneSignalClient: NSObject, IOneSignalClient {
166166 mockResponses [ request] = response
167167 }
168168
169- public func setMockFailureResponseForRequest( request: String , error: NSError ) {
169+ public func setMockFailureResponseForRequest( request: String , error: OneSignalClientError ) {
170170 mockFailureResponses [ request] = error
171171 }
172172}
0 commit comments