use ncryptf::{Request, Response};

use crate::ncryptfts::cases::get_expected_cipher;

use super::cases::{get_expected_v2_cipher, get_payload, CData};

#[test]
fn test_v2_encrypt_decrypt() {
    let c = CData::init();
    match Request::from(c.client_kp_secret, c.signature_kp_secret) {
        Ok(mut request) => {
            match request.encrypt_with_nonce(
                get_payload(),
                c.server_kp_public,
                Some(c.nonce.clone()),
                Some(2),
            ) {
                Ok(cipher) => {
                    assert_eq!(cipher, get_expected_v2_cipher());
                    match Response::from(c.server_kp_secret) {
                        Ok(response) => match response.decrypt(cipher, None, None) {
                            Ok(text) => {
                                assert_eq!(text, get_payload());
                            }
                            _ => assert!(false),
                        },
                        _ => assert!(false),
                    }
                }
                _ => assert!(false),
            };
        }
        _ => {
            assert!(false);
        }
    };
}

#[test]
fn test_v2_decrypt_empty_string() {
    let c = CData::init();
    match Request::from(c.client_kp_secret, c.signature_kp_secret) {
        Ok(mut request) => {
            match request.encrypt_with_nonce(
                "".to_string(),
                c.server_kp_public,
                Some(c.nonce.clone()),
                Some(2),
            ) {
                Ok(cipher) => match Response::from(c.server_kp_secret) {
                    Ok(response) => match response.decrypt(cipher, None, None) {
                        Ok(text) => {
                            assert_eq!(text, "".to_string());
                        }
                        _ => assert!(false),
                    },
                    _ => assert!(false),
                },
                _ => assert!(false),
            };
        }
        _ => {
            assert!(false);
        }
    };
}

#[test]
fn test_v1_encrypt_decrypt() {
    let c = CData::init();
    match Request::from(c.client_kp_secret, c.signature_kp_secret) {
        Ok(mut request) => {
            match request.encrypt_with_nonce(
                get_payload(),
                c.server_kp_public.clone(),
                Some(c.nonce.clone()),
                None,
            ) {
                Ok(cipher) => {
                    assert_eq!(cipher, get_expected_cipher());
                    match Response::from(c.server_kp_secret) {
                        Ok(response) => {
                            match response.decrypt(cipher, Some(c.client_kp_public), Some(c.nonce))
                            {
                                Ok(text) => {
                                    assert_eq!(text, get_payload());
                                }
                                _ => assert!(false),
                            }
                        }
                        _ => assert!(false),
                    }
                }
                _ => assert!(false),
            };
        }
        _ => {
            assert!(false);
        }
    };
}