def test_the_instance_should_be_able_to_unregister_multiple_realms(): rr = RespectfulRequester() realm_tuples = [ ["TEST123", 100, 300], ["TEST234", 200, 600], ["TEST345", 300, 900], ] rr.register_realms(realm_tuples) request_func = lambda: requests.get("http://google.com") rr._perform_request(request_func, realms=["TEST123", "TEST234", "TEST345"]) rr.unregister_realms(["TEST123", "TEST234", "TEST345"]) assert rr.redis.get(rr._realm_redis_key("TEST123")) is None assert not rr.redis.sismember("%s:REALMS" % rr.redis_prefix, "TEST123") assert not len( rr.redis.keys("%s:REQUESTS:%s:*" % (rr.redis_prefix, "TEST123"))) assert rr.redis.get(rr._realm_redis_key("TEST234")) is None assert not rr.redis.sismember("%s:REALMS" % rr.redis_prefix, "TEST234") assert not len( rr.redis.keys("%s:REQUESTS:%s:*" % (rr.redis_prefix, "TEST234"))) assert rr.redis.get(rr._realm_redis_key("TEST345")) is None assert not rr.redis.sismember("%s:REALMS" % rr.redis_prefix, "TEST345") assert not len( rr.redis.keys("%s:REQUESTS:%s:*" % (rr.redis_prefix, "TEST345")))
def test_the_instance_should_be_able_to_generate_a_redis_key_when_provided_with_a_realm(): rr = RespectfulRequester() assert rr._realm_redis_key("TEST") == "%s:REALMS:TEST" % rr.redis_prefix assert rr._realm_redis_key("TEST2") == "%s:REALMS:TEST2" % rr.redis_prefix assert rr._realm_redis_key("TEST SPACED") == "%s:REALMS:TEST SPACED" % rr.redis_prefix assert rr._realm_redis_key("TEST ÉÉÉ") == "%s:REALMS:TEST ÉÉÉ" % rr.redis_prefix
def test_the_instance_should_be_able_to_unregister_multiple_realms(): rr = RespectfulRequester() realm_tuples = [ ["TEST123", 100, 300], ["TEST234", 200, 600], ["TEST345", 300, 900], ] rr.register_realms(realm_tuples) request_func = lambda: requests.get("http://google.com") rr._perform_request(request_func, realms=["TEST123", "TEST234", "TEST345"]) rr.unregister_realms(["TEST123", "TEST234", "TEST345"]) assert rr.redis.get(rr._realm_redis_key("TEST123")) is None assert not rr.redis.sismember("%s:REALMS" % rr.redis_prefix, "TEST123") assert not len(rr.redis.keys("%s:REQUESTS:%s:*" % (rr.redis_prefix, "TEST123"))) assert rr.redis.get(rr._realm_redis_key("TEST234")) is None assert not rr.redis.sismember("%s:REALMS" % rr.redis_prefix, "TEST234") assert not len(rr.redis.keys("%s:REQUESTS:%s:*" % (rr.redis_prefix, "TEST234"))) assert rr.redis.get(rr._realm_redis_key("TEST345")) is None assert not rr.redis.sismember("%s:REALMS" % rr.redis_prefix, "TEST345") assert not len(rr.redis.keys("%s:REQUESTS:%s:*" % (rr.redis_prefix, "TEST345")))
def test_the_instance_should_be_able_to_unregister_a_realm(): rr = RespectfulRequester() rr.register_realm("TEST123", max_requests=100, timespan=300) request_func = lambda: requests.get("http://google.com") rr._perform_request(request_func, "TEST123") rr.unregister_realm("TEST123") assert rr.redis.get(rr._realm_redis_key("TEST123")) is None assert not rr.redis.sismember("%s:REALMS" % rr.redis_prefix, "TEST123") assert not len(rr.redis.keys("%s:REQUESTS:%s:*" % (rr.redis_prefix, "TEST123")))