[project @ Add support for openid.ax.count.ALIAS="unlimited"]

1 files changed, 28 insertions, 0 deletions

diff --git a/Tests/Auth/OpenID/AX.php b/Tests/Auth/OpenID/AX.php
index 96e7cfa..9c9c397 100644
--- a/Tests/Auth/OpenID/AX.php
+++ b/Tests/Auth/OpenID/AX.php
@@ -160,6 +160,34 @@ class ParseAXValuesTest extends PHPUnit_TestCase {
                                           'count.foo' => '1'));
     }
 
+    function test_invalidCountValue()
+    {
+        $msg = new Auth_OpenID_AX_FetchRequest();
+
+        $result = $msg->parseExtensionArgs(
+                    array('type.foo' => 'urn:foo',
+                          'count.foo' => 'bogus'));
+
+        $this->assertTrue(Auth_OpenID_AX::isError($result));
+    }
+
+    function test_requestUnlimitedValues()
+    {
+        $msg = new Auth_OpenID_AX_FetchRequest();
+
+        $result = $msg->parseExtensionArgs(
+               array('mode' => 'fetch_request',
+                     'required' => 'foo',
+                     'type.foo' => 'urn:foo',
+                     'count.foo' => Auth_OpenID_AX_UNLIMITED_VALUES));
+
+        $attrs = $msg->iterAttrs();
+        $foo = $attrs[0];
+
+        $this->assertTrue($foo->count == Auth_OpenID_AX_UNLIMITED_VALUES);
+        $this->assertTrue($foo->wantsUnlimitedValues());
+    }
+
     function test_countPresentAndIsZero()
     {
         $this->failUnlessAXValues(