mirror of
https://github.com/logos-storage/logos-storage-nim.git
synced 2026-01-11 01:43:07 +00:00
112 lines
4.0 KiB
Nim
112 lines
4.0 KiB
Nim
import os
|
|
import ../../asynctest
|
|
|
|
import pkg/chronos
|
|
import pkg/taskpools
|
|
|
|
import pkg/confutils/defs
|
|
import pkg/codex/conf
|
|
import pkg/codex/slots/proofs/backends
|
|
import pkg/codex/slots/proofs/proverfactory {.all.}
|
|
import pkg/codex/utils/natutils
|
|
|
|
import ../helpers
|
|
import ../examples
|
|
|
|
suite "Test BackendFactory":
|
|
let circuitDir = "testecircuitdir"
|
|
|
|
setup:
|
|
createDir(circuitDir)
|
|
|
|
teardown:
|
|
removeDir(circuitDir)
|
|
|
|
test "Should initialize with correct nimGroth16 config files":
|
|
let config = CodexConf(
|
|
cmd: StartUpCmd.persistence,
|
|
nat: NatConfig(hasExtIp: false, nat: NatNone),
|
|
metricsAddress: parseIpAddress("127.0.0.1"),
|
|
persistenceCmd: PersistenceCmd.prover,
|
|
marketplaceAddress: EthAddress.example.some,
|
|
proverBackendCmd: ProverBackendCmd.nimGroth16,
|
|
circomGraph: InputFile("tests/circuits/fixtures/proof_main.bin"),
|
|
circomR1cs: InputFile("tests/circuits/fixtures/proof_main.r1cs"),
|
|
circomZkey: InputFile("tests/circuits/fixtures/proof_main.zkey"),
|
|
)
|
|
|
|
check:
|
|
getGraphFile(config).tryGet == $config.circomGraph
|
|
getR1csFile(config).tryGet == $config.circomR1cs
|
|
getZkeyFile(config).tryGet == $config.circomZkey
|
|
|
|
test "Should initialize with correct circom compat config files":
|
|
let config = CodexConf(
|
|
cmd: StartUpCmd.persistence,
|
|
nat: NatConfig(hasExtIp: false, nat: NatNone),
|
|
metricsAddress: parseIpAddress("127.0.0.1"),
|
|
persistenceCmd: PersistenceCmd.prover,
|
|
marketplaceAddress: EthAddress.example.some,
|
|
proverBackendCmd: ProverBackendCmd.circomCompat,
|
|
circomWasm: InputFile("tests/circuits/fixtures/proof_main.wasm"),
|
|
circomR1cs: InputFile("tests/circuits/fixtures/proof_main.r1cs"),
|
|
circomZkey: InputFile("tests/circuits/fixtures/proof_main.zkey"),
|
|
)
|
|
|
|
check:
|
|
getWasmFile(config).tryGet == $config.circomWasm
|
|
getR1csFile(config).tryGet == $config.circomR1cs
|
|
getZkeyFile(config).tryGet == $config.circomZkey
|
|
|
|
test "Should initialize circom compat from local directory":
|
|
let config = CodexConf(
|
|
cmd: StartUpCmd.persistence,
|
|
nat: NatConfig(hasExtIp: false, nat: NatNone),
|
|
metricsAddress: parseIpAddress("127.0.0.1"),
|
|
persistenceCmd: PersistenceCmd.prover,
|
|
marketplaceAddress: EthAddress.example.some,
|
|
proverBackendCmd: ProverBackendCmd.circomCompat,
|
|
# Set the circuitDir such that the tests/circuits/fixtures/ files
|
|
# will be picked up as local files:
|
|
circuitDir: OutDir("tests/circuits/fixtures"),
|
|
)
|
|
|
|
check:
|
|
getR1csFile(config).tryGet == config.circuitDir / "proof_main.r1cs"
|
|
getWasmFile(config).tryGet == config.circuitDir / "proof_main.wasm"
|
|
getZkeyFile(config).tryGet == config.circuitDir / "proof_main.zkey"
|
|
|
|
test "Should initialize nim groth16 from local directory":
|
|
let config = CodexConf(
|
|
cmd: StartUpCmd.persistence,
|
|
nat: NatConfig(hasExtIp: false, nat: NatNone),
|
|
metricsAddress: parseIpAddress("127.0.0.1"),
|
|
persistenceCmd: PersistenceCmd.prover,
|
|
marketplaceAddress: EthAddress.example.some,
|
|
proverBackendCmd: ProverBackendCmd.nimGroth16,
|
|
# Set the circuitDir such that the tests/circuits/fixtures/ files
|
|
# will be picked up as local files:
|
|
circuitDir: OutDir("tests/circuits/fixtures"),
|
|
)
|
|
|
|
check:
|
|
getGraphFile(config).tryGet == config.circuitDir / "proof_main.bin"
|
|
getR1csFile(config).tryGet == config.circuitDir / "proof_main.r1cs"
|
|
getZkeyFile(config).tryGet == config.circuitDir / "proof_main.zkey"
|
|
|
|
test "Should suggest usage of downloader tool when files not available":
|
|
let
|
|
config = CodexConf(
|
|
cmd: StartUpCmd.persistence,
|
|
nat: NatConfig(hasExtIp: false, nat: NatNone),
|
|
metricsAddress: parseIpAddress("127.0.0.1"),
|
|
persistenceCmd: PersistenceCmd.prover,
|
|
proverBackendCmd: ProverBackendCmd.nimGroth16,
|
|
marketplaceAddress: EthAddress.example.some,
|
|
circuitDir: OutDir(circuitDir),
|
|
)
|
|
proverResult = config.initializeProver(Taskpool.new())
|
|
|
|
check:
|
|
proverResult.isErr
|