[krzysztofzablocki#1251] Swifty generated variable names + fixed gene…

…rated mocks compilation issues due to method generic parameters
rokridi · Jan 9, 2024 · 97507e9 · 97507e9
1 parent b62a8fc
commit 97507e9
Show file tree

Hide file tree

Showing 3 changed files with 40 additions and 2 deletions.
diff --git a/Templates/Templates/AutoMockable.stencil b/Templates/Templates/AutoMockable.stencil
@@ -16,8 +16,12 @@ import {{ import }}
 @testable import {{ import }}
 {% endfor %}
 
-{% macro cleanString string %}{{ string | replace:"(","_" | replace:")","" | replace:":","_" | replace:"`","" | replace:" ","_" | replace:"?","_" | replace:"!","_" | replace:",","_" | replace:"->","_" | replace:"@","_" | replace:".","_" | replace:"[","" | replace:"]","" | snakeToCamelCase }}{% endmacro %}
-{% macro swiftifyMethodName method %}{% call cleanString method.name | lowerFirstWord %}{% call cleanString method.actualReturnTypeName.name | upperFirstLetter %}{% endmacro %}
+{% macro cleanString string %}{{ string | replace:"(","_" | replace:")","" | replace:":","_" | replace:"`","" | replace:" ","_" | replace:"?","_" | replace:"!","_" | replace:",","_" | replace:"->","_" | replace:"@","_" | replace:".","_" | replace:"[","" | replace:"]","" | replace:"<","" | replace:">","" | snakeToCamelCase }}{% endmacro %}
+{%- macro swiftifyMethodName method -%}
+    {%- set cleanMethodName %}{% call cleanString method.name %}{% endset %}
+    {%- set cleanMethodReturnTypeName %}{% call cleanString method.actualReturnTypeName.name %}{% endset -%}
+    {{ cleanMethodName | lowerFirstLetter }}{{ cleanMethodReturnTypeName | upperFirstLetter }}
+{%- endmacro -%}
 
 {% macro accessLevel level %}{% if level != 'internal' %}{{ level }} {% endif %}{% endmacro %}
 

diff --git a/Templates/Tests/Context/AutoMockable.swift b/Templates/Tests/Context/AutoMockable.swift
@@ -235,3 +235,8 @@ protocol SubscriptProtocol {
     subscript<T: Hashable>(arg: T) -> T? { get set }
     subscript<T>(arg: String) -> T? where T: Cancellable { get throws }
 }
+
+// sourcery: AutoMockable
+public protocol ProtocolWithMethodWithGenericParameters {
+    func execute(param: Result<Int, Error>) -> Result<String, Error>
+}
diff --git a/Templates/Tests/Expected/AutoMockable.expected b/Templates/Tests/Expected/AutoMockable.expected
@@ -1526,4 +1526,33 @@ class VariablesProtocolMock: VariablesProtocol {
     var universityMarks: [String: Int] = [:]
 
 
+}
+class ProtocolWithMethodWithGenericParameters: SomeProtocol {
+
+
+
+
+    //MARK: - execute
+
+    var executeParamResultIntErrorResultStringErrorCallsCount = 0
+    var executeParamResultIntErrorResultStringErrorCalled: Bool {
+        return executeParamResultIntErrorResultStringErrorCallsCount > 0
+    }
+    var executeParamResultIntErrorResultStringErrorReceivedParam: (Result<Int, Error>)?
+    var executeParamResultIntErrorResultStringErrorReceivedInvocations: [(Result<Int, Error>)] = []
+    var executeParamResultIntErrorResultStringErrorReturnValue: Result<String, Error>!
+    var executeParamResultIntErrorResultStringErrorClosure: ((Result<Int, Error>) -> Result<String, Error>)?
+
+    func execute(param: Result<Int, Error>) -> Result<String, Error> {
+        executeParamResultIntErrorResultStringErrorCallsCount += 1
+        executeParamResultIntErrorResultStringErrorReceivedParam = param
+        executeParamResultIntErrorResultStringErrorReceivedInvocations.append(param)
+        if let executeParamResultIntErrorResultStringErrorClosure = executeParamResultIntErrorResultStringErrorClosure {
+            return executeParamResultIntErrorResultStringErrorClosure(param)
+        } else {
+            return executeParamResultIntErrorResultStringErrorReturnValue
+        }
+    }
+
+
 }