69 lines
1.7 KiB
V
69 lines
1.7 KiB
V
module dbfs
|
|
|
|
import os
|
|
import incubaid.herolib.core.texttools
|
|
import incubaid.herolib.crypt.secp256k1
|
|
import incubaid.herolib.ui.console
|
|
|
|
fn test_dbname1() {
|
|
data_dir := '/tmp/namedbtest'
|
|
|
|
mut ndb := namedb_new(data_dir)!
|
|
|
|
console.print_debug('delete ${data_dir}')
|
|
os.rmdir_all(data_dir) or {}
|
|
|
|
mut test_cases := []string{}
|
|
mut nr := 1
|
|
for i := 0; i < nr * 1000 + 1; i++ {
|
|
privkey := secp256k1.new()!
|
|
pubkey := privkey.public_key_base64()
|
|
if i % 1000 == 0 {
|
|
console.print_debug(i)
|
|
}
|
|
test_cases << pubkey
|
|
}
|
|
|
|
defer {
|
|
// os.rmdir_all(data_dir) or {}
|
|
console.print_debug('rmdir done')
|
|
}
|
|
|
|
// Register public keys and store their unique IDs
|
|
mut ids := []u32{}
|
|
mut i := 0
|
|
for pubkey in test_cases {
|
|
if i % 1000 == 0 {
|
|
console.print_debug('b${i}')
|
|
}
|
|
myid := ndb.set(pubkey, '${i}')!
|
|
ids << myid
|
|
i++
|
|
}
|
|
|
|
// Retrieve public keys using their unique IDs
|
|
console.print_debug('retrieve starts')
|
|
for i2, myid in ids {
|
|
retrieved_pubkey, data := ndb.get_from_id(myid)!
|
|
myid_found, data_found := ndb.get(retrieved_pubkey)!
|
|
assert myid_found == myid
|
|
assert data_found == data
|
|
tc := test_cases[i2] or {
|
|
panic("can't find ${i2} in test_cases with len: ${test_cases.len}")
|
|
}
|
|
assert retrieved_pubkey == tc, 'Retrieved pubkey doesn\'t match for ID: ${myid}'
|
|
}
|
|
|
|
console.print_debug('All tests passed!')
|
|
}
|
|
|
|
fn test_dbname2() {
|
|
assert namedb_dbid(0).str() == '(0, 0, 0)'
|
|
assert namedb_dbid(1).str() == '(0, 0, 1)'
|
|
assert namedb_dbid(255).str() == '(0, 0, 255)'
|
|
assert namedb_dbid(256).str() == '(0, 1, 0)'
|
|
assert namedb_dbid(256 * 256).str() == '(1, 0, 0)'
|
|
assert namedb_dbid(256 * 256 + 3).str() == '(1, 0, 3)'
|
|
assert namedb_dbid(256 * 256 + 3 + 256).str() == '(1, 1, 3)'
|
|
}
|