use criterion::{black_box, criterion_group, criterion_main, Criterion};
use hex_literal::hex;
use starknet_crypto::rfc6979_generate_k;
use starknet_types_core::felt::Felt;

pub fn criterion_benchmark(c: &mut Criterion) {
    let message_hash = hex!("010b559a3b4dc1b7137d90521cb413b397ff07963214d128a92d65aec7182f68");
    let private_key = hex!("07e3184f4bef18f371bc53fc412dff1b30dbc94f758490fb8e2349bae647a642");
    let seed = hex!("03fe27199aaad4e700559e2436a919f4de70def585a6deb2f4c087fdf6a27c1b");

    let message_hash = Felt::from_bytes_be(&message_hash);
    let private_key = Felt::from_bytes_be(&private_key);
    let seed = Felt::from_bytes_be(&seed);

    c.bench_function("rfc6979_generate_k", |b| {
        b.iter(|| {
            black_box(rfc6979_generate_k(&message_hash, &private_key, Some(&seed)));
        });
    });
}

criterion_group!(benches, criterion_benchmark);
criterion_main!(benches);