1
0
mirror of git://jb55.com/damus synced 2024-09-19 11:43:44 +00:00
damus/damusTests/Mocking/MockProfiles.swift

29 lines
746 B
Swift
Raw Normal View History

//
// MockProfiles.swift
// damusTests
//
// Created by Daniel DAquino on 2023-10-13.
//
import Foundation
@testable import damus
// A Mockable `Profiles` class that can be used for testing.
// Note: Not all methods are mocked. You might need to implement a method depending on the test you are writing.
class MockProfiles: Profiles {
var mocked_profiles: [Pubkey: Profile] = [:]
var ndb: Ndb
init?(mocked_profiles: [Pubkey : Profile], ndb: Ndb) {
self.mocked_profiles = mocked_profiles
self.ndb = ndb
super.init(ndb: ndb)
}
override func lookup(id: Pubkey) -> NdbTxn<Profile?> {
return NdbTxn(ndb: self.ndb) { txn in
return self.mocked_profiles[id]
}
}
}