diff --git a/src/rate_limit.c b/src/rate_limit.c index f7c33b4..3ed94bf 100644 --- a/src/rate_limit.c +++ b/src/rate_limit.c @@ -58,24 +58,3 @@ int uc_ratelimit_allow(struct RateLimit *r, long current_ts) return allowed; } -#if 0 -int main() -{ - int i; - struct RateLimit r; - time_t start; - - uc_ratelimit_init(&r, 10, 60, time(NULL)); - - for(i = 0; i < 120; i++) { - int k; - long allowed = 0; - sleep(1); - for (k = 0; k < 500; k++) - allowed += uc_ratelimit_allow(&r, time(NULL)); - printf("%ld of 500 can pass\n", allowed); - } - - return 0; -} -#endif diff --git a/test/rate_limit.c b/test/rate_limit.c new file mode 100644 index 0000000..1b5c714 --- /dev/null +++ b/test/rate_limit.c @@ -0,0 +1,25 @@ +#include +#include +#include +#include "ucore/rate_limit.h" + + +int main() +{ + int i; + struct RateLimit r; + time_t start; + + uc_ratelimit_init(&r, 10, 60, time(NULL)); + + for(i = 0; i < 120; i++) { + int k; + long allowed = 0; + sleep(1); + for (k = 0; k < 500; k++) + allowed += uc_ratelimit_allow(&r, time(NULL)); + printf("%ld of 500 can pass\n", allowed); + } + + return 0; +}