Skip to content

Commit 67c9d72

Browse files
committed
Fixed syntax. Now using config
1 parent 57b4274 commit 67c9d72

File tree

2 files changed

+17
-14
lines changed

2 files changed

+17
-14
lines changed

Sources/AWSLambdaTesting/Lambda+Testing.swift

Lines changed: 13 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -23,12 +23,12 @@ extension Lambda {
2323
public var requestId: String
2424
public var traceId: String
2525
public var invokedFunctionArn: String
26-
public var timeout: Double
26+
public var timeout: DispatchTimeInterval
2727

2828
public init(requestId: String = "\(DispatchTime.now().uptimeNanoseconds)",
2929
traceId: String = "Root=\(DispatchTime.now().uptimeNanoseconds);Parent=\(DispatchTime.now().uptimeNanoseconds);Sampled=1",
3030
invokedFunctionArn: String = "arn:aws:lambda:us-west-1:\(DispatchTime.now().uptimeNanoseconds):function:custom-runtime",
31-
timeout: Double = 5) {
31+
timeout: DispatchTimeInterval = .seconds(5)) {
3232
self.requestId = requestId
3333
self.traceId = traceId
3434
self.invokedFunctionArn = invokedFunctionArn
@@ -37,28 +37,30 @@ extension Lambda {
3737
}
3838

3939
public static func test(_ closure: @escaping StringLambdaClosure,
40-
config: TestConfig = .init(),
41-
with payload: String) throws -> String {
40+
with payload: String,
41+
using config: TestConfig = .init()) throws -> String {
4242
try Self.test(StringLambdaClosureWrapper(closure), config: config, with: payload)
4343
}
4444

4545
public static func test(_ closure: @escaping StringVoidLambdaClosure,
46-
config: TestConfig = .init(),
47-
with payload: String) throws {
46+
with payload: String,
47+
using config: TestConfig = .init()) throws {
4848
_ = try Self.test(StringVoidLambdaClosureWrapper(closure), config: config, with: payload)
4949
}
5050

5151
public static func test<In: Decodable, Out: Encodable>(
5252
_ closure: @escaping CodableLambdaClosure<In, Out>,
53-
config: TestConfig = .init(),
54-
with payload: In
53+
with payload: In,
54+
using config: TestConfig = .init()
5555
) throws -> Out {
5656
try Self.test(CodableLambdaClosureWrapper(closure), config: config, with: payload)
5757
}
5858

59-
public static func test<In: Decodable>(_ closure: @escaping CodableVoidLambdaClosure<In>,
60-
config: TestConfig = .init(),
61-
with payload: In) throws {
59+
public static func test<In: Decodable>(
60+
_ closure: @escaping CodableVoidLambdaClosure<In>,
61+
with payload: In,
62+
using config: TestConfig = .init()
63+
) throws {
6264
_ = try Self.test(CodableVoidLambdaClosureWrapper(closure), config: config, with: payload)
6365
}
6466

Tests/AWSLambdaTestingTests/Tests.swift

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -123,11 +123,12 @@ class LambdaTestingTests: XCTestCase {
123123
}
124124

125125
func testConfigValues() {
126+
let timeout: TimeInterval = 4
126127
let config = Lambda.TestConfig(
127128
requestId: "abc123",
128129
traceId: "hahahihi",
129130
invokedFunctionArn: "arn:hihi",
130-
timeout: 4
131+
timeout: .seconds(4)
131132
)
132133

133134
let myLambda = { (ctx: Lambda.Context, _: String, callback: @escaping (Result<Void, Error>) -> Void) in
@@ -136,11 +137,11 @@ class LambdaTestingTests: XCTestCase {
136137
XCTAssertEqual(ctx.invokedFunctionArn, config.invokedFunctionArn)
137138

138139
let secondsSinceEpoch = Double(Int64(bitPattern: ctx.deadline.rawValue)) / -1_000_000_000
139-
XCTAssertLessThanOrEqual(Date(timeIntervalSince1970: secondsSinceEpoch).timeIntervalSinceNow, config.timeout)
140+
XCTAssertEqual(Date(timeIntervalSince1970: secondsSinceEpoch).timeIntervalSinceNow, timeout, accuracy: 0.1)
140141

141142
callback(.success(()))
142143
}
143144

144-
XCTAssertNoThrow(try Lambda.test(myLambda, config: config, with: UUID().uuidString))
145+
XCTAssertNoThrow(try Lambda.test(myLambda, with: UUID().uuidString, using: config))
145146
}
146147
}

0 commit comments

Comments
 (0)