Improve token expiration test

ably · Jun 5, 2018 · 936955b · 936955b
1 parent c7efe08
commit 936955b
Showing 1 changed file with 3 additions and 9 deletions.
diff --git a/Spec/Auth.swift b/Spec/Auth.swift
@@ -3718,20 +3718,14 @@ class Auth : QuickSpec {
                     it ("receives a 40142 error from the server") {
                         waitUntil(timeout: testTimeout) { done in
                             client.connection.once(.connected) { stateChange in
-                                delay(tokenDuration + 1) {
+                                client.connection.once(.disconnected) { stateChange in
+                                    expect(stateChange!.reason!.code).to(equal(40142))
+                                    expect(stateChange!.reason!.description).to(contain("Key/token status changed (expire)"))
                                     done()
                                 }
                             }
                             client.connect()
                         }
-                        waitUntil(timeout: testTimeout) { done in
-                            client.connection.once(.disconnected) { stateChange in
-                                expect(stateChange!.reason!.code).to(equal(40142))
-                                expect(stateChange!.reason!.description).to(contain("Key/token status changed (expire)"))
-                                done()
-                            }
-                        }
-
                     }
                 }