Universe.swift

class ATests: XCTestCase {
    /// The testable `A` class.
    private var a: A!
    /// The mocked `B` class injected to `a`.
    private var b: BMock!
    
    override func setUp() {
        super.setUp()
        b = BMock()
        a = A(b: b)
    }
    
    // MARK: - Initialiser
    
    func testInitialiser() {
        XCTAssertFalse(a.hasAnswer)
    }
    
    // MARK: - checkAnswer()
    
    // When `b` returns true, `hasAnswer` should be true.
    func testCheckAnswerTrue() {
        b.validateAnswerResponse = true
        a.checkAnswer()
        XCTAssertTrue(a.hasAnswer)
    }
    
    // When `b` returns false, `hasAnswer` should be false.
    func testCheckAnswerFalse() {
        b.validateAnswerResponse = false
        a.checkAnswer()
        XCTAssertFalse(a.hasAnswer)
    }
}