// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
// Copyright by contributors to this project.
// SPDX-License-Identifier: (Apache-2.0 OR MIT)

use criterion::{BatchSize, BenchmarkId, Criterion};
use mls_rs::{test_utils::benchmarks::load_group_states, CipherSuite};

fn bench(c: &mut Criterion) {
    let cipher_suite = CipherSuite::CURVE25519_AES128;
    let group_states = load_group_states(cipher_suite);
    let mut bench_group = c.benchmark_group("group_receive_commit");

    for (i, mut group_states) in group_states.into_iter().enumerate() {
        bench_group.bench_with_input(
            BenchmarkId::new(format!("{cipher_suite:?}"), i),
            &i,
            |b, _| {
                b.iter_batched_ref(
                    || {
                        let commit = group_states.sender.commit(Vec::new()).unwrap();
                        group_states.sender.clear_pending_commit();
                        (commit, group_states.receiver.clone())
                    },
                    move |(commit, receiver)| {
                        receiver
                            .process_incoming_message(commit.commit_message.clone())
                            .unwrap();
                    },
                    BatchSize::SmallInput,
                )
            },
        );
    }

    bench_group.finish();
}

criterion::criterion_group!(benches, bench);
criterion::criterion_main!(benches);