import path from "path";
import {describeDirectorySpecTest, InputType} from "@chainsafe/lodestar-spec-test-util";
import {bytesToHex, hexToBytes} from "../../src/helpers";
import {SPEC_TESTS_DIR} from "../params";
import {describeForAllImplementations} from "../switch";
import {ZeroPrivateKeyError} from "../../src/errors";

interface ISignMessageTestCase {
  data: {
    input: {
      privkey: string;
      message: string;
    };
    output: string;
  };
}

describeForAllImplementations((bls) => {
  describeDirectorySpecTest<ISignMessageTestCase, string>(
    "bls/sign/small",
    path.join(SPEC_TESTS_DIR, "general/phase0/bls/sign/small"),
    (testCase) => {
      try {
        const {privkey, message} = testCase.data.input;
        const signature = bls.sign(hexToBytes(privkey), hexToBytes(message));
        return bytesToHex(signature);
      } catch (e) {
        if (e instanceof ZeroPrivateKeyError) return null;
        else throw e;
      }
    },
    {
      inputTypes: {data: InputType.YAML},
      getExpected: (testCase) => testCase.data.output,
    }
  );
});