Automatically create simple mocks for protocols in Swift

automock.stencil

{% for type in types.protocols.implementing.AutoMockable %}
{% if not type.name == "AutoMockable" %}
class {{ type.name }}Mock: {{ type.name }} {
    {% for method in type.allMethods %}
    var {{ method.shortName }}Called = false
    func {{ method.shortName }}({% for param in method.parameters %}{{ param.argumentLabel }}{% if not param.argumentLabel == param.name %} {{ param.name }}{% endif %}: {{param.typeName }}{% if not forloop.last%}, {% endif %}{% endfor %}) {
        {{ method.shortName }}Called = true
    }
    {% endfor %}
}
{% endif %}
{% endfor %}

DataService.swift

import Foundation

/// Protocols confoming to this protocol will have a mock generated for them.
protocol AutoMockable {}

protocol DataServiceProtocol: AutoMockable {
    
    func fetchResource(fromURL url: URL, onComplete: (Result)-> Void)
    
}

class DataService: DataServiceProtocol {
    
    func fetchResource(fromURL url: URL, onComplete: (Result)-> Void) {
        
        //Real implementation
        //Will get replaced in the generated file
        
    }
    
}

DataServiceMock.swift

// Generated using Sourcery 0.5.0 — https://github.com/krzysztofzablocki/Sourcery
// DO NOT EDIT






class DataServiceProtocolMock: DataServiceProtocol {
    
    var fetchResourceCalled = false
    func fetchResource(fromURL url: URL, onComplete: (Result)-> Void) {
        fetchResourceCalled = true
    }
    
}

description.md

Automatically generates a mock class implementing a specific protocol. For each protocol method, the mock will provide a methodNameCalled boolean. Files:

• DataService.swift - sample file
• lens.stencil - sourcery template
• DataServiceMock.swift - sample mock generated

You will need sourcery template for this to work.

Suggestions and improvements are welcome!

Inspired by Filip Zawada.