ios - Swift 单元测试模拟类静态方法

Question

我在类中有一个静态方法

class A {
    static func myStaticMethod() -> B {
        return B()
    }
}

class B {
    func getTitle() -> String {
        // some functionality here
    }
}

在我想测试的类方法中，我使用它:

func someBoolFunc() -> Bool {
    var b = A.myStaticMethod()
    if (b.getTitle() = “hello”) {
         return true
    }
    return false
}

如何为此编写模拟类...我尝试过:

class MockA: A {
    var myTitle:String
    // this seems incorrect, because i didn't override static function
    func myStaticMethod() -> MockB {
        var b = MockB()
        b.title = myTitle
        return b
    }
}

class MockB: B {
    var myTitle:String    

    func getTitle() -> String {

        return myTitle
    }
}

在测试中我想使用类似的东西:

func testExample() {
    A = MockA
    MockA.title = "My title"
    // And now this func should use my MockA instead of A
    someBoolFunc()
}

当然这只是理论上的:(

Answer 1

也许是这样？

protocol AProto {
  static func myStaticMethod() -> BProto
}

class A: AProto {
  static func myStaticMethod() -> BProto {
    return B()
  }
}

class MockA: AProto {
  static func myStaticMethod() -> BProto {
    return MockB()
  }
}

protocol BProto {
  func getTitle() -> String
}

class B: BProto {
  func getTitle() -> String {
    return "hello"
  }
}

class MockB: BProto {
  func getTitle() -> String {
    return "bye bye"
  }
}

func someBoolFunc(_ aProto: AProto.Type = A.self) -> Bool {
  var b = aProto.myStaticMethod()
  if (b.getTitle() == "hello") {
    return true
  }
  return false
}

print(someBoolFunc())
print(someBoolFunc(MockA.self))