mirror of
git://jb55.com/damus
synced 2024-09-05 21:03:51 +00:00
Changelog-Fixed: Fix compilation error on test target in UserSearchCacheTests Signed-off-by: Daniel D‘Aquino <daniel@daquino.me> Signed-off-by: William Casarin <jb55@jb55.com>
138 lines
6.4 KiB
Swift
138 lines
6.4 KiB
Swift
//
|
|
// UserSearchCacheTests.swift
|
|
// damusTests
|
|
//
|
|
// Created by Terry Yiu on 6/30/23.
|
|
//
|
|
|
|
import XCTest
|
|
@testable import damus
|
|
|
|
final class UserSearchCacheTests: XCTestCase {
|
|
|
|
var keypair: FullKeypair? = nil
|
|
let damusState = DamusState.empty
|
|
let nip05 = "_@somedomain.com"
|
|
|
|
override func setUpWithError() throws {
|
|
keypair = try XCTUnwrap(generate_new_keypair())
|
|
|
|
if let keypair {
|
|
let pubkey = keypair.pubkey
|
|
let validatedNip05 = try XCTUnwrap(NIP05.parse(nip05))
|
|
|
|
damusState.profiles.set_validated(pubkey, nip05: validatedNip05)
|
|
|
|
let profile = Profile(name: "tyiu", display_name: "Terry Yiu", about: nil, picture: nil, banner: nil, website: nil, lud06: nil, lud16: nil, nip05: nip05, damus_donation: nil)
|
|
let timestampedProfile = TimestampedProfile(profile: profile, timestamp: 0, event: test_note)
|
|
damusState.profiles.add(id: pubkey, profile: timestampedProfile)
|
|
|
|
// Lookup to synchronize access on profiles dictionary to avoid race conditions.
|
|
let _ = damusState.profiles.lookup(id: pubkey)
|
|
}
|
|
}
|
|
|
|
override func tearDown() {
|
|
keypair = nil
|
|
}
|
|
|
|
func testSearch() throws {
|
|
let keypair = try XCTUnwrap(keypair)
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "tyiu"), [keypair.pubkey])
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "ty"), [keypair.pubkey])
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "terry yiu"), [keypair.pubkey])
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "rry"), [keypair.pubkey])
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "somedomain"), [keypair.pubkey])
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "dom"), [keypair.pubkey])
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "i"), [keypair.pubkey])
|
|
}
|
|
|
|
func testUpdateProfile() throws {
|
|
let keypair = try XCTUnwrap(keypair)
|
|
|
|
let newNip05 = "_@other.xyz"
|
|
_ = try XCTUnwrap(NIP05.parse(newNip05))
|
|
|
|
damusState.profiles.set_validated(keypair.pubkey, nip05: NIP05.parse(newNip05))
|
|
|
|
let newProfile = Profile(name: "whoami", display_name: "T-DAWG", about: nil, picture: nil, banner: nil, website: nil, lud06: nil, lud16: nil, nip05: newNip05, damus_donation: nil)
|
|
let newTimestampedProfile = TimestampedProfile(profile: newProfile, timestamp: 1000, event: test_note)
|
|
damusState.profiles.add(id: keypair.pubkey, profile: newTimestampedProfile)
|
|
|
|
// Lookup to synchronize access on profiles dictionary to avoid race conditions.
|
|
let _ = damusState.profiles.lookup(id: keypair.pubkey)
|
|
|
|
// Old profile attributes are removed from cache.
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "tyiu"), [])
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "ty"), [])
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "Terry Yiu"), [])
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "rry"), [])
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "somedomain"), [])
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "dom"), [])
|
|
|
|
// New profile attributes are added to cache.
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "whoami"), [keypair.pubkey])
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "hoa"), [keypair.pubkey])
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "t-dawg"), [keypair.pubkey])
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "daw"), [keypair.pubkey])
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "other"), [keypair.pubkey])
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "xyz"), [keypair.pubkey])
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "the"), [keypair.pubkey])
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "y"), [keypair.pubkey])
|
|
}
|
|
|
|
func testUpdateOwnContactsPetnames() throws {
|
|
let keypair = try XCTUnwrap(keypair)
|
|
let damus = "3efdaebb1d8923ebd99c9e7ace3b4194ab45512e2be79c1b7d68d9243e0d2681"
|
|
let jb55 = "32e1827635450ebb3c5a7d12c1f8e7b2b514439ac10a67eef3d9fd9c5c68e245"
|
|
|
|
var pubkeysToPetnames = [Pubkey: String]()
|
|
pubkeysToPetnames[damus] = "damus"
|
|
pubkeysToPetnames[jb55] = "jb55"
|
|
|
|
let contactsEvent = try createContactsEventWithPetnames(pubkeysToPetnames: pubkeysToPetnames)
|
|
|
|
// Initial own contacts event caching on searchable petnames.
|
|
damusState.user_search_cache.updateOwnContactsPetnames(id: keypair.pubkey, oldEvent: nil, newEvent: contactsEvent)
|
|
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "damus"), [damus])
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "jb55"), [jb55])
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "5"), [jb55])
|
|
|
|
// Replace one of the petnames and verify if the cache updates accordingly.
|
|
|
|
pubkeysToPetnames.removeValue(forKey: jb55)
|
|
pubkeysToPetnames[jb55] = "bill"
|
|
let newContactsEvent = try createContactsEventWithPetnames(pubkeysToPetnames: pubkeysToPetnames)
|
|
|
|
damusState.user_search_cache.updateOwnContactsPetnames(id: keypair.pubkey, oldEvent: contactsEvent, newEvent: newContactsEvent)
|
|
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "damus"), [damus])
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "jb55"), [])
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "5"), [])
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "bill"), [jb55])
|
|
XCTAssertEqual(damusState.user_search_cache.search(key: "l"), [jb55])
|
|
}
|
|
|
|
private func createContactsEventWithPetnames(pubkeysToPetnames: [Pubkey: String]) throws -> NostrEvent {
|
|
let keypair = try XCTUnwrap(keypair)
|
|
|
|
let bootstrapRelays = load_bootstrap_relays(pubkey: keypair.pubkey)
|
|
let relayInfo = RelayInfo(read: true, write: true)
|
|
var relays: [String: RelayInfo] = [:]
|
|
|
|
for relay in bootstrapRelays {
|
|
relays[relay] = relayInfo
|
|
}
|
|
|
|
let relayJson = encode_json(relays)!
|
|
|
|
let tags = pubkeysToPetnames.enumerated().map {
|
|
["p", $0.element.key, "", $0.element.value]
|
|
}
|
|
|
|
return NostrEvent(content: relayJson, keypair: keypair.to_keypair(), kind: NostrKind.contacts.rawValue, tags: tags)!
|
|
}
|
|
|
|
}
|