src/test/test-ratelimit.c

   1 /***
   2   This file is part of systemd
   3
   4   Copyright 2014 Ronny Chevalier
   5
   6   systemd is free software; you can redistribute it and/or modify it
   7   under the terms of the GNU Lesser General Public License as published by
   8   the Free Software Foundation; either version 2.1 of the License, or
   9   (at your option) any later version.
  10
  11   systemd is distributed in the hope that it will be useful, but
  12   WITHOUT ANY WARRANTY; without even the implied warranty of
  13   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  14   Lesser General Public License for more details.
  15
  16   You should have received a copy of the GNU Lesser General Public License
  17   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  18 ***/
  19
  20 #include <unistd.h>
  21
  22 #include "ratelimit.h"
  23 #include "time-util.h"
  24 #include "macro.h"
  25
  26 static void test_ratelimit_test(void) {
  27         int i;
  28         RATELIMIT_DEFINE(ratelimit, 1 * USEC_PER_SEC, 10);
  29
  30         for (i = 0; i < 10; i++) {
  31                 assert_se(ratelimit_test(&ratelimit));
  32         }
  33         assert_se(!ratelimit_test(&ratelimit));
  34         sleep(1);
  35         for (i = 0; i < 10; i++) {
  36                 assert_se(ratelimit_test(&ratelimit));
  37         }
  38
  39         RATELIMIT_INIT(ratelimit, 0, 10);
  40         for (i = 0; i < 10000; i++) {
  41                 assert_se(ratelimit_test(&ratelimit));
  42         }
  43 }
  44
  45 int main(int argc, char *argv[]) {
  46         test_ratelimit_test();
  47
  48         return 0;
  49 }