Commit 20b70e3

mo khan <mo@mokhan.ca>
2025-03-26 19:18:50
feat: add lookup for Assertion
main
1 parent 789ffa1
Changed files (2)
lib
saml
kit
document.rb
spec
saml
kit
document_spec.rb
lib/saml/kit/document.rb
@@ -53,12 +53,14 @@ module Saml
 
       class << self
         CONSTRUCTORS = {
+          'Assertion' => -> { Saml::Kit::Assertion },
           'AuthnRequest' => -> { Saml::Kit::AuthenticationRequest },
           'LogoutRequest' => -> { Saml::Kit::LogoutRequest },
           'LogoutResponse' => -> { Saml::Kit::LogoutResponse },
           'Response' => -> { Saml::Kit::Response },
         }.freeze
         XPATH = [
+          '/saml:Assertion',
           '/samlp:AuthnRequest',
           '/samlp:LogoutRequest',
           '/samlp:LogoutResponse',
@@ -71,7 +73,10 @@ module Saml
         # @param configuration [Saml::Kit::Configuration] configuration to use
         # for unpacking the document.
         def to_saml_document(xml, configuration: Saml::Kit.configuration)
-          namespaces = { samlp: Namespaces::PROTOCOL }
+          namespaces = {
+            saml: Namespaces::ASSERTION,
+            samlp: Namespaces::PROTOCOL,
+          }
           element = Nokogiri::XML(xml).at_xpath(XPATH, namespaces)
           constructor = CONSTRUCTORS[element.name].try(:call) || InvalidDocument
           constructor.new(xml, configuration: configuration)
spec/saml/kit/document_spec.rb
@@ -50,6 +50,12 @@ RSpec.describe Saml::Kit::Document do
       expect(result).to be_instance_of(Saml::Kit::LogoutRequest)
     end
 
+    it 'returns an Assertion' do
+      xml = Saml::Kit::Response.build(user, request).assertion.to_xml
+      result = subject.to_saml_document(xml)
+      expect(result).to be_instance_of(Saml::Kit::Assertion)
+    end
+
     it 'returns an invalid document' do
       xml = <<-XML
       <html>