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

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

forEachImplementation((bls, implementation) => {
  describeDirectorySpecTest<ISignMessageTestCase, string>(
    `${implementation} - bls/sign/small`,
    path.join(SPEC_TESTS_DIR, "general/phase0/bls/sign/small"),
    (testCase) => {
      const {privkey, message} = testCase.data.input;
      const signature = bls.sign(hexToBytes(privkey), hexToBytes(message));
      return bytesToHex(signature);
    },
    {
      inputTypes: {data: InputType.YAML},
      getExpected: (testCase) => testCase.data.output,
    }
  );
});