Complete hash_computer coverage; removed unneeded dependency

main
Inga 🏳‍🌈 4 years ago
parent 2c68d8bc71
commit 2d71f1733b
  1. 2
      .gitignore
  2. 1
      Cargo.toml
  3. 88
      tests/hash_computer_test.rs

2
.gitignore vendored

@ -9,4 +9,6 @@ Cargo.lock
# These are backup files generated by rustfmt # These are backup files generated by rustfmt
**/*.rs.bk **/*.rs.bk
*.profraw
*.profdata
.vs .vs

@ -8,7 +8,6 @@ edition = "2018"
[dependencies] [dependencies]
bit_field = "0.10.1" bit_field = "0.10.1"
crunchy = "0.2.2"
md5 = "0.7.0" md5 = "0.7.0"
packed_simd = { version = "0.3.4", package = "packed_simd_2", features = ["into_bits"] } packed_simd = { version = "0.3.4", package = "packed_simd_2", features = ["into_bits"] }
permutohedron = "0.2.4" permutohedron = "0.2.4"

@ -1,31 +1,15 @@
#![feature(array_map)]
use packed_simd::u8x32; use packed_simd::u8x32;
extern crate trustpilot_challenge_rust; extern crate trustpilot_challenge_rust;
use trustpilot_challenge_rust::hash_computer; use trustpilot_challenge_rust::hash_computer;
fn prepare_message(message_string: &str) -> u8x32 { fn parse_hash(hash: u128) -> u32 {
let mut bytes_static: [u8; 32] = [0; 32]; ((hash >> 96) as u32).to_be()
let bytes = message_string.as_bytes();
for i in 0..bytes.len() {
bytes_static[i] = bytes[i];
} }
bytes_static[bytes.len()] = b' '; static MESSAGES: [&str; 8] = [
u8x32::from(bytes_static)
}
fn prepare_messages(message_strings: [&str; 8]) -> [u8x32; 8] {
let mut result: [u8x32; 8] = [u8x32::splat(0); 8];
for i in 0..8 {
result[i] = prepare_message(message_strings[i]);
}
result
}
#[test]
fn it_computes_hashes() {
let messages: [&str; 8] = /*[""; 8];*/[
"DAPUpOGHw620yalJA0vjFPK7ThgHyAN", "DAPUpOGHw620yalJA0vjFPK7ThgHyAN",
"4xRslaTeBCNyRu2EiIDueEx3BTbIP5H", "4xRslaTeBCNyRu2EiIDueEx3BTbIP5H",
"kFPd2zk60eEFpNwgEOZAcyDcxRVv0Y8", "kFPd2zk60eEFpNwgEOZAcyDcxRVv0Y8",
@ -36,7 +20,7 @@ fn it_computes_hashes() {
"VXqOIzYdLIqx6tw8LJbR7SqR5iYgTlQ" "VXqOIzYdLIqx6tw8LJbR7SqR5iYgTlQ"
]; ];
let expected: [u128; 8] = [ static EXPECTED_HASHES: [u128; 8] = [
0xC6F9E9B203CEA81A7BA28BE276B96A6F, 0xC6F9E9B203CEA81A7BA28BE276B96A6F,
0x45BF15D8B08E1AEADE1305B8E43B8F2C, 0x45BF15D8B08E1AEADE1305B8E43B8F2C,
0x7AA53B4627C8DD3714F2874EDE04DA7D, 0x7AA53B4627C8DD3714F2874EDE04DA7D,
@ -47,11 +31,65 @@ fn it_computes_hashes() {
0xCED45BA82BD1BDCF546255CB6A530FE3, 0xCED45BA82BD1BDCF546255CB6A530FE3,
]; ];
let messages_simd = prepare_messages(messages); // just some random u128 numbers
static WRONG_HASHES: [u128; 8] = [
0xF6999358632CA7EBC3E687329B5B330E,
0x2DFA25CE0B6E11E794A4BAE87EECBBE4,
0x24436F08BD36CD5E010C4469D6612462,
0x78A35F791CA6C57396C1C4CB6A773848,
0xB6906785D8BD81BAF1AA1E2A9F1D17E9,
0x7841A7132C2784CE7FAC83AD01A43B45,
0x10D0C197833BE4365F9D5C8262005C77,
0x654CB866EE50773F999CD32C123E6C12,
];
static MIXED_HASHES: [u128; 4] = [
WRONG_HASHES[0],
EXPECTED_HASHES[3],
WRONG_HASHES[1],
EXPECTED_HASHES[6],
];
fn prepare_message(message_string: &str) -> u8x32 {
let mut bytes_static: [u8; 32] = [0; 32];
let bytes = message_string.as_bytes();
for i in 0..bytes.len() {
bytes_static[i] = bytes[i];
}
let hashes = hash_computer::compute_hashes(messages_simd, messages[0].len()); bytes_static[bytes.len()] = b' ';
u8x32::from(bytes_static)
}
fn prepare_messages(message_strings: [&str; 8]) -> [u8x32; 8] {
let mut result: [u8x32; 8] = [u8x32::splat(0); 8];
for i in 0..8 { for i in 0..8 {
assert_eq!((expected[i] >> 96) as u32, hashes[i].to_be()); result[i] = prepare_message(message_strings[i]);
} }
result
}
#[test]
fn it_computes_hashes() {
let messages_simd = prepare_messages(MESSAGES);
let hashes = hash_computer::compute_hashes(&messages_simd, MESSAGES[0].len());
assert_eq!(EXPECTED_HASHES.map(parse_hash), hashes);
}
#[test]
fn it_verifies_hashes() {
let messages_simd = prepare_messages(MESSAGES);
let results = hash_computer::find_hashes(&messages_simd, MESSAGES[0].len(), &(EXPECTED_HASHES.map(parse_hash)));
assert!(results.is_some());
assert_eq!(results.unwrap(), messages_simd);
let results = hash_computer::find_hashes(&messages_simd, MESSAGES[0].len(), &(WRONG_HASHES.map(parse_hash)));
assert!(results.is_none());
let results = hash_computer::find_hashes(&messages_simd, MESSAGES[0].len(), &(MIXED_HASHES.map(parse_hash)));
assert!(results.is_some());
assert_eq!(results.unwrap(), [messages_simd[3], messages_simd[6]]);
} }
Loading…
Cancel
Save