// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
//
// Copyright (c) DUSK NETWORK. All rights reserved.

use dusk_bls12_381::BlsScalar;
use dusk_bytes::Serializable;
use dusk_schnorr::{PublicKeyVarGen, SecretKeyVarGen, SignatureVarGen};
use ff::Field;
use rand::rngs::StdRng;
use rand::SeedableRng;

#[test]
fn sign_verify() {
    let mut rng = StdRng::seed_from_u64(2321u64);

    let sk = SecretKeyVarGen::random(&mut rng);
    let message = BlsScalar::random(&mut rng);
    let pk = PublicKeyVarGen::from(&sk);

    let sig = sk.sign(&mut rng, message);

    assert!(pk.verify(&sig, message));
}

#[test]
fn test_wrong_keys() {
    let mut rng = StdRng::seed_from_u64(2321u64);

    let sk = SecretKeyVarGen::random(&mut rng);
    let message = BlsScalar::random(&mut rng);

    let sig = sk.sign(&mut rng, message);

    // Derive random public key
    let pk = PublicKeyVarGen::from(&SecretKeyVarGen::random(&mut rng));

    assert!(!pk.verify(&sig, message));
}

#[test]
fn to_from_bytes() {
    let mut rng = StdRng::seed_from_u64(2321u64);

    let sk = SecretKeyVarGen::random(&mut rng);
    let message = BlsScalar::random(&mut rng);

    let sig = sk.sign(&mut rng, message);
    assert_eq!(sig, SignatureVarGen::from_bytes(&sig.to_bytes()).unwrap());
}