98 lines
4.4 KiB
Rust
98 lines
4.4 KiB
Rust
// file: crates/ksp-raw-transaction-lib/unit_tests/signature.rs
|
|
// version: 4
|
|
|
|
use base64::Engine; // rust-rules: trait-import
|
|
|
|
#[test]
|
|
fn pre_002_signature_parser_accepts_exact_sixty_four_zero_bytes() {
|
|
let text = "1".repeat(crate::MIN_RAW_TRANSACTION_SIGNATURE_TEXT_BYTES);
|
|
let signature = crate::parse_raw_transaction_signature(text.as_str());
|
|
assert!(signature.is_ok());
|
|
if let std::result::Result::Ok(signature) = signature {
|
|
assert_eq!(signature.as_bytes(), &[0_u8; 64]);
|
|
}
|
|
return;
|
|
}
|
|
|
|
#[test]
|
|
fn pre_002_signature_parser_rejects_text_bounds_invalid_base58_and_noncanonical_lengths() {
|
|
for text in [
|
|
"1".repeat(crate::MIN_RAW_TRANSACTION_SIGNATURE_TEXT_BYTES - 1),
|
|
"1".repeat(crate::MAX_RAW_TRANSACTION_SIGNATURE_TEXT_BYTES + 1),
|
|
format!("{}0", "1".repeat(crate::MIN_RAW_TRANSACTION_SIGNATURE_TEXT_BYTES - 1)),
|
|
"1".repeat(65),
|
|
"z".repeat(crate::MAX_RAW_TRANSACTION_SIGNATURE_TEXT_BYTES),
|
|
] {
|
|
let result = crate::parse_raw_transaction_signature(text.as_str());
|
|
assert!(result.is_err());
|
|
if let std::result::Result::Err(error) = result {
|
|
assert_eq!(error.code(), crate::ERROR_CODE_RAW_TRANSACTION_SIGNATURE_INVALID);
|
|
let debug = std::format!("{error:?}");
|
|
assert!(!debug.contains(text.as_str()));
|
|
}
|
|
}
|
|
return;
|
|
}
|
|
|
|
#[test]
|
|
fn pre_004_embedded_signature_extracts_first_signature_from_canonical_transaction_base64() {
|
|
let encoded = "AQAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA";
|
|
let signature = crate::extract_raw_transaction_signature_from_binary_base64(encoded);
|
|
assert!(signature.is_ok());
|
|
if let std::result::Result::Ok(signature) = signature {
|
|
assert_eq!(signature.as_bytes(), &[0_u8; 64]);
|
|
}
|
|
return;
|
|
}
|
|
|
|
#[test]
|
|
fn pre_004_embedded_signature_rejects_invalid_base64_noncanonical_short_vec_and_truncation() {
|
|
for encoded in [
|
|
"not-base64",
|
|
"gQAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA==",
|
|
"AQAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA==",
|
|
"AQAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA=",
|
|
] {
|
|
let result = crate::extract_raw_transaction_signature_from_binary_base64(encoded);
|
|
assert!(result.is_err());
|
|
if let std::result::Result::Err(error) = result {
|
|
assert_eq!(error.code(), crate::ERROR_CODE_RAW_TRANSACTION_SIGNATURE_INVALID);
|
|
assert!(!error.to_string().contains(encoded));
|
|
}
|
|
}
|
|
return;
|
|
}
|
|
|
|
#[test]
|
|
fn pre_006_embedded_signature_extracts_terminal_v1_signature_and_rejects_trailing_bytes() {
|
|
let encoded = "gQEAAR8AAABRUVFRUVFRUVFRUVFRUVFRUVFRUVFRUVFRUVFRUVFRUQECQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFBQUFCQkJCQkJCQkJCQkJCQkJCQkJCQkJCQkJCQkJCQkJCQgcAAAAAAAAACAAAAAkAAAAAgAAAAQEDAAABAgOqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqq";
|
|
let signature = crate::extract_raw_transaction_signature_from_binary_base64(encoded);
|
|
assert!(signature.is_ok());
|
|
if let std::result::Result::Ok(signature) = signature {
|
|
assert_eq!(signature.as_bytes(), &[0xaa_u8; 64]);
|
|
}
|
|
let decoded = base64::engine::general_purpose::STANDARD.decode(encoded);
|
|
assert!(decoded.is_ok());
|
|
if let std::result::Result::Ok(mut decoded) = decoded {
|
|
decoded.push(0);
|
|
let trailing = base64::engine::general_purpose::STANDARD.encode(decoded);
|
|
assert!(crate::extract_raw_transaction_signature_from_binary_base64(trailing.as_str()).is_err());
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn pre_004_base58_formatter_round_trips_canonical_signatures() {
|
|
for bytes in [[0_u8; 64], [1_u8; 64], [0xAB_u8; 64]] {
|
|
let signature = ksp_store_api::RawTransactionSignature::new(bytes);
|
|
let text = crate::format_raw_transaction_signature(&signature);
|
|
assert!((crate::MIN_RAW_TRANSACTION_SIGNATURE_TEXT_BYTES..=crate::MAX_RAW_TRANSACTION_SIGNATURE_TEXT_BYTES).contains(&text.len()));
|
|
let parsed = crate::parse_raw_transaction_signature(text.as_str());
|
|
assert!(parsed.is_ok());
|
|
if let std::result::Result::Ok(parsed) = parsed {
|
|
assert_eq!(parsed, signature);
|
|
}
|
|
}
|
|
assert_eq!(crate::format_raw_transaction_signature(&ksp_store_api::RawTransactionSignature::new([0_u8; 64])), "1".repeat(64));
|
|
return;
|
|
}
|