example.rs

#[cfg(test)]
mod tests {
    use byte_marks::ByteMarker;
    use byte_marks::Marked;
    use rand::Rng;
    use std::io::Cursor;

    #[test]
    fn test_streaming_with_marks_and_tail() {
        //message_stream -> stream of messages -> msg|suffix|msg|suffix|msg|tail
        let message_stream = "StreamingsUfFiX withsUfFiX markssUfFiX and tailtAiL";

        //The following is for only showing validation
        let mut messages = Vec::new();

        messages.push("Streaming".as_bytes());
        messages.push(" with".as_bytes());
        messages.push(" marks".as_bytes());
        messages.push(" and tail".as_bytes());

        //Cursor is akin to a TcpStream
        let mut cursor = Cursor::new(message_stream.as_bytes());

        //From the stream create an iterator and validate the received message chunks
        let stream = Marked::new(&mut cursor, "sUfFiX", "tAiL");

        let zipped = stream.into_iter().zip(messages.iter());

        for (unmarked, message) in zipped {
            assert!(unmarked == message.to_vec());
        }
    }

    #[test]
    fn unmark_random_msg_test_1() {
        let random_texts = [
            "Some random",
            "strings from this array",
            "are getiing",
            "picked and being",
            "converted to",
            "to bytes",
            "and",
            "then suffixed",
            "with byte marks.",
            "These",
            "marked bytes",
            "are then getting",
            "stripped off of",
            "demarcating",
            "pattern",
        ];
        let mut randomizer = rand::thread_rng();
        let num_strings: usize = randomizer.gen_range(0..1000);

        let mut orig_strings = vec![];
        let mut marked_bytes = vec![];

        let marker = ByteMarker::new("sUfFiX", "");

        for _ in 0..num_strings {
            //Pick a random index
            let index = randomizer.gen_range(0..random_texts.len());
            //Pick a random string for the given index
            let picked_string = random_texts[index];

            //Get the bytes and demarcate with byte marks
            let mut bytes = picked_string.as_bytes().to_vec();

            marker.mark_bytes(&mut bytes);

            //Preserve the old string for later validation
            orig_strings.push(picked_string);

            //Keep extending the marked bytes
            marked_bytes.extend(bytes);
        }

        //Get the orginal strings back from marked bytes
        let unmarked = marker.unmark(&marked_bytes).unwrap().0;

        //We must get same bytes back with demarcating bytes removed
        //Lets reconstruct the strings back and validate

        for i in 0..unmarked.len() {
            assert_eq!(
                orig_strings[i],
                String::from_utf8(unmarked[i].to_vec()).unwrap()
            );
        }
    }
}