o1js/src/lib/testing/testing.unit-test.ts

TypeScript framework for zk-SNARKs and zkApps

import { expect } from 'expect';
import { jsLayout } from '../../bindings/mina-transaction/gen/v1/js-layout.js';
import { Signature } from '../../mina-signer/src/signature.js';
import {
  AccountUpdate,
  PublicKey,
  UInt32,
  UInt64,
  signableFromLayout,
  ZkappCommand,
  Json,
} from '../../bindings/mina-transaction/gen/v1/transaction-bigint.js';
import { test, Random } from './property.js';

// some trivial roundtrip tests
test(Random.accountUpdate, (accountUpdate, assert) => {
  let json = AccountUpdate.toJSON(accountUpdate);
  let jsonString = JSON.stringify(json);
  assert(jsonString === JSON.stringify(AccountUpdate.toJSON(AccountUpdate.fromJSON(json))));
  // TODO add back using `fromValue`
  // let fields = AccountUpdate.toFields(accountUpdate);
  // let auxiliary = AccountUpdate.toAuxiliary(accountUpdate);
  // let recovered = AccountUpdate.fromFields(fields, auxiliary);
  // assert(jsonString === JSON.stringify(AccountUpdate.toJSON(recovered)));
});
test(Random.json.accountUpdate, (json) => {
  let jsonString = JSON.stringify(json);
  expect(jsonString).toEqual(JSON.stringify(AccountUpdate.toJSON(AccountUpdate.fromJSON(json))));
});

// check that test fails for a property that does not hold in general
let testSilent = test.custom({ logFailures: false });
expect(() => {
  testSilent(Random.nat(100), Random.nat(100), (x, y, assert) => {
    assert(x !== y, 'two different numbers can never be the same');
  });
}).toThrow('two different numbers');

// check that invalid JSON cannot be parsed
// note: test.negative asserts that _every_ sample fails
test.negative(Random.json.uint64.invalid, UInt64.fromJSON);
test.negative(Random.json.uint32.invalid, UInt32.fromJSON);
test.negative(Random.json.publicKey.invalid, PublicKey.fromJSON);
test.negative(Random.json.signature.invalid, Signature.fromBase58);

test.custom({ negative: true, timeBudget: 1000 })(
  Random.json.accountUpdate.invalid!,
  AccountUpdate.fromJSON
);

const FeePayer = signableFromLayout<ZkappCommand['feePayer'], Json.ZkappCommand['feePayer']>(
  jsLayout.ZkappCommand.entries.feePayer as any
);

test.negative(Random.json.feePayer.invalid!, FeePayer.fromJSON);