Test suite: fix intermittent failure of "ARTPushRegistererDelegate mu…

…st be implemented"
ably · Sep 27, 2019 · 3cbca52 · 3cbca52
1 parent 702c28e
commit 3cbca52
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/Spec/Push.swift b/Spec/Push.swift
@@ -16,13 +16,17 @@ class Push : QuickSpec {
         var rest: ARTRest!
         var mockHttpExecutor: MockHTTPExecutor!
         var storage: MockDeviceStorage!
+        var stateMachineDelegate: StateMachineDelegate!
 
         beforeEach {
             rest = ARTRest(key: "xxxx:xxxx")
+            rest.internal.resetDeviceSingleton()
             mockHttpExecutor = MockHTTPExecutor()
             rest.internal.httpExecutor = mockHttpExecutor
             storage = MockDeviceStorage()
             rest.internal.storage = storage
+            stateMachineDelegate = StateMachineDelegate()
+            rest.push.internal.activationMachine().delegate = stateMachineDelegate
         }
 
         // RSH2

diff --git a/Spec/PushActivationStateMachine.swift b/Spec/PushActivationStateMachine.swift
@@ -1598,7 +1598,7 @@ class StateMachineDelegate: NSObject, ARTPushRegistererDelegate {
 
 typealias ARTDeviceId = String
 
-class StateMachineDelegateCustomCallbacks: StateMachineDelegate {
+private class StateMachineDelegateCustomCallbacks: StateMachineDelegate {
 
     var onPushCustomRegister: ((ARTErrorInfo?, ARTDeviceDetails?) -> NSError?)?
     var onPushCustomDeregister: ((ARTErrorInfo?, ARTDeviceId?) -> NSError?)?