From 0f76e3b48a117c78bb89298fabea22aedf454f12 Mon Sep 17 00:00:00 2001
From: Johan Lundberg <lundberg@sunet.se>
Date: Mon, 11 Dec 2023 09:39:08 +0100
Subject: [PATCH] add eduPersonEntitlement attribute to tests

---
 src/auth_server/tests/test_saml_sp.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/auth_server/tests/test_saml_sp.py b/src/auth_server/tests/test_saml_sp.py
index 28b5a4b..406dbc9 100644
--- a/src/auth_server/tests/test_saml_sp.py
+++ b/src/auth_server/tests/test_saml_sp.py
@@ -99,6 +99,9 @@ def setUp(self) -> None:
                 surname="Testsson",
                 targeted_id="https://idp.example.com/simplesaml/saml2/idp/metadata.php!http://test.localhost/saml2-metadata!398f4967ef4ec07985d93a9200d3891184b9c0f6c79db53280894ae75673eab8",
                 unique_id="eppn@idp.example.com",
+                entitlement=[
+                    "urn:mace:swamid.se:example.com:role:member",
+                ],
             ),
         )
 
@@ -171,12 +174,15 @@ def setUp(self) -> None:
               </saml:Attribute>
               <saml:Attribute Name="urn:oid:0.9.2342.19200300.100.1.3" NameFormat="urn:oasis:names:tc:SAML:2.0:attrname-format:uri" FriendlyName="mail">
                 <saml:AttributeValue xsi:type="xs:string">testsson@example.com</saml:AttributeValue>
-              </saml:Attribute>              
+              </saml:Attribute>
               <saml:Attribute Name="urn:oid:1.3.6.1.4.1.5923.1.1.1.10" NameFormat="urn:oasis:names:tc:SAML:2.0:attrname-format:uri" FriendlyName="eduPersonTargetedID">
                 <saml:AttributeValue>
                   <saml:NameID Format="urn:oasis:names:tc:SAML:2.0:nameid-format:persistent" NameQualifier="https://idp.example.com/simplesaml/saml2/idp/metadata.php" SPNameQualifier="http://test.localhost/saml2-metadata">398f4967ef4ec07985d93a9200d3891184b9c0f6c79db53280894ae75673eab8</saml:NameID>
                 </saml:AttributeValue>
               </saml:Attribute>
+              <saml:Attribute Name="urn:oid:1.3.6.1.4.1.5923.1.1.1.7" NameFormat="urn:oasis:names:tc:SAML:2.0:attrname-format:uri" FriendlyName="eduPersonEntitlement">
+                <saml:AttributeValue xsi:type="xs:string">urn:mace:swamid.se:example.com:role:member</saml:AttributeValue>
+              </saml:Attribute>
             </saml:AttributeStatement>
           </saml:Assertion>
         </samlp:Response>"""