Ver código fonte

Another attempt at test reliability.

Jon Shier 7 anos atrás
pai
commit
69c5fde631
1 arquivos alterados com 3 adições e 3 exclusões
  1. 3 3
      Tests/SessionTests.swift

+ 3 - 3
Tests/SessionTests.swift

@@ -327,7 +327,7 @@ class SessionTestCase: BaseTestCase {
         // Then
         XCTAssertNotNil(response, "response should not be nil")
         XCTAssertTrue(request.isCancelled)
-        XCTAssertTrue((request.task == nil) || (request.task?.state == .canceling))
+        XCTAssertTrue((request.task == nil) || (request.task?.state == .canceling || request.task?.state == .completed))
         guard let error = request.error?.asAFError, case .explicitlyCancelled = error else {
             XCTFail("Request should have an .explicitlyCancelled error.")
             return
@@ -360,7 +360,7 @@ class SessionTestCase: BaseTestCase {
         // Then
         XCTAssertNotNil(response, "response should not be nil")
         XCTAssertTrue(request.isCancelled)
-        XCTAssertTrue((request.task == nil) || (request.task?.state == .canceling))
+        XCTAssertTrue((request.task == nil) || (request.task?.state == .canceling || request.task?.state == .completed))
         guard let error = request.error?.asAFError, case .explicitlyCancelled = error else {
             XCTFail("Request should have an .explicitlyCancelled error.")
             return
@@ -393,7 +393,7 @@ class SessionTestCase: BaseTestCase {
         // Then
         XCTAssertNotNil(response, "response should not be nil")
         XCTAssertTrue(request.isCancelled)
-        XCTAssertTrue((request.task == nil) || (request.task?.state == .canceling))
+        XCTAssertTrue((request.task == nil) || (request.task?.state == .canceling || request.task?.state == .completed))
         guard let error = request.error?.asAFError, case .explicitlyCancelled = error else {
             XCTFail("Request should have an .explicitlyCancelled error.")
             return