Fixes for subscripts support in AutoMockable template

krzysztofzablocki · Dec 22, 2023 · b76e38b · b76e38b
1 parent a09bdf4
commit b76e38b
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 4 deletions.
diff --git a/Templates/Templates/AutoMockable.stencil b/Templates/Templates/AutoMockable.stencil
@@ -120,8 +120,9 @@ import {{ import }}
 {% endif %}
 {% endmacro %}
 
-{% macro mockSubscript subscript %}
-    {% call accessLevel variable.readAccess %}{{ 'final ' if subscript.isFinal }}subscript{% if subscript.isGeneric %}<{% for genericParameter in subscript.genericParameters %}{{ genericParameter.name }}{% if genericParameter.typeName %}: {{ genericParameter.typeName.name }}{% endif %}{{ ', ' if not forloop.last }}{% endfor %}>{% endif %}({% for parameters in subscript.parameters }{{ parameter.asSource }}{{ ', ' if not forloop.last }}{% endfor %}) -> {{ subscript.returnTypeName.name }}{% if subscript.genericRequirements|count != 0 %} where {% for requirement in subscript.genericRequirements %}{{ requirement.leftType.name }} {{ requirement.relationshipSyntax }} {{ requirement.rightType.typeName.name }}{{ ', ' if not forloop.last }}{% endfor %}{% endif %} {
+{% macro mockSubscript subscript index %}
+    //MARK: - Subscript #{{ index }}
+    {% call accessLevel variable.readAccess %}subscript{% if subscript.isGeneric %}<{% for genericParameter in subscript.genericParameters %}{{ genericParameter.name }}{% if genericParameter.inheritedTypeName %}: {{ genericParameter.inheritedTypeName.name }}{% endif %}{{ ', ' if not forloop.last }}{% endfor %}>{% endif %}({% for parameter in subscript.parameters %}{{ parameter.asSource }}{{ ', ' if not forloop.last }}{% endfor %}) -> {{ subscript.returnTypeName.name }}{% if subscript.genericRequirements|count != 0 %} where {% for requirement in subscript.genericRequirements %}{{ requirement.leftType.name }} {{ requirement.relationshipSyntax }} {{ requirement.rightType.typeName.name }}{{ ', ' if not forloop.last }}{% endfor %}{% endif %} {
     {% if subscript.readAccess %} get { fatalError("Subscripts are not fully supported yet") }{% endif %}
     {% if subscript.writeAccess %} set { fatalError("Subscripts are not fully supported yet") }{% endif %}
     }
@@ -296,7 +297,7 @@ import {{ import }}
 {% endfor %}
 
 {% for subscript in type.allSubscripts|!definedInExtension %}
-    {% call mockSubscript subscript %}
+    {% call mockSubscript subscript forloop.counter %}
 {% endfor %}
 }
 {% endif %}{% endfor %}
diff --git a/Templates/Tests/Context/AutoMockable.swift b/Templates/Tests/Context/AutoMockable.swift
@@ -227,7 +227,8 @@ public protocol ProtocolWithOverrides {
     func doSomething(_ data: String) throws -> (([Int], [Any]) -> Void)
 }
 
-protocol SubscriptProtocol: AutoMockable {
+// sourcery: AutoMockable
+protocol SubscriptProtocol {
     subscript(arg: Int) -> String { get set }
     subscript<T>(arg: T) -> Int { get }
     subscript<T: Hashable>(arg: T) -> T? { get set }

diff --git a/Templates/Tests/Expected/AutoMockable.expected b/Templates/Tests/Expected/AutoMockable.expected
@@ -1391,6 +1391,26 @@ class StaticMethodProtocolMock: StaticMethodProtocol {
     }
 
 }
+class SubscriptProtocolMock: SubscriptProtocol {
+    //MARK: - Subscript #1
+     subscript(arg: Int) -> String {
+     get { fatalError("Subscripts are not fully supported yet") }
+     set { fatalError("Subscripts are not fully supported yet") }
+    }
+    //MARK: - Subscript #2
+     subscript<T>(arg: T) -> Int {
+     get { fatalError("Subscripts are not fully supported yet") }
+    }
+    //MARK: - Subscript #3
+     subscript<T: Hashable>(arg: T) -> T? {
+     get { fatalError("Subscripts are not fully supported yet") }
+     set { fatalError("Subscripts are not fully supported yet") }
+    }
+    //MARK: - Subscript #4
+     subscript<T>(arg: String) -> T? where T : Cancellable {
+     get { fatalError("Subscripts are not fully supported yet") }
+    }
+}
 class ThrowableProtocolMock: ThrowableProtocol {