Add unit tests for isc_buffer_copyregion()

Add some basic checks for isc_buffer_copyregion() to ensure it behaves
as expected for both fixed-size buffers and buffers which can be
automatically reallocated.  Adjust the list of headers included by
lib/isc/tests/buffer_test.c so that it matches what that test program
really uses.

diff --git a/lib/isc/tests/buffer_test.c b/lib/isc/tests/buffer_test.c
index 34145d7bccf73cbe4c84e920e4f50299d98eef7c..89009e0a9d12fc5481fa1a5e693bb250d64c9c5d 100644 (file)
--- a/lib/isc/tests/buffer_test.c
+++ b/lib/isc/tests/buffer_test.c
@@ -10,17 +10,20 @@
  */
 
 #include <config.h>
+
+#include <limits.h>
+#include <stdbool.h>
 #include <stdlib.h>
-#include <unistd.h>
-#include <fcntl.h>
+#include <string.h>
 
 #include <atf-c.h>
 
 #include "isctest.h"
 
 #include <isc/buffer.h>
-#include <isc/print.h>
+#include <isc/region.h>
 #include <isc/result.h>
+#include <isc/types.h>
 
 ATF_TC(isc_buffer_reserve);
 ATF_TC_HEAD(isc_buffer_reserve, tc) {
@@ -158,6 +161,51 @@ ATF_TC_BODY(isc_buffer_dynamic, tc) {
        isc_test_end();
 }
 
+ATF_TC(isc_buffer_copyregion);
+ATF_TC_HEAD(isc_buffer_copyregion, tc) {
+       atf_tc_set_md_var(tc, "descr", "copy a region into a buffer");
+}
+
+ATF_TC_BODY(isc_buffer_copyregion, tc) {
+       unsigned char data[] = { 0x11, 0x22, 0x33, 0x44 };
+       isc_buffer_t *b = NULL;
+       isc_result_t result;
+
+       isc_region_t r = {
+               .base = data,
+               .length = sizeof(data),
+       };
+
+       result = isc_test_begin(NULL, true, 0);
+       ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
+
+       result = isc_buffer_allocate(mctx, &b, sizeof(data));
+       ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
+
+       /*
+        * Fill originally allocated buffer space.
+        */
+       result = isc_buffer_copyregion(b, &r);
+       ATF_CHECK_EQ(result, ISC_R_SUCCESS);
+
+       /*
+        * Appending more data to the buffer should fail.
+        */
+       result = isc_buffer_copyregion(b, &r);
+       ATF_CHECK_EQ(result, ISC_R_NOSPACE);
+
+       /*
+        * Enable auto reallocation and retry.  Appending should now succeed.
+        */
+       isc_buffer_setautorealloc(b, true);
+       result = isc_buffer_copyregion(b, &r);
+       ATF_CHECK_EQ(result, ISC_R_SUCCESS);
+
+       isc_buffer_free(&b);
+
+       isc_test_end();
+}
+
 ATF_TC(isc_buffer_printf);
 ATF_TC_HEAD(isc_buffer_printf, tc) {
        atf_tc_set_md_var(tc, "descr", "printf() into a buffer");
@@ -278,6 +326,7 @@ ATF_TC_BODY(isc_buffer_printf, tc) {
 ATF_TP_ADD_TCS(tp) {
        ATF_TP_ADD_TC(tp, isc_buffer_reserve);
        ATF_TP_ADD_TC(tp, isc_buffer_dynamic);
+       ATF_TP_ADD_TC(tp, isc_buffer_copyregion);
        ATF_TP_ADD_TC(tp, isc_buffer_printf);
        return (atf_no_error());
 }