Improve partial class mock tests

typealiased · Aug 5, 2021 · 4e0ead8 · 4e0ead8
1 parent 606799a
commit 4e0ead8
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/Sources/MockingbirdTestsHost/MinimalTestTypes.swift b/Sources/MockingbirdTestsHost/MinimalTestTypes.swift
@@ -14,5 +14,5 @@ protocol MinimalProtocol {
 
 class MinimalClass {
   var property: String = "super"
-  func method(value: String) -> String { return "super-" + value }
+  func method(value: String) -> String { return "\(value)-\(property)" }
 }
diff --git a/Tests/MockingbirdTests/Framework/PartialMockTests.swift b/Tests/MockingbirdTests/Framework/PartialMockTests.swift
@@ -40,6 +40,13 @@ class PartialMockTests: BaseTestCase {
     }
   }
 
+  class SelfReferencingImplementer: MinimalProtocol {
+    var property: String = "foobar"
+    func method(value: String) -> String {
+      return property
+    }
+  }
+
   class UnrelatedType {}
 
   override func setUpWithError() throws {
@@ -108,13 +115,17 @@ class PartialMockTests: BaseTestCase {
   }
 
   func testForwardMethodToSuperclass() throws {
+    given(classMock.property).willReturn("world")
     given(classMock.method(value: any())).willForwardToSuper()
-    XCTAssertEqual(classMock.method(value: "hello"), "super-hello")
+    XCTAssertEqual(classMock.method(value: "hello"), "hello-world")
+    verify(classMock.property).wasCalled()
     verify(classMock.method(value: "hello")).wasCalled()
   }
   func testForwardMethodToSuperclass_stubbingOperator() throws {
+    given(classMock.property) ~> "world"
     given(classMock.method(value: any())) ~> forwardToSuper()
-    XCTAssertEqual(classMock.method(value: "hello"), "super-hello")
+    XCTAssertEqual(classMock.method(value: "hello"), "hello-world")
+    verify(classMock.property).wasCalled()
     verify(classMock.method(value: "hello")).wasCalled()
   }