IdP hinting improvements

- fix: exception accessing to qs_params
- chore: improved exacmple configuration
- feat: added unit tests

Author: peppelinux

example/plugins/microservices/idp_hinting.yaml.example

@@ -4,3 +4,4 @@ config:
   allowed_params:
     - idp_hinting
     - idp_hint
+    - idphint

src/satosa/micro_services/idp_hinting.py

@@ -51,10 +51,9 @@ def process(self, context, data):
         hints = (
             entity_id
             for param_name in self.idp_hint_param_names
-            for qs_param_name, entity_id in qs_params
+            for qs_param_name, entity_id in qs_params.items()
             if param_name == qs_param_name
         )
         hint = next(hints, None)
-
         context.decorate(context.KEY_TARGET_ENTITYID, hint)
         return super().process(context, data)

tests/satosa/micro_services/test_idp_hinting.py

@@ -0,0 +1,40 @@
+from unittest import TestCase
+
+import pytest
+
+from satosa.context import Context
+from satosa.state import State
+from satosa.micro_services.idp_hinting import IdpHinting
+
+
+class TestIdpHinting(TestCase):
+    def setUp(self):
+        context = Context()
+        context.state = State()
+
+        config = {
+              'allowed_params': ["idp_hinting", "idp_hint", "idphint"]
+        }
+
+        plugin = IdpHinting(
+            config=config,
+            name='test_idphinting',
+            base_url='https://satosa.example.org',
+        )
+        plugin.next = lambda ctx, data: (ctx, data)
+
+        self.config = config
+        self.context = context
+        self.plugin = plugin
+
+    def test_idp_hinting(self):
+        self.context.request = {}
+        _target = 'https://localhost:8080'
+        self.context.qs_params = {'idphint': _target}
+        res = self.plugin.process(self.context, data={})
+        assert res[0].internal_data.get('target_entity_id') == _target
+
+    def test_no_idp_hinting(self):
+        self.context.request = {}
+        res = self.plugin.process(self.context, data={})
+        assert not res[0].internal_data.get('target_entity_id')