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