Привет Прадип , добро пожаловать в StackOverflow. ?
Вы можете определить протокол с подмножеством интересующих вас функций из AWSAPIGatewayResponse
, заставить AWSAPIGatewayResponse
соответствовать ему и заставить apiResponseResults(response:, session:)
ожидать значение этого типа.
protocol Response {
var statusCode: Int { get }
}
extension AWSAPIGatewayResponse: Response { }
func apiResponseResults(response: Response, sessionObject: Session) { ... }
В ваших тестах вы можете определить двойной тест, соответствующий Response
, и использовать его в качестве входного параметра для вашего метода.
struct TestResponse: Response {
let statusCode: Int
}
// Use it like this
let response = TestResponse(statusCode: 200)
Надеюсь, это поможет.