Add rate limit tests