Skip to content

Commit 9c63706

Browse files
add wasCalledTimesVariable in func mocker
1 parent 7fd31ae commit 9c63706

File tree

2 files changed

+9
-0
lines changed

2 files changed

+9
-0
lines changed

CodeGeneratorTests/FuncMockTests.swift

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,9 @@ class FuncMockerTests: XCTestCase {
3131
XCTAssertEqual(funcMocker.sensibleVariables[3], VarSignature(string: " var testSuccessHandlerParam1: Int!"))
3232
XCTAssertEqual(funcMocker.sensibleVariables[4], VarSignature(string: " var testSuccessHandlerDidReturn: String?"))
3333
XCTAssertEqual(funcMocker.wasCalledVariable, VarSignature(string: " var testFuncMockWasCalled: Bool?"))
34+
var wasCalledTimesVar = VarSignature(declaration: "var", name: "testFuncMockWasCalledTimes", type: "Int", accessLevel: .internal)
35+
wasCalledTimesVar.initValue = "0"
36+
XCTAssertEqual(funcMocker.wasCalledTimesVariable, wasCalledTimesVar)
3437
XCTAssertEqual(funcMocker.returnVariable, VarSignature(string: " var testFuncMockShouldReturn: String!"))
3538

3639
var expectedBodyLines = string(from: "funcMock", ofType: "txt").components(separatedBy: "\n")

Generate.../FuncMocker.swift

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,12 @@ struct FuncMocker {
2828
return VarSignature(declaration: "var", name: funcSignature.readableName + "WasCalled", type: "Bool?")
2929
}
3030

31+
var wasCalledTimesVariable: VarSignature {
32+
var varSignature = VarSignature(declaration: "var", name: funcSignature.readableName + "WasCalledTimes", type: "Int")
33+
varSignature.initValue = "0"
34+
return varSignature
35+
}
36+
3137
var returnVariable: VarSignature? {
3238
if funcSignature.isReturnVoid {
3339
return nil

0 commit comments

Comments
 (0)