128 lines
No EOL
4.7 KiB
JavaScript
128 lines
No EOL
4.7 KiB
JavaScript
const assert = require('assert');
|
|
const { initialize } = require('../node/index.js');
|
|
|
|
describe('tests', function() {
|
|
|
|
// initialize once before running tests
|
|
before(function (done) {
|
|
initialize().then(zokrates => {
|
|
this.zokrates = zokrates;
|
|
done();
|
|
});
|
|
});
|
|
|
|
describe("compilation", () => {
|
|
it('should compile', function() {
|
|
assert.doesNotThrow(() => {
|
|
const artifacts = this.zokrates.compile("def main() -> (field): return 42", "main");
|
|
assert.ok(artifacts !== undefined);
|
|
})
|
|
});
|
|
|
|
it('should throw on invalid code', function() {
|
|
assert.throws(() => this.zokrates.compile(":-)", "main"));
|
|
});
|
|
|
|
it('should resolve stdlib module', function() {
|
|
assert.doesNotThrow(() => {
|
|
const code = `
|
|
import "hashes/sha256/512bit" as sha256
|
|
import "ecc/edwardsAdd" as edwardsAdd
|
|
def main() -> ():
|
|
return
|
|
`;
|
|
this.zokrates.compile(code, "main");
|
|
})
|
|
});
|
|
|
|
it('should resolve user module', function() {
|
|
assert.doesNotThrow(() => {
|
|
const code = 'import "test" as test\ndef main() -> (field): return test()';
|
|
this.zokrates.compile(code, "main", (_, path) => {
|
|
return {
|
|
source: "def main() -> (field): return 1",
|
|
location: path
|
|
}
|
|
});
|
|
})
|
|
});
|
|
|
|
it('should throw on invalid module', function() {
|
|
assert.throws(() => {
|
|
const code = 'import "test" as test\ndef main() -> (field): return test()';
|
|
this.zokrates.compile(code, "main", (_loc, _path) => null);
|
|
})
|
|
});
|
|
})
|
|
|
|
describe("computation", () => {
|
|
it('should compute with valid inputs', function() {
|
|
assert.doesNotThrow(() => {
|
|
const code = 'def main(private field a) -> (field): return a * a';
|
|
const artifacts = this.zokrates.compile(code, "main", null);
|
|
|
|
const result = this.zokrates.computeWitness(artifacts, ["2"])
|
|
const output = JSON.parse(result.output);
|
|
|
|
assert.deepEqual(output, ["4"]);
|
|
})
|
|
});
|
|
|
|
it('should throw on invalid input count', function() {
|
|
assert.throws(() => {
|
|
const code = 'def main(private field a) -> (field): return a * a';
|
|
const artifacts = this.zokrates.compile(code, "main", null);
|
|
|
|
this.zokrates.computeWitness(artifacts, ["1", "2"])
|
|
})
|
|
});
|
|
|
|
it('should throw on invalid input type', function() {
|
|
assert.throws(() => {
|
|
const code = 'def main(private field a) -> (field): return a * a';
|
|
const artifacts = this.zokrates.compile(code, "main", null);
|
|
|
|
this.zokrates.computeWitness(artifacts, [true])
|
|
})
|
|
});
|
|
})
|
|
|
|
describe("setup", () => {
|
|
it('should run setup', function() {
|
|
assert.doesNotThrow(() => {
|
|
const code = 'def main(private field a) -> (field): return a * a';
|
|
const artifacts = this.zokrates.compile(code, "main", null);
|
|
|
|
this.zokrates.setup(artifacts.program);
|
|
})
|
|
});
|
|
})
|
|
|
|
describe("export-verifier", () => {
|
|
it('should export solidity verifier', function() {
|
|
assert.doesNotThrow(() => {
|
|
const code = 'def main(private field a) -> (field): return a * a';
|
|
const artifacts = this.zokrates.compile(code, "main", null);
|
|
const keypair = this.zokrates.setup(artifacts.program);
|
|
|
|
const verifier = this.zokrates.exportSolidityVerifier(keypair.vk, "v1");
|
|
assert.ok(verifier.length > 0)
|
|
})
|
|
});
|
|
})
|
|
|
|
describe("generate-proof", () => {
|
|
it('should generate proof', function() {
|
|
assert.doesNotThrow(() => {
|
|
const code = 'def main(private field a) -> (field): return a * a';
|
|
const artifacts = this.zokrates.compile(code, "main", null);
|
|
const computationResult = this.zokrates.computeWitness(artifacts, ["2"])
|
|
const keypair = this.zokrates.setup(artifacts.program);
|
|
const proof = this.zokrates.generateProof(artifacts.program, computationResult.witness, keypair.pk);
|
|
|
|
assert.ok(proof !== undefined);
|
|
assert.deepEqual(proof.inputs, ["0x0000000000000000000000000000000000000000000000000000000000000004"])
|
|
})
|
|
});
|
|
});
|
|
}); |