diff --git a/set1/chal1/src/main.rs b/set1/chal1/src/main.rs index 38086f6..cfeb3ed 100644 --- a/set1/chal1/src/main.rs +++ b/set1/chal1/src/main.rs @@ -11,7 +11,9 @@ use std::env; /// # Examples /// /// ~~~ -/// hex_to_string("49") => I +/// let test: Vec = vec![String::from("41"), +/// String::from("42")]; +/// assert_eq!(hex_to_string(&test), "AB"); /// ~~~ fn hex_to_string(hex_char: &Vec) -> String { let mut output = String::new(); @@ -30,7 +32,7 @@ fn hex_to_string(hex_char: &Vec) -> String { /// # Examples /// /// ~~~ -/// let test: Vec = vec![0x41, 0x42]; +/// let test: Vec = vec![0x41, 0x42]; // "[A,B] in ascii" /// let b64_encode = base64_encode(&test); /// assert_eq!(b64_encode, "QUI="); /// ~~~ @@ -91,11 +93,16 @@ fn base64_encode(input: &Vec) -> String { } fn main() { - // Test Base64 encode - let test: Vec = vec![0x41, 0x42]; // "[A,B] in ascii" + // Test base64 encode() + let test: Vec = vec![0x41, 0x42]; let b64_encode = base64_encode(&test); assert_eq!(b64_encode, "QUI="); + // Test hex_to_string() + let test: Vec = vec![String::from("41"), + String::from("42")]; + assert_eq!(hex_to_string(&test), "AB"); + let args: Vec = env::args().collect(); let input = &args[1];