Skip to content

Commit 0f27e9a

Browse files
art-divinRuslan Alikhamov
authored and
Ruslan Alikhamov
committed
Added missing definition
1 parent a46bf91 commit 0f27e9a

File tree

1 file changed

+27
-0
lines changed

1 file changed

+27
-0
lines changed

Templates/MethodWrapper.swift

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,33 @@ func replacingSelf(_ value: String, current: Current) -> String {
4040
.replacingOccurrences(of: ",Self?", with: ",\(current.selfType)?")
4141
}
4242

43+
class MethodRegistrar {
44+
var registered: [String: Int] = [:]
45+
var suffixes: [String: Int] = [:]
46+
var suffixesWithoutReturnType: [String: Int] = [:]
47+
48+
func register(_ name: String, _ uniqueName: String, _ uniqueNameWithReturnType: String) {
49+
if let count = registered[name] {
50+
registered[name] = count + 1
51+
suffixes[uniqueNameWithReturnType] = count + 1
52+
} else {
53+
registered[name] = 1
54+
suffixes[uniqueNameWithReturnType] = 1
55+
}
56+
57+
if let count = suffixesWithoutReturnType[uniqueName] {
58+
suffixesWithoutReturnType[uniqueName] = count + 1
59+
} else {
60+
suffixesWithoutReturnType[uniqueName] = 1
61+
}
62+
}
63+
64+
func returnTypeMatters(uniqueName: String) -> Bool {
65+
let count = suffixesWithoutReturnType[uniqueName] ?? 0
66+
return count > 1
67+
}
68+
}
69+
4370
class MethodWrapper {
4471
private var noStubDefinedMessage: String {
4572
let methodName = method.name.condenseWhitespace()

0 commit comments

Comments
 (0)