21 lines
572 B
Swift
21 lines
572 B
Swift
//
|
|
// StubFunctionThenTrait.swift
|
|
// Cuckoo
|
|
//
|
|
// Created by Filip Dolnik on 27.06.16.
|
|
// Copyright © 2016 Brightify. All rights reserved.
|
|
//
|
|
|
|
public protocol StubFunctionThenTrait: BaseStubFunctionTrait {
|
|
/// Invokes `implementation` when invoked.
|
|
func then(_ implementation: @escaping (InputType) -> OutputType) -> Self
|
|
}
|
|
|
|
public extension StubFunctionThenTrait {
|
|
@discardableResult
|
|
func then(_ implementation: @escaping (InputType) -> OutputType) -> Self {
|
|
stub.appendAction(.callImplementation(implementation))
|
|
return self
|
|
}
|
|
}
|