ENH Add a status flag for companies that we can behat test against

silverstripe · Nov 21, 2024 · 5c92057 · 5c92057
1 parent 5ca1ed5
commit 5c92057
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/code/Company.php b/code/Company.php
@@ -368,4 +368,15 @@ public function scaffoldSearchField()
         return DropdownField::create('CompanyID', 'Company', Company::get()->map())->setEmptyString('');
     }
 
+    public function getStatusFlags(bool $cached = true): array
+    {
+        $this->beforeExtending('updateStatusFlags', function (array &$flags) {
+            $flags['company-status-flag1'] = 'string-flag';
+            $flags['company-status-flag2'] = [
+                'title' => 'a flag with a title',
+                'text' => 'array-flag',
+            ];
+        });
+        return parent::getStatusFlags($cached);
+    }
 }