Add support for xs:dateTime AttributeValue and improve tests

simplesamlphp · Sep 2, 2024 · 166cd28 · 166cd28
1 parent a30604e
commit 166cd28
Show file tree

Hide file tree

Showing 4 changed files with 59 additions and 3 deletions.
diff --git a/src/SAML2/XML/saml/AttributeValue.php b/src/SAML2/XML/saml/AttributeValue.php
@@ -4,6 +4,8 @@
 
 namespace SimpleSAML\SAML2\XML\saml;
 
+use DateTimeImmutable;
+use DateTimeInterface;
 use DOMElement;
 use SimpleSAML\Assert\Assert;
 use SimpleSAML\SAML2\Constants as C;
@@ -27,7 +29,7 @@ class AttributeValue extends AbstractSamlElement
  /**
  * Create an AttributeValue.
  *
- * @param string|int|null|\SimpleSAML\XML\AbstractElement $value The value of this element. Can be one of:
+ * @param string|int|null|\DateTimeInterface|\SimpleSAML\XML\AbstractElement $value The value of this element. Can be one of:
  * - string
  * - int
  * - null
@@ -36,7 +38,7 @@ class AttributeValue extends AbstractSamlElement
  * @throws \SimpleSAML\Assert\AssertionFailedException if the supplied value is neither a string or a DOMElement
  */
  final public function __construct(
- protected string|int|null|AbstractElement $value,
+ protected string|int|null|DateTimeInterface|AbstractElement $value,
  ) {
  }
 
@@ -56,6 +58,10 @@ public function getXsiType(): string
  case "NULL":
  return "xs:nil";
  case "object":
+ if ($this->value instanceof DateTimeInterface) {
+ return 'xs:dateTime';
+ }
+
  return sprintf(
  '%s:%s',
  $this->value::getNamespacePrefix(),
@@ -111,14 +117,25 @@ public static function fromXML(DOMElement $xml): static
  $xml->hasAttributeNS(C::NS_XSI, "type") &&
  $xml->getAttributeNS(C::NS_XSI, "type") === "xs:integer"
  ) {
+ Assert::numeric($xml->textContent);
+
  // we have an integer as value
  $value = intval($xml->textContent);
+ } elseif (
+ $xml->hasAttributeNS(C::NS_XSI, "type") &&
+ $xml->getAttributeNS(C::NS_XSI, "type") === "xs:dateTime"
+ ) {
+ Assert::validDateTime($xml->textContent);
+
+ // we have a dateTime as value
+ $value = new DateTimeImmutable($xml->textContent);
  } elseif (
  // null value
  $xml->hasAttributeNS(C::NS_XSI, "nil") &&
  ($xml->getAttributeNS(C::NS_XSI, "nil") === "1" ||
  $xml->getAttributeNS(C::NS_XSI, "nil") === "true")
  ) {
+ Assert::isEmpty($xml->textContent);
  $value = null;
  } else {
  $value = $xml->textContent;
@@ -155,7 +172,14 @@ public function toXML(DOMElement $parent = null): DOMElement
  $e->textContent = '';
  break;
  case "object":
- $this->getValue()->toXML($e);
+ if ($this->value instanceof DateTimeInterface) {
+ $e->setAttributeNS('http://www.w3.org/2000/xmlns/', 'xmlns:xsi', C::NS_XSI);
+ $e->setAttributeNS('http://www.w3.org/2000/xmlns/', 'xmlns:xs', C::NS_XS);
+ $e->setAttributeNS(C::NS_XSI, 'xsi:type', 'xs:dateTime');
+ $e->textContent = $this->value->format(C::DATETIME_FORMAT);
+ } else {
+ $this->getValue()->toXML($e);
+ }
  break;
  default: // string
  $e->textContent = $this->getValue();

diff --git a/tests/SAML2/XML/saml/AttributeTest.php b/tests/SAML2/XML/saml/AttributeTest.php
@@ -4,6 +4,7 @@
 
 namespace SimpleSAML\Test\SAML2\XML\saml;
 
+use DateTimeImmutable;
 use PHPUnit\Framework\Attributes\CoversClass;
 use PHPUnit\Framework\Attributes\Group;
 use PHPUnit\Framework\TestCase;
@@ -88,6 +89,9 @@ public function testMarshalling(): void
  [
  new AttributeValue('FirstValue'),
  new AttributeValue('SecondValue'),
+ new AttributeValue(3),
+ new AttributeValue(new DateTimeImmutable('2024-04-04T04:44:44Z')),
+ new AttributeValue(null),
  ],
  [$attr1, $attr2],
  );

diff --git a/tests/SAML2/XML/saml/AttributeValueTest.php b/tests/SAML2/XML/saml/AttributeValueTest.php
@@ -4,6 +4,7 @@
 
 namespace SimpleSAML\Test\SAML2\XML\saml;
 
+use DateTimeImmutable;
 use PHPUnit\Framework\Attributes\CoversClass;
 use PHPUnit\Framework\Attributes\Group;
 use PHPUnit\Framework\TestCase;
@@ -77,6 +78,30 @@ public function testMarshallingString(): void
  }
 
 
+ /**
+ * Test creating an AttributeValue from scratch using an integer.
+ */
+ public function testMarshallingInteger(): void
+ {
+ $av = new AttributeValue(3);
+
+ $this->assertEquals(3, $av->getValue());
+ $this->assertEquals('xs:integer', $av->getXsiType());
+ }
+
+
+ /**
+ * Test creating an AttributeValue from scratch using an dateTime.
+ */
+ public function testMarshallingDateTime(): void
+ {
+ $av = new AttributeValue(new DateTimeImmutable("2024-04-04T04:44:44Z"));
+
+ $this->assertEquals('2024-04-04T04:44:44Z', $av->getValue()->format(C::DATETIME_FORMAT));
+ $this->assertEquals('xs:dateTime', $av->getXsiType());
+ }
+
+
  /**
  */
  public function testMarshallingNull(): void

diff --git a/tests/resources/xml/saml_Attribute.xml b/tests/resources/xml/saml_Attribute.xml
@@ -1,4 +1,7 @@
 <saml:Attribute xmlns:saml="urn:oasis:names:tc:SAML:2.0:assertion" Name="TheName" NameFormat="urn:oasis:names:tc:SAML:2.0:attrname-format:basic" FriendlyName="TheFriendlyName" test:attr1="testval1" test:attr2="testval2" xmlns:test="urn:test:something">
  <saml:AttributeValue>FirstValue</saml:AttributeValue>
  <saml:AttributeValue>SecondValue</saml:AttributeValue>
+ <saml:AttributeValue xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:xs="http://www.w3.org/2001/XMLSchema" xsi:type="xs:integer">3</saml:AttributeValue>
+ <saml:AttributeValue xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:xs="http://www.w3.org/2001/XMLSchema" xsi:type="xs:dateTime">2024-04-04T04:44:44Z</saml:AttributeValue>
+ <saml:AttributeValue xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:nil="1"/>
 </saml:Attribute>