# HG changeset patch
# User Mike Becker <universe@uap-core.de>
# Date 1399366614 -7200
# Node ID aed60ba37acf6851146a3a3bdcde09dc2b9d2dfc
# Parent  350a0e3898bd9e60997b758214303decc3b147b7
ucx_buffer_extract tests

diff -r 350a0e3898bd -r aed60ba37acf test/buffer_tests.c
--- a/test/buffer_tests.c	Mon May 05 15:56:39 2014 +0200
+++ b/test/buffer_tests.c	Tue May 06 10:56:54 2014 +0200
@@ -503,25 +503,57 @@
 
     UcxBuffer *src = ucx_buffer_new(buffer, 16, UCX_BUFFER_AUTOFREE);
     src->size = 15;
-    UcxBuffer *dst = ucx_buffer_extract(src, 5, 5, UCX_BUFFER_DEFAULT);
+    UcxBuffer *dst = ucx_buffer_extract(src, 5, 5, UCX_BUFFER_AUTOEXTEND);
 
     UCX_TEST_BEGIN
     UCX_TEST_ASSERT(dst != NULL, "ucx_buffer_extract returned NULL");
-            
-    UCX_TEST_ASSERT((dst->flags & UCX_BUFFER_AUTOFREE) == UCX_BUFFER_AUTOFREE,
+    UCX_TEST_ASSERT(dst->flags == (UCX_BUFFER_AUTOEXTEND | UCX_BUFFER_AUTOFREE),
             "autofree flag shall be enforced");
     UCX_TEST_ASSERT(dst->size == 5, "wrong size for new buffer");
+    UCX_TEST_ASSERT(dst->capacity == 5, "wrong capacity for new buffer");
+    UCX_TEST_ASSERT(dst->pos == 0, "wrong position for new buffer");
     char rb[5];
     ucx_buffer_read(rb, 1, 5, dst);
     UCX_TEST_ASSERT(memcmp(rb, "is a ", 5) == 0,
             "new buffer has incorrect content");
 
-    UCX_TEST_ASSERT(ucx_buffer_extract(dst, 3, 3, UCX_BUFFER_DEFAULT) == NULL,
-            "extract shall fail on invalid bounds");
+    UCX_TEST_END
+
+    ucx_buffer_free(dst);
+    ucx_buffer_free(src);
+}
+
+UCX_TEST(test_ucx_buffer_extract_oob) {
+    char *buffer = (char*) malloc(16);
+    strcpy(buffer, "this is a test!");
+
+    UcxBuffer *src = ucx_buffer_new(buffer, 16, UCX_BUFFER_AUTOFREE);
+    UCX_TEST_BEGIN
+    
+    UCX_TEST_ASSERT(ucx_buffer_extract(src, 5, 0, UCX_BUFFER_DEFAULT) == NULL,
+            "extract shall fail on zero length");
+    UCX_TEST_ASSERT(ucx_buffer_extract(src, 10, 10, UCX_BUFFER_DEFAULT) == NULL,
+            "extract shall fail on invalid bounds (size exceeds limits)");
+    UCX_TEST_ASSERT(ucx_buffer_extract(src, 20, -7, UCX_BUFFER_DEFAULT) == NULL,
+            "extract shall fail on invalid bounds (start exceeds limits)");
 
     UCX_TEST_END
 
-    ucx_buffer_free(dst);
+    ucx_buffer_free(src);
+}
+
+UCX_TEST(test_ucx_buffer_extract_overflow) {
+    char *buffer = (char*) malloc(16);
+    strcpy(buffer, "this is a test!");
+
+    UcxBuffer *src = ucx_buffer_new(buffer, 16, UCX_BUFFER_AUTOFREE);
+    UCX_TEST_BEGIN
+    
+    UCX_TEST_ASSERT(ucx_buffer_extract(src, 5, (size_t)-4,
+        UCX_BUFFER_DEFAULT) == NULL, "extract shall fail on integer overflow");
+
+    UCX_TEST_END
+
     ucx_buffer_free(src);
 }
 
diff -r 350a0e3898bd -r aed60ba37acf test/buffer_tests.h
--- a/test/buffer_tests.h	Mon May 05 15:56:39 2014 +0200
+++ b/test/buffer_tests.h	Tue May 06 10:56:54 2014 +0200
@@ -53,11 +53,12 @@
 UCX_TEST(test_ucx_buffer_getc);
 UCX_TEST(test_ucx_buffer_read);
 UCX_TEST(test_ucx_buffer_read_oob);
+UCX_TEST(test_ucx_buffer_extract);
+UCX_TEST(test_ucx_buffer_extract_oob);
+UCX_TEST(test_ucx_buffer_extract_overflow);
 
 UCX_TEST(test_ucx_buffer_write);
 UCX_TEST(test_ucx_buffer_write_ax);
-
-UCX_TEST(test_ucx_buffer_extract);
 UCX_TEST(test_ucx_stream_copy);
 
 #ifdef	__cplusplus
diff -r 350a0e3898bd -r aed60ba37acf test/main.c
--- a/test/main.c	Mon May 05 15:56:39 2014 +0200
+++ b/test/main.c	Tue May 06 10:56:54 2014 +0200
@@ -194,9 +194,11 @@
         ucx_test_register(suite, test_ucx_buffer_getc);
         ucx_test_register(suite, test_ucx_buffer_read);
         ucx_test_register(suite, test_ucx_buffer_read_oob);
+        ucx_test_register(suite, test_ucx_buffer_extract);
+        ucx_test_register(suite, test_ucx_buffer_extract_oob);
+        ucx_test_register(suite, test_ucx_buffer_extract_overflow);
         ucx_test_register(suite, test_ucx_buffer_write);
         ucx_test_register(suite, test_ucx_buffer_write_ax);
-        ucx_test_register(suite, test_ucx_buffer_extract);
         ucx_test_register(suite, test_ucx_stream_copy);
         
         /* Utils Tests*/