Adding unit test for rate limiting