chore: add unchecked Sendable conformance to AutoMockable

krzysztofzablocki · Aug 1, 2024 · 2a26b88 · 2a26b88
1 parent 81033ec
commit 2a26b88
Show file tree

Hide file tree

Showing 4 changed files with 65 additions and 1 deletion.
diff --git a/Templates/Templates/AutoMockable.stencil b/Templates/Templates/AutoMockable.stencil
@@ -369,8 +369,12 @@ import {{ import }}
     {% endif %}
 {%- endmacro %}
 
+{% macro extractRequiredProtocolConformance type -%}
+    {% if type.based.Sendable %}, @unchecked Sendable{% endif %}
+{%- endmacro %}
+
 {% for type in types.protocols where type.based.AutoMockable or type|annotated:"AutoMockable" %}{% if type.name != "AutoMockable" %}
-{% call accessLevel type.accessLevel %}class {{ type.name }}Mock{% set generics %}{% call extractProtocolCompositionFromAssociatedTypes type %}{% endset %}{{ generics | replace:",>",">"}}: {{ type.name }} {%- set requirements -%}{% call extractProtocolRequirementsFromType type %}{%- endset -%} {{ requirements|replace:",{","{"|replace:"{"," {" }}
+{% call accessLevel type.accessLevel %}class {{ type.name }}Mock{% set generics %}{% call extractProtocolCompositionFromAssociatedTypes type %}{% endset %}{{ generics | replace:",>",">"}}: {{ type.name }}{% call extractRequiredProtocolConformance type %} {%- set requirements -%}{% call extractProtocolRequirementsFromType type %}{%- endset -%} {{ requirements|replace:",{","{"|replace:"{"," {" }}
 {% for associatedType in type.associatedTypes|sortedValuesByKeys %}
     {% if associatedType.type.kind == nil or not associatedType.type.kind|contains:"protocol" %}
     typealias {{ associatedType.name }} = {% if associatedType.type != nil %}{{ associatedType.type.name }}{% elif associatedType.typeName != nil %}{{ associatedType.typeName.name }}{% else %}Any{% endif %}

diff --git a/Templates/Tests/Context/AutoMockable.swift b/Templates/Tests/Context/AutoMockable.swift
@@ -292,3 +292,15 @@ protocol TestProtocol {
 
     func getValue6() -> Value6
 }
+
+// sourcery: AutoMockable
+protocol SendableProtocol: Sendable {
+  var value: Any { get }
+}
+
+protocol NotMockedSendableProtocol: Sendable {}
+
+// sourcery: AutoMockable
+protocol SendableSendableProtocol: NotMockedSendableProtocol {
+  var value: Any { get }
+}
diff --git a/Templates/Tests/Expected/AutoMockable.expected b/Templates/Tests/Expected/AutoMockable.expected
@@ -1456,6 +1456,30 @@ class SameShortMethodNamesProtocolMock: SameShortMethodNamesProtocol {
     }
 
 
+}
+class SendableProtocolMock: SendableProtocol, @unchecked Sendable {
+
+
+    var value: Any {
+        get { return underlyingValue }
+        set(value) { underlyingValue = value }
+    }
+    var underlyingValue: (Any)!
+
+
+
+}
+class SendableSendableProtocolMock: SendableSendableProtocol, @unchecked Sendable {
+
+
+    var value: Any {
+        get { return underlyingValue }
+        set(value) { underlyingValue = value }
+    }
+    var underlyingValue: (Any)!
+
+
+
 }
 class SingleOptionalParameterFunctionMock: SingleOptionalParameterFunction {
 

diff --git a/Templates/Tests/Generated/AutoMockable.generated.swift b/Templates/Tests/Generated/AutoMockable.generated.swift
@@ -1456,6 +1456,30 @@ class SameShortMethodNamesProtocolMock: SameShortMethodNamesProtocol {
     }
 
 
+}
+class SendableProtocolMock: SendableProtocol, @unchecked Sendable {
+
+
+    var value: Any {
+        get { return underlyingValue }
+        set(value) { underlyingValue = value }
+    }
+    var underlyingValue: (Any)!
+
+
+
+}
+class SendableSendableProtocolMock: SendableSendableProtocol, @unchecked Sendable {
+
+
+    var value: Any {
+        get { return underlyingValue }
+        set(value) { underlyingValue = value }
+    }
+    var underlyingValue: (Any)!
+
+
+
 }
 class SingleOptionalParameterFunctionMock: SingleOptionalParameterFunction {