Modified overrides tests to adjust to overrides being applicable to H…

…TTPRoutes now

Signed-off-by: Martin Hesko <[email protected]>

testsuite/tests/singlecluster/overrides/test_basic_auth.py

@@ -3,17 +3,15 @@
 import pytest
 
 from testsuite.httpx.auth import HttpxOidcClientAuth
-from testsuite.kuadrant.policy.authorization.auth_policy import AuthPolicy
 
 pytestmark = [pytest.mark.kuadrant_only]
 
 
 @pytest.fixture(scope="module")
-def authorization(route, gateway, blame, cluster, label, oidc_provider):  # pylint: disable=unused-argument
-    """Add oidc identity to overrides block of gateway-attached AuthPolicy"""
-    auth_policy = AuthPolicy.create_instance(cluster, blame("authz"), gateway, labels={"testRun": label})
-    auth_policy.overrides.identity.add_oidc("override", oidc_provider.well_known["issuer"])
-    return auth_policy
+def authorization(authorization, oidc_provider):
+    """Add oidc identity to defaults block of AuthPolicy"""
+    authorization.overrides.identity.add_oidc("override", oidc_provider.well_known["issuer"])
+    return authorization
 
 
 @pytest.fixture(scope="module")
@@ -28,12 +26,12 @@ def rate_limit():
     return None
 
 
+@pytest.mark.parametrize("authorization", ["route", "gateway"], indirect=True)
 def test_basic_auth(route, authorization, client, auth):
     """Test if rules inside overrides block of Gateway's AuthPolicy are inherited by the HTTPRoute
     and enforced like any other normal rule"""
     route.refresh()
     assert route.is_affected_by(authorization)
 
-    response = client.get("/get")
-    assert response.status_code == 401
+    assert client.get("/get").status_code == 401
     assert client.get("/get", auth=auth).status_code == 200  # assert that AuthPolicy is enforced

testsuite/tests/singlecluster/overrides/test_basic_rate_limit.py

@@ -1,12 +1,12 @@
-"""Test basic enforcement of the rules inside the 'overrides' block of the RateLimitPolicy assigned to a Gateway"""
+"""Test enforcement of the rules inside the 'overrides' block of the RateLimitPolicy assigned to a Gateway/HTTPRoute"""
 
 import pytest
 
 from testsuite.kuadrant.policy.rate_limit import Limit, RateLimitPolicy
 
 pytestmark = [pytest.mark.kuadrant_only, pytest.mark.limitador]
 
-GATEWAY_LIMIT = Limit(3, "5s")
+OVERRIDE_LIMIT = Limit(3, "5s")
 ROUTE_LIMIT = Limit(2, "5s")
 
 
@@ -16,33 +16,34 @@ def authorization():
     return None
 
 
-@pytest.fixture(scope="module")
-def rate_limit_gw(request, cluster, blame, module_label, gateway):
-    """Add a RateLimitPolicy to the Gateway with an overrides block to override the Route-level policy."""
-    rate_limit_gateway = RateLimitPolicy.create_instance(
-        cluster, blame("limit-gateway"), gateway, labels={"testRun": module_label}
+@pytest.fixture(scope="function")
+def rate_limit_route(request, cluster, blame, module_label, route):
+    """Add a RateLimitPolicy to the HTTPRoute with a basic limit to be overriden."""
+    rate_limit_route = RateLimitPolicy.create_instance(
+        cluster, blame("limit-route"), route, labels={"testRun": module_label}
     )
-    rate_limit_gateway.overrides.add_limit("basic", [GATEWAY_LIMIT])
-    request.addfinalizer(rate_limit_gateway.delete)
-    rate_limit_gateway.commit()
-    rate_limit_gateway.wait_for_ready()
-    return rate_limit_gateway
+    rate_limit_route.add_limit("basic", [ROUTE_LIMIT])
+    request.addfinalizer(rate_limit_route.delete)
+    rate_limit_route.commit()
+    rate_limit_route.wait_for_accepted()
+    return rate_limit_route
 
 
 @pytest.fixture(scope="module")
 def rate_limit(rate_limit):
-    """Add basic requests limit to RateLimitPolicy"""
-    rate_limit.add_limit("basic", [ROUTE_LIMIT])
+    """Add an override to RateLimitPolicy"""
+    rate_limit.overrides.add_limit("override-limit", [OVERRIDE_LIMIT])
     return rate_limit
 
 
-def test_basic_rate_limit(rate_limit, rate_limit_gw, route, client):
-    """Test if rules inside overrides block of Gateway's RateLimitPolicy are inherited by the HTTPRoute
-    and enforced like any other normal rule"""
+@pytest.mark.parametrize("rate_limit", ["route", "gateway"], indirect=True)
+def test_basic_rate_limit(rate_limit, rate_limit_route, route, client):
+    """Test if rules inside overrides block of Gateway/HTTPRoute RateLimitPolicy are inherited by the HTTPRoute
+    and override the rate limit targeting the route."""
     route.refresh()
     assert route.is_affected_by(rate_limit)
-    rate_limit_gw.wait_for_full_enforced()
+    assert route.is_affected_by(rate_limit_route)
 
-    responses = client.get_many("/get", GATEWAY_LIMIT.limit)
+    responses = client.get_many("/get", OVERRIDE_LIMIT.limit)
     responses.assert_all(status_code=200)
     assert client.get("/get").status_code == 429  # assert that RateLimitPolicy is enforced