test: add logout arguments to satosa/frontends/test_saml2 to fix tests

Author: sebulibah

tests/satosa/frontends/test_saml2.py

@@ -69,6 +69,7 @@ def setup_for_authn_req(self, context, idp_conf, sp_conf, nameid_format=None, re
 
         base_url = self.construct_base_url_from_entity_id(idp_conf["entityid"])
         samlfrontend = SAMLFrontend(lambda ctx, internal_req: (ctx, internal_req),
+                                    lambda ctx, internal_logout_req: (ctx, internal_logout_req),
                                     internal_attributes, config, base_url, "saml_frontend")
         samlfrontend.register_endpoints(["saml"])
 
@@ -119,7 +120,8 @@ def get_auth_response(self, samlfrontend, context, internal_response, sp_conf, i
     ])
     def test_config_error_handling(self, conf):
         with pytest.raises(ValueError):
-            SAMLFrontend(lambda ctx, req: None, INTERNAL_ATTRIBUTES, conf, "base_url", "saml_frontend")
+            SAMLFrontend(lambda ctx, req: None, lambda ctx, req: None,
+                         INTERNAL_ATTRIBUTES, conf, "base_url", "saml_frontend")
 
     def test_register_endpoints(self, idp_conf):
         """
@@ -133,6 +135,7 @@ def get_path_from_url(url):
 
         base_url = self.construct_base_url_from_entity_id(idp_conf["entityid"])
         samlfrontend = SAMLFrontend(lambda context, internal_req: (context, internal_req),
+                                    lambda context, internal_logout_req: (context, internal_logout_req),
                                     INTERNAL_ATTRIBUTES, config, base_url, "saml_frontend")
 
         providers = ["foo", "bar"]
@@ -247,7 +250,7 @@ def test_get_filter_attributes_with_sp_requested_attributes_without_friendlyname
                                                "eduPersonAffiliation", "mail", "displayName", "sn",
                                                "givenName"]}}  # no op mapping for saml attribute names
 
-        samlfrontend = SAMLFrontend(None, internal_attributes, conf, base_url, "saml_frontend")
+        samlfrontend = SAMLFrontend(None, None, internal_attributes, conf, base_url, "saml_frontend")
         samlfrontend.register_endpoints(["testprovider"])
 
         internal_req = InternalData(
@@ -357,7 +360,8 @@ def test_sp_metadata_without_uiinfo(self, context, idp_conf, sp_conf):
 
     def test_metadata_endpoint(self, context, idp_conf):
         conf = {"idp_config": idp_conf, "endpoints": ENDPOINTS}
-        samlfrontend = SAMLFrontend(lambda ctx, req: None, INTERNAL_ATTRIBUTES, conf, "base_url", "saml_frontend")
+        samlfrontend = SAMLFrontend(lambda ctx, req: None, lambda ctx, req: None,
+                                    INTERNAL_ATTRIBUTES, conf, "base_url", "saml_frontend")
         samlfrontend.register_endpoints(["todo"])
         resp = samlfrontend._metadata_endpoint(context)
         headers = dict(resp.headers)
@@ -399,7 +403,8 @@ class TestSAMLMirrorFrontend:
     @pytest.fixture(autouse=True)
     def create_frontend(self, idp_conf):
         conf = {"idp_config": idp_conf, "endpoints": ENDPOINTS}
-        self.frontend = SAMLMirrorFrontend(lambda ctx, req: None, INTERNAL_ATTRIBUTES, conf, BASE_URL,
+        self.frontend = SAMLMirrorFrontend(lambda ctx, req: None, lambda ctx, req: None,
+                                           INTERNAL_ATTRIBUTES, conf, BASE_URL,
                                            "saml_mirror_frontend")
         self.frontend.register_endpoints([self.BACKEND])
 
@@ -490,6 +495,7 @@ def frontend(self, idp_conf, sp_conf):
         # Create, register the endpoints, and then return the frontend
         # instance.
         frontend = SAMLVirtualCoFrontend(lambda ctx, req: None,
+                                         lambda ctx, logout_req: None,
                                          internal_attributes,
                                          conf,
                                          BASE_URL,