import XCTest
@testable import RouterOSAssistant

final class SSHHostKeyTrustStoreTests: XCTestCase {
    override func tearDown() {
        UserDefaults.standard.removeObject(forKey: "RouterOSAssistant.TrustedSSHHostKeyFingerprints")
        super.tearDown()
    }

    func testUntrustedByDefault() {
        let store = SSHHostKeyTrustStore()
        XCTAssertFalse(store.isTrusted(host: "192.0.2.1", fingerprint: "AA:BB"))
    }

    func testTrustPersistsForSameHostAndFingerprint() {
        let store = SSHHostKeyTrustStore()
        store.trust(host: "192.0.2.1", fingerprint: "AA:BB")

        XCTAssertTrue(store.isTrusted(host: "192.0.2.1", fingerprint: "AA:BB"))
        XCTAssertFalse(store.isTrusted(host: "192.0.2.1", fingerprint: "CC:DD"))
        XCTAssertFalse(store.isTrusted(host: "192.0.2.2", fingerprint: "AA:BB"))
    }
}