lib/tests/test_utils.c

   1 /*****************************************************************************
   2 *
   3 * This program is free software: you can redistribute it and/or modify
   4 * it under the terms of the GNU General Public License as published by
   5 * the Free Software Foundation, either version 3 of the License, or
   6 * (at your option) any later version.
   7 *
   8 * This program is distributed in the hope that it will be useful,
   9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11 * GNU General Public License for more details.
  12 *
  13 * You should have received a copy of the GNU General Public License
  14 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  15 *
  16 *
  17 *****************************************************************************/
  18
  19 #include "common.h"
  20 #include "utils_base.h"
  21
  22 #include "tap.h"
  23
  24 int
  25 main (int argc, char **argv)
  26 {
  27         range   *range;
  28         double  temp;
  29         thresholds *thresholds = NULL;
  30         int     rc;
  31
  32         plan_tests(81+23);
  33
  34         range = parse_range_string("6");
  35         ok( range != NULL, "'6' is valid range");
  36         ok( range->start == 0, "Start correct");
  37         ok( range->start_infinity == FALSE, "Not using negative infinity");
  38         ok( range->end == 6, "End correct");
  39         ok( range->end_infinity == FALSE, "Not using infinity");
  40         free(range);
  41
  42         range = parse_range_string("1:12%%");
  43         ok( range != NULL, "'1:12%%' is valid - percentages are ignored");
  44         ok( range->start == 1, "Start correct");
  45         ok( range->start_infinity == FALSE, "Not using negative infinity");
  46         ok( range->end == 12, "End correct");
  47         ok( range->end_infinity == FALSE, "Not using infinity");
  48         free(range);
  49
  50         range = parse_range_string("-7:23");
  51         ok( range != NULL, "'-7:23' is valid range");
  52         ok( range->start == -7, "Start correct");
  53         ok( range->start_infinity == FALSE, "Not using negative infinity");
  54         ok( range->end == 23, "End correct");
  55         ok( range->end_infinity == FALSE, "Not using infinity");
  56         free(range);
  57
  58         range = parse_range_string(":5.75");
  59         ok( range != NULL, "':5.75' is valid range");
  60         ok( range->start == 0, "Start correct");
  61         ok( range->start_infinity == FALSE, "Not using negative infinity");
  62         ok( range->end == 5.75, "End correct");
  63         ok( range->end_infinity == FALSE, "Not using infinity");
  64         free(range);
  65
  66         range = parse_range_string("~:-95.99");
  67         ok( range != NULL, "~:-95.99' is valid range");
  68         ok( range->start_infinity == TRUE, "Using negative infinity");
  69         ok( range->end == -95.99, "End correct (with rounding errors)");
  70         ok( range->end_infinity == FALSE, "Not using infinity");
  71         free(range);
  72
  73         range = parse_range_string("12345678901234567890:");
  74         temp = atof("12345678901234567890");            /* Can't just use this because number too large */
  75         ok( range != NULL, "'12345678901234567890:' is valid range");
  76         ok( range->start == temp, "Start correct");
  77         ok( range->start_infinity == FALSE, "Not using negative infinity");
  78         ok( range->end_infinity == TRUE, "Using infinity");
  79         /* Cannot do a "-1" on temp, as it appears to be same value */
  80         ok( check_range(temp/1.1, range) == TRUE, "12345678901234567890/1.1 - alert");
  81         ok( check_range(temp, range) == FALSE, "12345678901234567890 - no alert");
  82         ok( check_range(temp*2, range) == FALSE, "12345678901234567890*2 - no alert");
  83         free(range);
  84
  85         range = parse_range_string("~:0");
  86         ok( range != NULL, "'~:0' is valid range");
  87         ok( range->start_infinity == TRUE, "Using negative infinity");
  88         ok( range->end == 0, "End correct");
  89         ok( range->end_infinity == FALSE, "Not using infinity");
  90         ok( range->alert_on == OUTSIDE, "Will alert on outside of this range");
  91         ok( check_range(0.5, range) == TRUE, "0.5 - alert");
  92         ok( check_range(-10, range) == FALSE, "-10 - no alert");
  93         ok( check_range(0, range)   == FALSE, "0 - no alert");
  94         free(range);
  95
  96         range = parse_range_string("@0:657.8210567");
  97         ok( range != 0, "@0:657.8210567' is a valid range");
  98         ok( range->start == 0, "Start correct");
  99         ok( range->start_infinity == FALSE, "Not using negative infinity");
 100         ok( range->end == 657.8210567, "End correct");
 101         ok( range->end_infinity == FALSE, "Not using infinity");
 102         ok( range->alert_on == INSIDE, "Will alert on inside of this range" );
 103         ok( check_range(32.88, range) == TRUE, "32.88 - alert");
 104         ok( check_range(-2, range)    == FALSE, "-2 - no alert");
 105         ok( check_range(657.8210567, range) == TRUE, "657.8210567 - alert");
 106         ok( check_range(0, range)     == TRUE, "0 - alert");
 107         free(range);
 108
 109         range = parse_range_string("1:1");
 110         ok( range != NULL, "'1:1' is a valid range");
 111         ok( range->start == 1, "Start correct");
 112         ok( range->start_infinity == FALSE, "Not using negative infinity");
 113         ok( range->end == 1, "End correct");
 114         ok( range->end_infinity == FALSE, "Not using infinity");
 115         ok( check_range(0.5, range) == TRUE, "0.5 - alert");
 116         ok( check_range(1, range) == FALSE, "1 - no alert");
 117         ok( check_range(5.2, range) == TRUE, "5.2 - alert");
 118         free(range);
 119
 120         range = parse_range_string("2:1");
 121         ok( range == NULL, "'2:1' rejected");
 122
 123         rc = _set_thresholds(&thresholds, NULL, NULL);
 124         ok( rc == 0, "Thresholds (NULL, NULL) set");
 125         ok( thresholds->warning == NULL, "Warning not set");
 126         ok( thresholds->critical == NULL, "Critical not set");
 127
 128         rc = _set_thresholds(&thresholds, NULL, "80");
 129         ok( rc == 0, "Thresholds (NULL, '80') set");
 130         ok( thresholds->warning == NULL, "Warning not set");
 131         ok( thresholds->critical->end == 80, "Critical set correctly");
 132
 133         rc = _set_thresholds(&thresholds, "5:33", NULL);
 134         ok( rc == 0, "Thresholds ('5:33', NULL) set");
 135         ok( thresholds->warning->start == 5, "Warning start set");
 136         ok( thresholds->warning->end == 33, "Warning end set");
 137         ok( thresholds->critical == NULL, "Critical not set");
 138
 139         rc = _set_thresholds(&thresholds, "30", "60");
 140         ok( rc == 0, "Thresholds ('30', '60') set");
 141         ok( thresholds->warning->end == 30, "Warning set correctly");
 142         ok( thresholds->critical->end == 60, "Critical set correctly");
 143         ok( get_status(15.3, thresholds) == STATE_OK, "15.3 - ok");
 144         ok( get_status(30.0001, thresholds) == STATE_WARNING, "30.0001 - warning");
 145         ok( get_status(69, thresholds) == STATE_CRITICAL, "69 - critical");
 146
 147         char *test;
 148         test = np_escaped_string("bob\\n");
 149         ok( strcmp(test, "bob\n") == 0, "bob\\n ok");
 150         free(test);
 151
 152         test = np_escaped_string("rhuba\\rb");
 153         ok( strcmp(test, "rhuba\rb") == 0, "rhuba\\rb okay");
 154         free(test);
 155
 156         test = np_escaped_string("ba\\nge\\r");
 157         ok( strcmp(test, "ba\nge\r") == 0, "ba\\nge\\r okay");
 158         free(test);
 159
 160         test = np_escaped_string("\\rabbi\\t");
 161         ok( strcmp(test, "\rabbi\t") == 0, "\\rabbi\\t okay");
 162         free(test);
 163
 164         test = np_escaped_string("and\\\\or");
 165         ok( strcmp(test, "and\\or") == 0, "and\\\\or okay");
 166         free(test);
 167
 168         test = np_escaped_string("bo\\gus");
 169         ok( strcmp(test, "bogus") == 0, "bo\\gus okay");
 170         free(test);
 171
 172         test = np_escaped_string("everything");
 173         ok( strcmp(test, "everything") == 0, "everything okay");
 174
 175         /* np_extract_ntpvar tests (23) */
 176         test=np_extract_ntpvar("foo=bar, bar=foo, foobar=barfoo\n", "foo");
 177         ok(test && !strcmp(test, "bar"), "1st test as expected");
 178         free(test);
 179
 180         test=np_extract_ntpvar("foo=bar,bar=foo,foobar=barfoo\n", "bar");
 181         ok(test && !strcmp(test, "foo"), "2nd test as expected");
 182         free(test);
 183
 184         test=np_extract_ntpvar("foo=bar, bar=foo, foobar=barfoo\n", "foobar");
 185         ok(test && !strcmp(test, "barfoo"), "3rd test as expected");
 186         free(test);
 187
 188         test=np_extract_ntpvar("foo=bar\n", "foo");
 189         ok(test && !strcmp(test, "bar"), "Single test as expected");
 190         free(test);
 191
 192         test=np_extract_ntpvar("foo=bar, bar=foo, foobar=barfooi\n", "abcd");
 193         ok(!test, "Key not found 1");
 194
 195         test=np_extract_ntpvar("foo=bar\n", "abcd");
 196         ok(!test, "Key not found 2");
 197
 198         test=np_extract_ntpvar("foo=bar=foobar", "foo");
 199         ok(test && !strcmp(test, "bar=foobar"), "Strange string 1");
 200         free(test);
 201
 202         test=np_extract_ntpvar("foo", "foo");
 203         ok(!test, "Malformed string 1");
 204
 205         test=np_extract_ntpvar("foo,", "foo");
 206         ok(!test, "Malformed string 2");
 207
 208         test=np_extract_ntpvar("foo=", "foo");
 209         ok(!test, "Malformed string 3");
 210
 211         test=np_extract_ntpvar("foo=,bar=foo", "foo");
 212         ok(!test, "Malformed string 4");
 213
 214         test=np_extract_ntpvar(",foo", "foo");
 215         ok(!test, "Malformed string 5");
 216
 217         test=np_extract_ntpvar("=foo", "foo");
 218         ok(!test, "Malformed string 6");
 219
 220         test=np_extract_ntpvar("=foo,", "foo");
 221         ok(!test, "Malformed string 7");
 222
 223         test=np_extract_ntpvar(",,,", "foo");
 224         ok(!test, "Malformed string 8");
 225
 226         test=np_extract_ntpvar("===", "foo");
 227         ok(!test, "Malformed string 9");
 228
 229         test=np_extract_ntpvar(",=,=,", "foo");
 230         ok(!test, "Malformed string 10");
 231
 232         test=np_extract_ntpvar("=,=,=", "foo");
 233         ok(!test, "Malformed string 11");
 234
 235         test=np_extract_ntpvar("  foo=bar  ,\n bar=foo\n , foobar=barfoo  \n  ", "foo");
 236         ok(test && !strcmp(test, "bar"), "Random spaces and newlines 1");
 237         free(test);
 238
 239         test=np_extract_ntpvar("  foo=bar  ,\n bar=foo\n , foobar=barfoo  \n  ", "bar");
 240         ok(test && !strcmp(test, "foo"), "Random spaces and newlines 2");
 241         free(test);
 242
 243         test=np_extract_ntpvar("  foo=bar  ,\n bar=foo\n , foobar=barfoo  \n  ", "foobar");
 244         ok(test && !strcmp(test, "barfoo"), "Random spaces and newlines 3");
 245         free(test);
 246
 247         test=np_extract_ntpvar("  foo=bar  ,\n bar\n \n= \n foo\n , foobar=barfoo  \n  ", "bar");
 248         ok(test && !strcmp(test, "foo"), "Random spaces and newlines 4");
 249         free(test);
 250
 251         test=np_extract_ntpvar("", "foo");
 252         ok(!test, "Empty string return NULL");
 253
 254         return exit_status();
 255 }