nim-codex/tests/codex/stores/testkeyutils.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 == "*"