Attempt at a database unittest

Author: ghalse

.travis.yml

@@ -11,6 +11,27 @@ matrix:
   allow_failures:
   - php: hhvm
 
+services:
+ - mysql
+
+# Set up a test database we can use for the the unit tests
+before_install:
+  - mysql -e 'CREATE DATABASE IF NOT EXISTS simplesamlphp;'
+  - mysql -e 'CREATE TABLE `AttributeFromSQL` (
+        `uid` VARCHAR(100) NOT NULL,
+        `sp` VARCHAR(250) DEFAULT "%",
+        `attribute` VARCHAR(30) NOT NULL,
+        `value` TEXT
+    ) DEFAULT CHARSET=utf8;
+    GRANT ALL ON `simplesamlphp`.* TO `phpunit`@`localhost` IDENTIFIED BY "phpunit";
+    ' simplesamlphp
+  - mysql -e "
+    INSERT INTO AttributeFromSQL (uid, sp, attribute, value) VALUES ('user@example.org', '%', 'eduPersonEntitlement', 'urn:mace:exampleIdP.org:demoservice:demo-admin');
+    INSERT INTO AttributeFromSQL (uid, sp, attribute, value) VALUES ('user@example.org', 'https://idp.example.org/idp/shibboleth', 'eduPersonEntitlement', 'urn:mace:grnet.gr:eduroam:admin');
+    INSERT INTO AttributeFromSQL (uid, sp, attribute, value) VALUES ('user@example.org', '%', 'eduPersonAffiliation', 'faculty');
+    INSERT INTO AttributeFromSQL (uid, attribute, value) VALUES ('user@example.org', 'mail', 'user@example.org');
+    " simplesamlphp
+
 before_script:
   composer update --dev
 

tests/lib/Auth/Process/AttributeFromSQLTest.php

@@ -25,8 +25,82 @@ protected function setUp()
         \SimpleSAML_Configuration::loadFromArray(array(), '[ARRAY]', 'simplesaml');
     }
 
-    public function testAny()
+    /**
+     * Test the example from docs
+     */
+    public function testExample()
+    {
+        $config = array(
+            'attribute' => 'eduPersonPrincipalName',
+            'limit' => array('eduPersonEntitlement', 'eduPersonAffiliation'),
+            'replace' => false,
+            'database' => array(
+                'username' => 'phpunit',
+                'password' => 'phpunit',
+            ),
+        );
+        $request = array(
+            'Attributes' => array(
+                'eduPersonPrincipalName' => array('user@example.org'),
+                'eduPersonAffiliation' => array('member'),
+                'displayName' => array('Example User'),
+            ),
+            'Destination' => array(
+                'entityid' => 'https://idp.example.org/idp/shibboleth',
+            ),
+        );
+        $result = self::processFilter($config, $request);
+        $attributes = $result['Attributes'];
+        $expectedData = array(
+            'eduPersonPrincipalName' => array('user@example.org'),
+            'displayName' => array('Example User'),
+            'eduPersonEntitlement' => array(
+                'urn:mace:exampleIdP.org:demoservice:demo-admin',
+                'urn:mace:grnet.gr:eduroam:admin',
+            ),
+            'eduPersonAffiliation' => array(
+                'member',
+                'faculty',
+            ),
+        );
+        $this->assertEquals($expectedData, $attributes, "Expected data was not correct");
+    }
+
+    /**
+     * Test attribute replacement
+     */
+    public function testReplace()
     {
-        $this->assertTrue(true, 'Just for travis.yml test');
+        $config = array(
+            'attribute' => 'eduPersonPrincipalName',
+            'limit' => array('mail', 'eduPersonAffiliation'),
+            'replace' => true,
+            'database' => array(
+                'username' => 'phpunit',
+                'password' => 'phpunit',
+            ),
+        );
+        $request = array(
+            'Attributes' => array(
+                'eduPersonPrincipalName' => array('user@example.org'),
+                'eduPersonAffiliation' => array('member'),
+                'displayName' => array('Example User'),
+            ),
+            'Destination' => array(
+                'entityid' => 'https://idp.example.org/idp/shibboleth',
+            ),
+        );
+        $result = self::processFilter($config, $request);
+        $attributes = $result['Attributes'];
+        $expectedData = array(
+            'eduPersonPrincipalName' => array('user@example.org'),
+            'displayName' => array('Example User'),
+            'eduPersonAffiliation' => array(
+                'faculty',
+            ),
+            'mail' => array('user@example.org'),
+        );
+        $this->assertEquals($expectedData, $attributes, "Expected data was not correct");
     }
+
 }