ios/AuthenticatorShared/UI/Platform/Application/Utilities/RootViewControllerTests.swift

import BitwardenKitMocks
import XCTest

@testable import AuthenticatorShared

// MARK: - RootViewControllerTests

class RootViewControllerTests: BitwardenTestCase {
    // MARK: Properties

    var subject: RootViewController!

    // MARK: Setup & Teardown

    override func setUp() {
        super.setUp()
        subject = RootViewController()
        setKeyWindowRoot(viewController: subject)
    }

    override func tearDown() {
        super.tearDown()
        subject = nil
    }

    // MARK: Tests

    /// `childViewController` swaps between different view controllers.
    @MainActor
    func test_childViewController_withNewViewController() {
        let viewController1 = UIViewController()
        subject.childViewController = viewController1
        XCTAssertTrue(subject.children.contains(viewController1))

        let viewController2 = UIViewController()
        subject.childViewController = viewController2
        XCTAssertTrue(subject.children.contains(viewController2))
        XCTAssertFalse(subject.children.contains(viewController1))
    }

    /// `childViewController` removes the current view controller when set to `nil`.
    @MainActor
    func test_childViewController_nil() {
        let viewController1 = UIViewController()
        subject.childViewController = viewController1
        XCTAssertTrue(subject.children.contains(viewController1))

        subject.childViewController = nil
        XCTAssertTrue(subject.children.isEmpty)
        XCTAssertTrue(subject.view.subviews.isEmpty)
    }

    /// `rootViewController` returns itself, instead of the current `childViewController`.
    @MainActor
    func test_rootViewController() {
        let viewController = UIViewController()
        subject.childViewController = viewController
        XCTAssertIdentical(subject.rootViewController, subject)
    }

    /// `show(child:)` sets `childViewController` to the `rootViewController` on the child navigator.
    @MainActor
    func test_show() {
        let navigator = MockStackNavigator()
        navigator.rootViewController = UIViewController()
        subject.show(child: navigator)
        XCTAssertIdentical(subject.childViewController, navigator.rootViewController)
    }
}