chore: update tests

danthorpe · Dec 8, 2024 · 9939094 · 9939094
1 parent d83ebee
commit 9939094
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/Tests/OAuthTests/OAuthDelegateTests.swift b/Tests/OAuthTests/OAuthDelegateTests.swift
@@ -43,6 +43,11 @@ final class OAuthDelegateTests: OAuthTestCase {
         .server(authenticationMethod: .stub)
         .authenticated(oauth: stub)
 
+      // Configure Network
+      try await network.stubOAuthSystem {
+        await $0.set(presentationContext: DefaultPresentationContext())
+      }
+
       // Make a request to protected resource
       var request = HTTPRequestData(path: "/protected-resource")
       request.authenticationMethod = .stub

diff --git a/Tests/OAuthTests/StubOAuthSystem.swift b/Tests/OAuthTests/StubOAuthSystem.swift
@@ -21,7 +21,7 @@ struct StubOAuthSystem: StandardOAuthSystem {
 
 extension NetworkingComponent {
   func stubOAuthSystem<ReturnValue>(
-    perform: (any OAuthProxy<StubOAuthSystem.Credentials>) async throws -> ReturnValue
+    perform: @MainActor (any OAuthProxy<StubOAuthSystem.Credentials>) async throws -> ReturnValue
   ) async throws -> ReturnValue {
     try await oauth(of: StubOAuthSystem.Credentials.self, perform: perform)
   }