aboutsummaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests')
-rw-r--r--tests/unit/curlcheck.h7
-rw-r--r--tests/unit/unit1300.c5
-rw-r--r--tests/unit/unit1301.c2
-rw-r--r--tests/unit/unit1302.c29
-rw-r--r--tests/unit/unit1303.c6
5 files changed, 32 insertions, 17 deletions
diff --git a/tests/unit/curlcheck.h b/tests/unit/curlcheck.h
index 57babe5c7..9d738430c 100644
--- a/tests/unit/curlcheck.h
+++ b/tests/unit/curlcheck.h
@@ -46,12 +46,13 @@ extern int unitfail;
int test(char *unused) \
{ \
(void)unused; \
- unit_setup(); \
- {
+ if (unit_setup()) { \
+ fail("unit_setup() failure"); \
+ } else {
#define UNITTEST_STOP \
+ unit_stop(); \
} \
- unit_stop(); \
return unitfail; \
}
diff --git a/tests/unit/unit1300.c b/tests/unit/unit1300.c
index 92c0a7ac9..093ef850a 100644
--- a/tests/unit/unit1300.c
+++ b/tests/unit/unit1300.c
@@ -14,9 +14,12 @@ static void test_curl_llist_dtor(void *key , void *value)
(void)value;
}
-static void unit_setup( void )
+static CURLcode unit_setup( void )
{
llist = Curl_llist_alloc( test_curl_llist_dtor );
+ if (!llist)
+ return CURLE_OUT_OF_MEMORY;
+ return CURLE_OK;
}
static void unit_stop( void )
diff --git a/tests/unit/unit1301.c b/tests/unit/unit1301.c
index 6f01e2dae..7de6f069a 100644
--- a/tests/unit/unit1301.c
+++ b/tests/unit/unit1301.c
@@ -5,7 +5,7 @@
#include "strequal.h"
#include "curlcheck.h"
-static void unit_setup( void ) {}
+static CURLcode unit_setup( void ) {return CURLE_OK;}
static void unit_stop( void ) {}
UNITTEST_START
diff --git a/tests/unit/unit1302.c b/tests/unit/unit1302.c
index d300a14ee..1a6bf4e54 100644
--- a/tests/unit/unit1302.c
+++ b/tests/unit/unit1302.c
@@ -3,16 +3,21 @@
#include "setup.h"
#include "urldata.h"
+#include "url.h" /* for Curl_safefree */
#include "curl_base64.h"
#include "curlcheck.h"
#include "memdebug.h" /* LAST include file */
static struct SessionHandle *data;
-static void unit_setup( void )
+static CURLcode unit_setup( void )
{
data = curl_easy_init();
+ if (!data)
+ return CURLE_OUT_OF_MEMORY;
+ return CURLE_OK;
}
+
static void unit_stop( void )
{
curl_easy_cleanup(data);
@@ -27,56 +32,58 @@ size_t rc;
rc = Curl_base64_encode(data, "i", 1, &output);
fail_unless( rc == 4 , "return code should be 4" );
verify_memory( output, "aQ==", 4);
-free(output);
+Curl_safefree(output);
rc = Curl_base64_encode(data, "ii", 2, &output);
fail_unless( rc == 4 , "return code should be 4" );
verify_memory( output, "aWk=", 4);
-free(output);
+Curl_safefree(output);
rc = Curl_base64_encode(data, "iii", 3, &output);
fail_unless( rc == 4 , "return code should be 4" );
verify_memory( output, "aWlp", 4);
-free(output);
+Curl_safefree(output);
rc = Curl_base64_encode(data, "iiii", 4, &output);
fail_unless( rc == 8 , "return code should be 8" );
verify_memory( output, "aWlpaQ==", 8);
-free(output);
+Curl_safefree(output);
/* 0 length makes it do strlen() */
rc = Curl_base64_encode(data, "iiii", 0, &output);
fail_unless( rc == 8 , "return code should be 8" );
verify_memory( output, "aWlpaQ==", 8);
-free(output);
+Curl_safefree(output);
rc = Curl_base64_decode("aWlpaQ==", &decoded);
fail_unless(rc == 4, "return code should be 4");
verify_memory(decoded, "iiii", 4);
-free(decoded);
+Curl_safefree(decoded);
rc = Curl_base64_decode("aWlp", &decoded);
fail_unless(rc == 3, "return code should be 3");
verify_memory(decoded, "iii", 3);
-free(decoded);
+Curl_safefree(decoded);
rc = Curl_base64_decode("aWk=", &decoded);
fail_unless(rc == 2, "return code should be 2");
verify_memory(decoded, "ii", 2);
-free(decoded);
+Curl_safefree(decoded);
rc = Curl_base64_decode("aQ==", &decoded);
fail_unless(rc == 1, "return code should be 1");
verify_memory(decoded, "i", 2);
-free(decoded);
+Curl_safefree(decoded);
/* this is an illegal input */
+decoded = NULL;
rc = Curl_base64_decode("aQ", &decoded);
fail_unless(rc == 0, "return code should be 0");
+fail_if(decoded, "returned pointer should be NULL");
/* this is garbage input that libcurl decodes as far as possible */
rc = Curl_base64_decode("a\x1f==", &decoded);
fail_unless(rc == 1, "return code should be 1");
-free(decoded);
+Curl_safefree(decoded);
UNITTEST_STOP
diff --git a/tests/unit/unit1303.c b/tests/unit/unit1303.c
index 01fdda1f9..cabee85c2 100644
--- a/tests/unit/unit1303.c
+++ b/tests/unit/unit1303.c
@@ -9,10 +9,14 @@
static struct SessionHandle *data;
-static void unit_setup( void )
+static CURLcode unit_setup( void )
{
data = curl_easy_init();
+ if (!data)
+ return CURLE_OUT_OF_MEMORY;
+ return CURLE_OK;
}
+
static void unit_stop( void )
{
curl_easy_cleanup(data);