Add coverage of spans allowing user requested data amounts

cppalliance · Jan 15, 2025 · e553f9e · e553f9e
1 parent a082b7e
commit e553f9e
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/test/test_shake256.cpp b/test/test_shake256.cpp
@@ -200,6 +200,21 @@ void test_class()
     hasher.finalize();
     const auto array_return4 {hasher.get_digest(small_container_span)};
     BOOST_TEST(array_return4 == boost::crypt::state::success);
+
+    // And the range interfaces
+    hasher.init();
+    hasher.process_bytes(bad_update_msg);
+    BOOST_TEST(hasher.get_digest(small_container, 10U) == boost::crypt::state::state_error);
+    hasher.finalize();
+    BOOST_TEST(hasher.get_digest(small_container, 10U) == boost::crypt::state::insufficient_output_length);
+    BOOST_TEST(hasher.get_digest(small_container, 5U) == boost::crypt::state::success);
+
+    hasher.init();
+    hasher.process_bytes(bad_update_msg);
+    BOOST_TEST(hasher.get_digest(small_container_span, 10U) == boost::crypt::state::state_error);
+    hasher.finalize();
+    BOOST_TEST(hasher.get_digest(small_container_span, 10U) == boost::crypt::state::insufficient_output_length);
+    BOOST_TEST(hasher.get_digest(small_container_span, 5U) == boost::crypt::state::success);
 }
 
 void test_user_container()