Skip to content

Commit

Permalink
Implemented VecAssertions for referenced vectors
Browse files Browse the repository at this point in the history
  • Loading branch information
Kirill Pimenov authored and woshilapin committed Oct 5, 2021
1 parent 8262886 commit 35856a0
Show file tree
Hide file tree
Showing 2 changed files with 45 additions and 0 deletions.
5 changes: 5 additions & 0 deletions src/option.rs
Original file line number Diff line number Diff line change
Expand Up @@ -126,6 +126,11 @@ mod tests {
assert_that(&option).is_some().is_equal_to(&"Hello");
}

#[test]
fn should_be_able_to_unwrap_referenced_vec() {
assert_that(&Some(&vec![1, 2, 3])).is_some().has_length(3);
}

#[test]
fn contains_value_should_allow_multiple_borrow_types() {
let option = Some("Hello");
Expand Down
40 changes: 40 additions & 0 deletions src/vec.rs
Original file line number Diff line number Diff line change
Expand Up @@ -42,6 +42,42 @@ impl<'s, T> VecAssertions for Spec<'s, Vec<T>> {
}
}

impl<'s, T> VecAssertions for Spec<'s, &'s Vec<T>> {
/// Asserts that the length of the subject vector is equal to the provided length. The subject
/// type must be of `&Vec` with a matching lifetime.
///
/// ```rust
/// assert_that(&&vec![1, 2, 3, 4]).has_length(4);
/// ```
fn has_length(&mut self, expected: usize) {
let length = self.subject.len();
if length != expected {
AssertionFailure::from_spec(self)
.with_expected(format!("vec to have length <{}>", expected))
.with_actual(format!("<{}>", length))
.fail();
}
}

/// Asserts that the subject vector is empty. The subject type must be of `&Vec` with a
/// matching lifetime.
///
/// ```rust
/// let test_vec: &Vec<u8> = &vec![];
/// assert_that(&test_vec).is_empty();
/// ```
fn is_empty(&mut self) {
let subject = self.subject;

if !subject.is_empty() {
AssertionFailure::from_spec(self)
.with_expected("an empty vec".to_string())
.with_actual(format!("a vec with length <{:?}>", subject.len()))
.fail();
}
}
}

#[cfg(test)]
mod tests {

Expand All @@ -51,25 +87,29 @@ mod tests {
fn should_not_panic_if_vec_length_matches_expected() {
let test_vec = vec![1, 2, 3];
assert_that(&test_vec).has_length(3);
assert_that(&&test_vec).has_length(3);
}

#[test]
#[should_panic(expected = "\n\texpected: vec to have length <1>\n\t but was: <3>")]
fn should_panic_if_vec_length_does_not_match_expected() {
let test_vec = vec![1, 2, 3];
assert_that(&test_vec).has_length(1);
assert_that(&&test_vec).has_length(1);
}

#[test]
fn should_not_panic_if_vec_was_expected_to_be_empty_and_is() {
let test_vec: Vec<u8> = vec![];
assert_that(&test_vec).is_empty();
assert_that(&&test_vec).is_empty();
}

#[test]
#[should_panic(expected = "\n\texpected: an empty vec\
\n\t but was: a vec with length <1>")]
fn should_panic_if_vec_was_expected_to_be_empty_and_is_not() {
assert_that(&vec![1]).is_empty();
assert_that(&&vec![1]).is_empty();
}
}

0 comments on commit 35856a0

Please sign in to comment.