Update tests

uber · Jun 30, 2024 · 41de219 · 41de219
1 parent f5713af
commit 41de219
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 10 deletions.
diff --git a/Tests/TestFuncs/TestFuncThrow/FixtureFuncThrow.swift b/Tests/TestFuncs/TestFuncThrow/FixtureFuncThrow.swift
@@ -104,16 +104,12 @@ class FuncThrowMock: FuncThrow {
 let funcTypedThrow = """
 import Foundation
 
-enum FuncError: Error {
-	case foo
-}
-
 /// \(String.mockAnnotation)
 protocol FuncTypedThrow {
 	func f1(arg: Int) throws(any LocalizedError) -> String
 	func f2(arg: Int) throws(any LocalizedError)
-	func f3(arg: Int) throws(FuncError) -> String
-	func f4(arg: Int) throws(FuncError)
+	func f3(arg: Int) throws(SomeError) -> String
+	func f4(arg: Int) throws(SomeError)
 }
 """
 let funcTypedThrowMock = """
@@ -146,8 +142,8 @@ class FuncTypedThrowMock: FuncTypedThrow {
 	}
 
 	private(set) var f3CallCount = 0
-	var f3Handler: ((Int) throws(FuncError) -> (String))?
-	func f3(arg: Int) throws(FuncError) -> String {
+	var f3Handler: ((Int) throws(SomeError) -> (String))?
+	func f3(arg: Int) throws(SomeError) -> String {
 		f3CallCount += 1
 		if let f3Handler = f3Handler {
 			return try f3Handler(arg)
@@ -156,8 +152,8 @@ class FuncTypedThrowMock: FuncTypedThrow {
 	}
 
 	private(set) var f4CallCount = 0
-	var f4Handler: ((Int) throws(FuncError) -> ())?
-	func f4(arg: Int) throws(FuncError)  {
+	var f4Handler: ((Int) throws(SomeError) -> ())?
+	func f4(arg: Int) throws(SomeError)  {
 		f4CallCount += 1
 		if let f4Handler = f4Handler {
 			try f4Handler(arg)

diff --git a/Tests/TestInit/FixtureInit.swift b/Tests/TestInit/FixtureInit.swift
@@ -326,3 +326,23 @@ class MyProtocolMock: MyProtocol {
 
 }
 """
+
+let throwableInit = """
+/// \(String.mockAnnotation)
+protocol MyProtocol {
+	init(param: String) throws(SomeError)
+}
+"""
+
+let throwableInitMock = """
+
+
+
+class MyProtocolMock: MyProtocol {
+	private var _param: String!
+	init() { }
+	required init(param: String = "") throws(SomeError) {
+		self._param = param
+	}
+}
+"""
diff --git a/Tests/TestInit/InitTests.swift b/Tests/TestInit/InitTests.swift
@@ -41,4 +41,11 @@ class InitTests: MockoloTestCase {
             dstContent: initWithSameParamNameButDifferentTypeMock
         )
     }
+
+	func testThrowableInit() {
+		verify(
+			srcContent: throwableInit,
+			dstContent: throwableInitMock
+		)
+	}
 }