95 lines
2.7 KiB
Nim
95 lines
2.7 KiB
Nim
## Nim-Codex
|
|
## Copyright (c) 2023 Status Research & Development GmbH
|
|
## Licensed under either of
|
|
## * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE))
|
|
## * MIT license ([LICENSE-MIT](LICENSE-MIT))
|
|
## at your option.
|
|
## This file may not be copied, modified, or distributed except according to
|
|
## those terms.
|
|
|
|
import std/random
|
|
import std/sequtils
|
|
import pkg/chronos
|
|
import pkg/libp2p
|
|
import pkg/asynctest
|
|
import pkg/questionable
|
|
import pkg/questionable/results
|
|
import pkg/codex/blocktype as bt
|
|
import pkg/codex/stores/repostore
|
|
import pkg/codex/clock
|
|
|
|
import ../helpers/mocktimer
|
|
import ../helpers/mockrepostore
|
|
import ../helpers/mockclock
|
|
import ../examples
|
|
|
|
import codex/namespaces
|
|
import codex/stores/keyutils
|
|
|
|
proc createManifestCid(): ?!Cid =
|
|
let
|
|
length = rand(4096)
|
|
bytes = newSeqWith(length, rand(uint8))
|
|
mcodec = multiCodec("sha2-256")
|
|
codec = multiCodec("dag-pb")
|
|
version = CIDv1
|
|
|
|
let hash = ? MultiHash.digest($mcodec, bytes).mapFailure
|
|
let cid = ? Cid.init(version, codec, hash).mapFailure
|
|
return success cid
|
|
|
|
suite "KeyUtils":
|
|
test "makePrefixKey should create block key":
|
|
let length = 6
|
|
let cid = Cid.example
|
|
let expectedPrefix = ($cid)[^length..^1]
|
|
let expectedPostfix = $cid
|
|
|
|
let key = !makePrefixKey(length, cid).option
|
|
let namespaces = key.namespaces
|
|
|
|
check:
|
|
namespaces.len == 4
|
|
namespaces[0].value == CodexRepoNamespace
|
|
namespaces[1].value == "blocks"
|
|
namespaces[2].value == expectedPrefix
|
|
namespaces[3].value == expectedPostfix
|
|
|
|
test "makePrefixKey should create manifest key":
|
|
let length = 6
|
|
let cid = !createManifestCid().option
|
|
let expectedPrefix = ($cid)[^length..^1]
|
|
let expectedPostfix = $cid
|
|
|
|
let key = !makePrefixKey(length, cid).option
|
|
let namespaces = key.namespaces
|
|
|
|
check:
|
|
namespaces.len == 4
|
|
namespaces[0].value == CodexRepoNamespace
|
|
namespaces[1].value == "manifests"
|
|
namespaces[2].value == expectedPrefix
|
|
namespaces[3].value == expectedPostfix
|
|
|
|
test "createBlockExpirationMetadataKey should create block TTL key":
|
|
let cid = Cid.example
|
|
|
|
let key = !createBlockExpirationMetadataKey(cid).option
|
|
let namespaces = key.namespaces
|
|
|
|
check:
|
|
namespaces.len == 3
|
|
namespaces[0].value == CodexMetaNamespace
|
|
namespaces[1].value == "ttl"
|
|
namespaces[2].value == $cid
|
|
|
|
test "createBlockExpirationMetadataQueryKey should create key for all block TTL entries":
|
|
let key = !createBlockExpirationMetadataQueryKey().option
|
|
let namespaces = key.namespaces
|
|
|
|
check:
|
|
namespaces.len == 3
|
|
namespaces[0].value == CodexMetaNamespace
|
|
namespaces[1].value == "ttl"
|
|
namespaces[2].value == "*"
|