Skip to content

Commit

Permalink
Last non async call
Browse files Browse the repository at this point in the history
  • Loading branch information
adam-fowler committed Jun 16, 2023
1 parent 7663bdd commit 7153ea1
Showing 1 changed file with 8 additions and 8 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -31,24 +31,24 @@ class ConfigFileCredentialProviderTests: XCTestCase {
return (.init(httpClient: httpClient, eventLoop: eventLoop, logger: TestEnvironment.logger, options: .init()), eventLoopGroup, httpClient)
}

func testCredentialProviderStatic() {
func testCredentialProviderStatic() async throws {
let credentials = ConfigFileLoader.SharedCredentials.staticCredential(credential: StaticCredential(accessKeyId: "foo", secretAccessKey: "bar"))
let (context, eventLoopGroup, httpClient) = self.makeContext()

let provider = try? ConfigFileCredentialProvider.credentialProvider(
let provider = try ConfigFileCredentialProvider.credentialProvider(
from: credentials,
context: context,
endpoint: nil
)
XCTAssertEqual((provider as? StaticCredential)?.accessKeyId, "foo")
XCTAssertEqual((provider as? StaticCredential)?.secretAccessKey, "bar")

XCTAssertNoThrow(try provider?.shutdown(on: context.eventLoop).wait())
XCTAssertNoThrow(try httpClient.syncShutdown())
try await provider.shutdown(on: context.eventLoop).get()
try await httpClient.shutdown()
XCTAssertNoThrow(try eventLoopGroup.syncShutdownGracefully())
}

func testCredentialProviderSTSAssumeRole() {
func testCredentialProviderSTSAssumeRole() async throws {
let credentials = ConfigFileLoader.SharedCredentials.assumeRole(
roleArn: "arn",
sessionName: "baz",
Expand All @@ -57,16 +57,16 @@ class ConfigFileCredentialProviderTests: XCTestCase {
)
let (context, eventLoopGroup, httpClient) = self.makeContext()

let provider = try? ConfigFileCredentialProvider.credentialProvider(
let provider = try ConfigFileCredentialProvider.credentialProvider(
from: credentials,
context: context,
endpoint: nil
)
XCTAssertTrue(provider is STSAssumeRoleCredentialProvider)
XCTAssertEqual((provider as? STSAssumeRoleCredentialProvider)?.request.roleArn, "arn")

XCTAssertNoThrow(try provider?.shutdown(on: context.eventLoop).wait())
XCTAssertNoThrow(try httpClient.syncShutdown())
try await provider.shutdown(on: context.eventLoop).get()
try await httpClient.shutdown()
XCTAssertNoThrow(try eventLoopGroup.syncShutdownGracefully())
}

Expand Down

0 comments on commit 7153ea1

Please sign in to comment.