-
Notifications
You must be signed in to change notification settings - Fork 739
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
add test to make sure updating cache policy in error interceptor work…
…s as advertised
- Loading branch information
1 parent
66e24ea
commit a48bfcf
Showing
2 changed files
with
132 additions
and
1 deletion.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
127 changes: 127 additions & 0 deletions
127
Tests/ApolloCacheDependentTests/CacheDependentInterceptorTests.swift
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,127 @@ | ||
| // | ||
| // CacheDependentInterceptorTests.swift | ||
| // ApolloCacheDependentTests | ||
| // | ||
| // Created by Ellen Shapiro on 12/9/20. | ||
| // Copyright © 2020 Apollo GraphQL. All rights reserved. | ||
| // | ||
|
|
||
| import XCTest | ||
| import Apollo | ||
| import ApolloTestSupport | ||
| import StarWarsAPI | ||
|
|
||
| class CacheDependentInterceptorTests: XCTestCase, CacheDependentTesting { | ||
| var cacheType: TestCacheProvider.Type { | ||
| InMemoryTestCacheProvider.self | ||
| } | ||
| var defaultWaitTimeout: TimeInterval = 5 | ||
|
|
||
|
|
||
| var cache: NormalizedCache! | ||
| var store: ApolloStore! | ||
| var client: ApolloClient! | ||
|
|
||
| override func setUpWithError() throws { | ||
| try super.setUpWithError() | ||
|
|
||
| cache = try makeNormalizedCache() | ||
| store = ApolloStore(cache: cache) | ||
| } | ||
|
|
||
| func testChangingCachePolicyInErrorInterceptorWorks() { | ||
| // Set up initial cache state | ||
| mergeRecordsIntoCache([ | ||
| "QUERY_ROOT": ["hero": Reference(key: "hero")], | ||
| "hero": ["__typename": "Droid", "name": "R2-D2"] | ||
| ]) | ||
|
|
||
| /// This interceptor will reroute anything that fails with a response code error to retry hitting only the cache | ||
| class RerouteToCacheErrorInterceptor: ApolloErrorInterceptor { | ||
| var handledError: Error? | ||
|
|
||
| func handleErrorAsync<Operation: GraphQLOperation>( | ||
| error: Error, | ||
| chain: RequestChain, | ||
| request: HTTPRequest<Operation>, | ||
| response: HTTPResponse<Operation>?, | ||
| completion: @escaping (Result<GraphQLResult<Operation.Data>, Error>) -> Void) { | ||
|
|
||
| self.handledError = error | ||
|
|
||
| switch error { | ||
| case ResponseCodeInterceptor.ResponseCodeError.invalidResponseCode: | ||
| request.cachePolicy = .returnCacheDataDontFetch | ||
| chain.retry(request: request, completion: completion) | ||
| default: | ||
| completion(.failure(error)) | ||
| } | ||
| } | ||
| } | ||
|
|
||
|
|
||
| class TestProvider: LegacyInterceptorProvider { | ||
| init(store: ApolloStore) { | ||
| super.init(client: self.mockClient, | ||
| store: store) | ||
| } | ||
|
|
||
| let mockClient: MockURLSessionClient = { | ||
| let client = MockURLSessionClient() | ||
| client.response = HTTPURLResponse(url: TestURL.mockServer.url, | ||
| statusCode: 401, | ||
| httpVersion: nil, | ||
| headerFields: nil) | ||
| client.data = Data() | ||
| return client | ||
| }() | ||
|
|
||
| let additionalInterceptor = RerouteToCacheErrorInterceptor() | ||
|
|
||
| override func additionalErrorInterceptor<Operation: GraphQLOperation>(for operation: Operation) -> ApolloErrorInterceptor? { | ||
| self.additionalInterceptor | ||
| } | ||
| } | ||
|
|
||
| let testProvider = TestProvider(store: self.store) | ||
| let network = RequestChainNetworkTransport(interceptorProvider: testProvider, | ||
| endpointURL: TestURL.mockServer.url) | ||
|
|
||
| let expectation = self.expectation(description: "Request sent") | ||
|
|
||
| // Send the initial request ignoring cache data so it doesn't initially get the data from the cache, | ||
| _ = network.send(operation: HeroNameQuery(), cachePolicy: .fetchIgnoringCacheData) { result in | ||
| defer { | ||
| expectation.fulfill() | ||
| } | ||
|
|
||
| // Check that the final result is what we expected | ||
| switch result { | ||
| case .failure(let error): | ||
| XCTFail("Unexpected error: \(error)") | ||
| case .success(let graphQLResult): | ||
| guard let heroName = graphQLResult.data?.hero?.name else { | ||
| XCTFail("Could not access hero name from returned result") | ||
| return | ||
| } | ||
|
|
||
| XCTAssertEqual(heroName, "R2-D2") | ||
| } | ||
|
|
||
| // Validate that there was a handled error before we went to the cache and we didn't just go straight to the cache | ||
| guard let handledError = testProvider.additionalInterceptor.handledError else { | ||
| XCTFail("No error was handled!") | ||
| return | ||
| } | ||
| switch handledError { | ||
| case ResponseCodeInterceptor.ResponseCodeError.invalidResponseCode(let response, _): | ||
| XCTAssertEqual(response?.statusCode, 401) | ||
| default: | ||
| XCTFail("Unexpected error on the additional error handler: \(handledError)") | ||
| } | ||
| } | ||
|
|
||
| self.wait(for: [expectation], timeout: self.defaultWaitTimeout) | ||
| } | ||
| } | ||
|
|