95 lines
3.0 KiB
Nim
95 lines
3.0 KiB
Nim
import pkg/codex/rest/json
|
|
import ./twonodes
|
|
import ../codex/examples
|
|
import json
|
|
from pkg/libp2p import Cid, `$`
|
|
|
|
twonodessuite "Uploads and downloads":
|
|
test "node allows local file downloads", twoNodesConfig:
|
|
let content1 = "some file contents"
|
|
let content2 = "some other contents"
|
|
|
|
let cid1 = client1.upload(content1).get
|
|
let cid2 = client2.upload(content2).get
|
|
|
|
let resp1 = client1.download(cid1, local = true).get
|
|
let resp2 = client2.download(cid2, local = true).get
|
|
|
|
check:
|
|
content1 == resp1
|
|
content2 == resp2
|
|
|
|
test "node allows remote file downloads", twoNodesConfig:
|
|
let content1 = "some file contents"
|
|
let content2 = "some other contents"
|
|
|
|
let cid1 = client1.upload(content1).get
|
|
let cid2 = client2.upload(content2).get
|
|
|
|
let resp2 = client1.download(cid2, local = false).get
|
|
let resp1 = client2.download(cid1, local = false).get
|
|
|
|
check:
|
|
content1 == resp1
|
|
content2 == resp2
|
|
|
|
test "node fails retrieving non-existing local file", twoNodesConfig:
|
|
let content1 = "some file contents"
|
|
let cid1 = client1.upload(content1).get # upload to first node
|
|
let resp2 = client2.download(cid1, local = true) # try retrieving from second node
|
|
|
|
check:
|
|
resp2.error.msg == "404 Not Found"
|
|
|
|
proc checkRestContent(cid: Cid, content: ?!string) =
|
|
let c = content.tryGet()
|
|
|
|
# tried to JSON (very easy) and checking the resulting object (would be much nicer)
|
|
# spent an hour to try and make it work.
|
|
let jsonData = parseJson(c)
|
|
|
|
check jsonData.hasKey("cid") == true
|
|
|
|
check jsonData["cid"].getStr() == $cid
|
|
check jsonData.hasKey("manifest") == true
|
|
|
|
let manifest = jsonData["manifest"]
|
|
|
|
check manifest.hasKey("treeCid") == true
|
|
check manifest["treeCid"].getStr() == "zDzSvJTezk7bJNQqFq8k1iHXY84psNuUfZVusA5bBQQUSuyzDSVL"
|
|
check manifest.hasKey("datasetSize") == true
|
|
check manifest["datasetSize"].getInt() == 18
|
|
check manifest.hasKey("blockSize") == true
|
|
check manifest["blockSize"].getInt() == 65536
|
|
check manifest.hasKey("protected") == true
|
|
check manifest["protected"].getBool() == false
|
|
|
|
test "node allows downloading only manifest", twoNodesConfig:
|
|
let content1 = "some file contents"
|
|
let cid1 = client1.upload(content1).get
|
|
|
|
let resp2 = client1.downloadManifestOnly(cid1)
|
|
checkRestContent(cid1, resp2)
|
|
|
|
test "node allows downloading content without stream", twoNodesConfig:
|
|
let content1 = "some file contents"
|
|
let cid1 = client1.upload(content1).get
|
|
|
|
let resp1 = client2.downloadNoStream(cid1)
|
|
checkRestContent(cid1, resp1)
|
|
let resp2 = client2.download(cid1, local = true).get
|
|
check:
|
|
content1 == resp2
|
|
|
|
test "reliable transfer test", twoNodesConfig:
|
|
proc transferTest(a: CodexClient, b: CodexClient) {.async.} =
|
|
let data = await RandomChunker.example(blocks=8)
|
|
let cid = a.upload(data).get
|
|
let response = b.download(cid).get
|
|
check:
|
|
response == data
|
|
|
|
for run in 0..10:
|
|
await transferTest(client1, client2)
|
|
await transferTest(client2, client1)
|