From 4aef1a11cbc9df342997530c9e338e25508aeb14 Mon Sep 17 00:00:00 2001
From: Guy Sartorelli <36352093+GuySartorelli@users.noreply.github.com>
Date: Thu, 21 Mar 2024 12:49:19 +1300
Subject: [PATCH] ENH Add ability to create roles for behat tests (#267)

---
 src/Context/FixtureContext.php | 44 ++++++++++++++++++++++++++++++++++
 1 file changed, 44 insertions(+)

diff --git a/src/Context/FixtureContext.php b/src/Context/FixtureContext.php
index d6bcd47..8300b44 100644
--- a/src/Context/FixtureContext.php
+++ b/src/Context/FixtureContext.php
@@ -31,6 +31,8 @@ use SilverStripe\Security\Member;
 use SilverStripe\Security\Permission;
 use SilverStripe\Versioned\Versioned;
 use SilverStripe\Core\Config\Config;
+use SilverStripe\Security\PermissionRole;
+use SilverStripe\Security\PermissionRoleCode;
 
 /**
  * Context used to create fixtures in the SilverStripe ORM.
@@ -620,6 +622,48 @@ class FixtureContext implements Context
         }
     }
 
+    /**
+     * Example: Given a "role" "Some role" with permissions "Access to 'Pages' section" and "Access to 'Files' section"
+     *
+     * @Given /^(?:an|a|the) "role" "([^"]+)" (?:with|has) permissions (.*)$/
+     * @param string $id
+     * @param string $permissionStr
+     */
+    public function stepCreateRoleWithPermissions($id, $permissionStr)
+    {
+        // Convert natural language permissions to codes
+        preg_match_all('/"([^"]+)"/', $permissionStr ?? '', $matches);
+        $permissions = $matches[1];
+        $codes = Permission::get_codes(false);
+
+        $role = $this->getFixtureFactory()->get(PermissionRole::class, $id);
+        if (!$role) {
+            $role = $this->getFixtureFactory()->createObject(PermissionRole::class, $id);
+        }
+
+        foreach ($permissions as $permission) {
+            $found = false;
+            foreach ($codes as $code => $details) {
+                if ($permission == $code
+                    || $permission == $details['name']
+                ) {
+                    $permissionRoleCode = PermissionRoleCode::create([
+                        'RoleID' => $role->ID,
+                        'Code' => $code,
+                    ]);
+                    $permissionRoleCode->write();
+                    $found = true;
+                }
+            }
+            if (!$found) {
+                throw new InvalidArgumentException(sprintf(
+                    'No permission found for "%s"',
+                    $permission
+                ));
+            }
+        }
+    }
+
     /**
      * Navigates to a record based on its identifier set during fixture creation,
      * using its RelativeLink() method to map the record to a URL.