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

import BitwardenKitMocks
import XCTest

@testable import BitwardenShared

// 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.
    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` dismisses a presented view controller before swapping between
    /// different view controllers.
    func test_childViewController_withPresentedViewController() {
        let viewController1 = UIViewController()
        subject.childViewController = viewController1
        viewController1.present(UIViewController(), animated: false)
        XCTAssertTrue(subject.children.contains(viewController1))
        XCTAssertNotNil(viewController1.presentedViewController)

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

        waitFor(subject.presentedViewController == nil)
        XCTAssertNil(subject.presentedViewController)
    }

    /// `childViewController` removes the current view controller when set to `nil`.
    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)
    }

    /// `preferredStatusBarStyle` returns the preferred status bar style for the given theme.
    func test_preferredStatusBarStyle() {
        subject.appTheme = .dark
        XCTAssertEqual(subject.preferredStatusBarStyle, .lightContent)

        subject.appTheme = .default
        XCTAssertEqual(subject.preferredStatusBarStyle, .default)

        subject.appTheme = .light
        XCTAssertEqual(subject.preferredStatusBarStyle, .darkContent)
    }

    /// `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)
    }
}