diff options
| -rw-r--r-- | .gitignore | 1 | ||||
| -rw-r--r-- | Makefile.am | 9 | ||||
| -rw-r--r-- | src/test/test-ratelimit.c | 49 |
3 files changed, 58 insertions, 1 deletions
diff --git a/.gitignore b/.gitignore index e42fbcbbab..36c67e2e2b 100644 --- a/.gitignore +++ b/.gitignore @@ -190,6 +190,7 @@ /test-ns /test-path-util /test-prioq +/test-ratelimit /test-replace-var /test-resolve /test-ring diff --git a/Makefile.am b/Makefile.am index 20b6082531..0e4edf771b 100644 --- a/Makefile.am +++ b/Makefile.am @@ -1247,7 +1247,8 @@ tests += \ test-fdset \ test-conf-files \ test-capability \ - test-async + test-async \ + test-ratelimit EXTRA_DIST += \ test/sched_idle_bad.service \ @@ -1370,6 +1371,12 @@ test_fdset_SOURCES = \ test_fdset_LDADD = \ libsystemd-core.la +test_ratelimit_SOURCES = \ + src/test/test-ratelimit.c + +test_ratelimit_LDADD = \ + libsystemd-shared.la + test_util_SOURCES = \ src/test/test-util.c diff --git a/src/test/test-ratelimit.c b/src/test/test-ratelimit.c new file mode 100644 index 0000000000..b7f6dfe246 --- /dev/null +++ b/src/test/test-ratelimit.c @@ -0,0 +1,49 @@ +/*** + This file is part of systemd + + Copyright 2014 Ronny Chevalier + + systemd is free software; you can redistribute it and/or modify it + under the terms of the GNU Lesser General Public License as published by + the Free Software Foundation; either version 2.1 of the License, or + (at your option) any later version. + + systemd is distributed in the hope that it will be useful, but + WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + Lesser General Public License for more details. + + You should have received a copy of the GNU Lesser General Public License + along with systemd; If not, see <http://www.gnu.org/licenses/>. +***/ + +#include <unistd.h> + +#include "ratelimit.h" +#include "time-util.h" +#include "macro.h" + +static void test_ratelimit_test(void) { + int i; + RATELIMIT_DEFINE(ratelimit, 1 * USEC_PER_SEC, 10); + + for (i = 0; i < 10; i++) { + assert_se(ratelimit_test(&ratelimit)); + } + assert_se(!ratelimit_test(&ratelimit)); + sleep(1); + for (i = 0; i < 10; i++) { + assert_se(ratelimit_test(&ratelimit)); + } + + RATELIMIT_INIT(ratelimit, 0, 10); + for (i = 0; i < 10000; i++) { + assert_se(ratelimit_test(&ratelimit)); + } +} + +int main(int argc, char *argv[]) { + test_ratelimit_test(); + + return 0; +} |