bitchat/bitchatTests/Mocks/MockKeychain.swift

//
// MockKeychain.swift
// bitchat
//
// This is free and unencumbered software released into the public domain.
// For more information, see <https://unlicense.org>
//

import Foundation
@testable import bitchat

final class MockKeychain: KeychainManagerProtocol {
    private var storage: [String: Data] = [:]
    
    func saveIdentityKey(_ keyData: Data, forKey key: String) -> Bool {
        storage[key] = keyData
        return true
    }
    
    func getIdentityKey(forKey key: String) -> Data? {
        storage[key]
    }
    
    func deleteIdentityKey(forKey key: String) -> Bool {
        storage.removeValue(forKey: key)
        return true
    }
    
    func deleteAllKeychainData() -> Bool {
        storage.removeAll()
        return true
    }
    
    func secureClear(_ data: inout Data) {
        //
        data = Data()
    }
    
    func secureClear(_ string: inout String) {
        string = ""
    }
    
    func verifyIdentityKeyExists() -> Bool {
        storage["identity_noiseStaticKey"] != nil
    }
}

final class MockKeychainHelper: KeychainHelperProtocol {
    private typealias Service = String
    private typealias Key = String
    private var storage: [Service: [Key: Data]] = [:]
    
    func save(key: String, data: Data, service: String, accessible: CFString?) {
        storage[service]?[key] = data
    }
    
    func load(key: String, service: String) -> Data? {
        storage[service]?[key]
    }
    
    func delete(key: String, service: String) {
        storage[service]?.removeValue(forKey: key)
    }
}