use rand_core::OsRng;

use group::{ff::{Field, PrimeFieldBits}, prime::PrimeGroup};
use multiexp::BatchVerifier;

use transcript::{Transcript, RecommendedTranscript};

use crate::cross_group::schnorr::SchnorrPoK;

fn test_schnorr<G: PrimeGroup>() where G::Scalar: PrimeFieldBits {
  let private = G::Scalar::random(&mut OsRng);

  let transcript = RecommendedTranscript::new(b"Schnorr Test");
  let mut batch = BatchVerifier::new(3);
  SchnorrPoK::prove(
    &mut OsRng,
    &mut transcript.clone(),
    G::generator(),
    private
  ).verify(
    &mut OsRng,
    &mut transcript.clone(),
    G::generator(),
    G::generator() * private,
    &mut batch
  );
  assert!(batch.verify_vartime());
}

#[test]
fn test_secp256k1() {
  test_schnorr::<k256::ProjectivePoint>();
}

#[test]
fn test_ed25519() {
  test_schnorr::<dalek_ff_group::EdwardsPoint>();
}