1
0
mirror of https://github.com/moparisthebest/curl synced 2024-11-05 09:05:04 -05:00
curl/tests/unit/unit1606.c
Dan Fandrich 8089dcfc5d tests: fixed OOM handling of unit tests to abort test
It's dangerous to continue to run the test when a memory alloc fails.
2017-04-30 10:55:20 +02:00

92 lines
2.6 KiB
C

/***************************************************************************
* _ _ ____ _
* Project ___| | | | _ \| |
* / __| | | | |_) | |
* | (__| |_| | _ <| |___
* \___|\___/|_| \_\_____|
*
* Copyright (C) 1998 - 2017, Daniel Stenberg, <daniel@haxx.se>, et al.
*
* This software is licensed as described in the file COPYING, which
* you should have received as part of this distribution. The terms
* are also available at https://curl.haxx.se/docs/copyright.html.
*
* You may opt to use, copy, modify, merge, publish, distribute and/or sell
* copies of the Software, and permit persons to whom the Software is
* furnished to do so, under the terms of the COPYING file.
*
* This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
* KIND, either express or implied.
*
***************************************************************************/
#include "curlcheck.h"
#include "speedcheck.h"
#include "urldata.h"
static CURLcode unit_setup(void)
{
return CURLE_OK;
}
static void unit_stop(void)
{
}
static int runawhile(struct Curl_easy *easy,
long time_limit,
long speed_limit,
curl_off_t speed,
int dec)
{
int counter = 1;
struct timeval now = {1, 0};
CURLcode result;
int finaltime;
curl_easy_setopt(easy, CURLOPT_LOW_SPEED_LIMIT, speed_limit);
curl_easy_setopt(easy, CURLOPT_LOW_SPEED_TIME, time_limit);
Curl_speedinit(easy);
do {
/* fake the current transfer speed */
easy->progress.current_speed = speed;
result = Curl_speedcheck(easy, now);
if(result)
break;
/* step the time */
now.tv_sec = ++counter;
speed -= dec;
} while(counter < 100);
finaltime = (int)(now.tv_sec - 1);
printf("%s\n", easy->state.buffer);
return finaltime;
}
UNITTEST_START
{
struct Curl_easy *easy = curl_easy_init();
abort_unless(easy, "out of memory");
fail_unless(runawhile(easy, 41, 41, 40, 0) == 41,
"wrong low speed timeout");
fail_unless(runawhile(easy, 21, 21, 20, 0) == 21,
"wrong low speed timeout");
fail_unless(runawhile(easy, 60, 60, 40, 0) == 60,
"wrong log speed timeout");
fail_unless(runawhile(easy, 50, 50, 40, 0) == 50,
"wrong log speed timeout");
fail_unless(runawhile(easy, 40, 40, 40, 0) == 99,
"should not time out");
fail_unless(runawhile(easy, 10, 50, 100, 2) == 36,
"bad timeout");
curl_easy_cleanup(easy);
return 0;
}
UNITTEST_STOP