Fix tests

configcat · Jan 29, 2020 · 100475a · 100475a
1 parent e9293cf
commit 100475a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 8 deletions.
diff --git a/Tests/AutoPollingTests.swift b/Tests/AutoPollingTests.swift
@@ -12,8 +12,6 @@ class AutoPollingTests: XCTestCase {
         let fetcher = ConfigFetcher(session: mockSession, apiKey: "", mode: mode)
         let policy = mode.accept(visitor: RefreshPolicyFactory(fetcher: fetcher, cache: InMemoryConfigCache()))
 
-        sleep(1)
-
         XCTAssertEqual("test", try policy.getConfiguration().get())
 
         sleep(3)
@@ -30,8 +28,6 @@ class AutoPollingTests: XCTestCase {
         let fetcher = ConfigFetcher(session: mockSession, apiKey: "", mode: mode)
         let policy = mode.accept(visitor: RefreshPolicyFactory(fetcher: fetcher, cache: InMemoryConfigCache()))
 
-        sleep(1)
-
         XCTAssertEqual("test", try policy.getConfiguration().get())
 
         sleep(3)
@@ -47,9 +43,7 @@ class AutoPollingTests: XCTestCase {
         let mode = PollingModes.autoPoll(autoPollIntervalInSeconds: 2)
         let fetcher = ConfigFetcher(session: mockSession, apiKey: "", mode: mode)
         let policy = mode.accept(visitor: RefreshPolicyFactory(fetcher: fetcher, cache: InMemoryConfigCache()))
-
-        sleep(1)
-
+
         XCTAssertEqual("test", try policy.getConfiguration().get())
 
         sleep(3)

diff --git a/Tests/Mock.swift b/Tests/Mock.swift
@@ -28,7 +28,7 @@ class MockURLSession: URLSession {
         completionHandler: @escaping CompletionHandler
         ) -> URLSessionDataTask {
         self.requests.append(request)
-        let response = self.responses.removeFirst()
+        let response = self.responses.count == 1 ? self.responses[0] : self.responses.removeFirst()
         let semaphore = DispatchSemaphore(value: 0)
         return MockURLSessionDataTask {
             self.queue.async {