PrivacyProtectionStatusTests.swift

//
//  PrivacyProtectionStatusTests.swift
//
//  Copyright © 2022 DuckDuckGo. All rights reserved.
//
//  Licensed under the Apache License, Version 2.0 (the "License");
//  you may not use this file except in compliance with the License.
//  You may obtain a copy of the License at
//
//  http://www.apache.org/licenses/LICENSE-2.0
//
//  Unless required by applicable law or agreed to in writing, software
//  distributed under the License is distributed on an "AS IS" BASIS,
//  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//  See the License for the specific language governing permissions and
//  limitations under the License.
//

import XCTest
import Combine
@testable import DuckDuckGo_Privacy_Browser

class PrivacyProtectionStatusTests: XCTestCase {

    var cancellables: Set<AnyCancellable> = []

    override func tearDown() {
        super.tearDown()
        cancellables.removeAll()
    }

    func testWhenStatusUpdatedFromPublisherThenReflectsNewStatus() {
        // Example of how you might test updates from a publisher
        let subject = PassthroughSubject<Bool, Never>()
        let status = PrivacyProtectionStatus(statusPublisher: subject) { isOn in
            return isOn ? .on : .off
        }

        // Initial value should be nil
        XCTAssertNil(status.status)

        let expectation = XCTestExpectation(description: "Status updates to .on")
        status.$status.sink { newStatus in
            if newStatus == .on {
                expectation.fulfill()
            }
        }.store(in: &cancellables)

        subject.send(true)

        wait(for: [expectation], timeout: 1.0)
    }

    func testWhenInitializedWithStaticStatusIndicatorThenSetsStatusCorrectly() {
        // Test initialization with a static status indicator
        let status = PrivacyProtectionStatus(statusIndicator: .on)

        XCTAssertEqual(status.status, .on)
    }

    func testWhenInitializedWithoutStatusThenStatusIsNil() {
        // Test initialization without a status
        let status = PrivacyProtectionStatus()

        XCTAssertNil(status.status)
    }
}