Merge pull request #62 from raptorxcz/feature/stub-naming

Add option for custom stub naming
raptorxcz · Mar 26, 2024 · c07023b · c07023b
2 parents 41e0af3 + da7677d
commit c07023b
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/Sources/Rubicon/Generator/StubGenerator.swift b/Sources/Rubicon/Generator/StubGenerator.swift
@@ -19,11 +19,11 @@ final class StubGenerator {
         self.initGenerator = initGenerator
     }
 
-    func generate(from protocolType: ProtocolDeclaration) -> String {
+    func generate(from protocolType: ProtocolDeclaration, stubSuffix: String = "Stub") -> String {
         let content = generateBody(from: protocolType)
         return protocolGenerator.makeProtocol(
             from: protocolType,
-            stub: "Stub",
+            stub: stubSuffix,
             content: content
         ).joined(separator: "\n") + "\n"
     }

diff --git a/Sources/Rubicon/Integration/Rubicon.swift b/Sources/Rubicon/Integration/Rubicon.swift
@@ -40,7 +40,7 @@ public final class Rubicon {
         let generator = makeStubGenerator(accessLevel: accessLevel, indentStep: indentStep)
         do {
             let protocols = try parser.parse(text: code)
-            return protocols.map(generator.generate(from:))
+            return protocols.map { generator.generate(from: $0) }
         } catch {
             return []
         }

diff --git a/Tests/RubiconTests/Generator/StubGeneratorTests.swift b/Tests/RubiconTests/Generator/StubGeneratorTests.swift
@@ -38,6 +38,12 @@ final class StubGeneratorTests: XCTestCase {
         XCTAssertEqual(result, "result\n")
     }
 
+    func test_givenEmptyProtocolWithCustomSuffix_whenGenerate_thenGenerateCode() {
+        let result = sut.generate(from: .makeStub(), stubSuffix: "Mock")
+
+        XCTAssertEqual(protocolGeneratorSpy.makeProtocol.first?.stub, "Mock")
+    }
+
     func test_givenProtocolWithVariable_whenGenerate_thenGenerateStub() {
         let protocolDeclaration = ProtocolDeclaration.makeStub(variables: [.makeStub()])