MNT Add custom validator for test cases

silverstripe · Nov 9, 2023 · 74b06c1 · 74b06c1
1 parent a5aeaa3
commit 74b06c1
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/code/Company.php b/code/Company.php
@@ -13,6 +13,8 @@
 use SilverStripe\Versioned\Versioned;
 use RelationFieldsTestPage;
 use GridFieldTestPage;
+use SilverStripe\Forms\CompositeValidator;
+use SilverStripe\Forms\RequiredFields;
 
 /**
  *
@@ -100,6 +102,16 @@ public function getCMSFields()
         return $fields;
     }
 
+    public function getCMSCompositeValidator(): CompositeValidator
+    {
+        $validator = parent::getCMSCompositeValidator();
+        $validator->addValidator(RequiredFields::create([
+            'Name',
+        ]));
+
+        return $validator;
+    }
+
     public function validate()
     {
         $result = parent::validate();

diff --git a/code/Employee.php b/code/Employee.php
@@ -3,8 +3,10 @@
 namespace SilverStripe\FrameworkTest\Model;
 
 use SilverStripe\Assets\Image;
+use SilverStripe\Forms\CompositeValidator;
 use SilverStripe\Forms\EmailField;
 use SilverStripe\Forms\NumericField;
+use SilverStripe\Forms\RequiredFields;
 use SilverStripe\Forms\TextField;
 use SilverStripe\ORM\Connect\MySQLSchemaManager;
 use SilverStripe\ORM\DataObject;
@@ -141,6 +143,15 @@ public function requireDefaultRecords()
         srand();
     }
 
+    public function getCMSCompositeValidator(): CompositeValidator
+    {
+        $validator = parent::getCMSCompositeValidator();
+        $validator->addValidator(RequiredFields::create([
+            'Name',
+        ]));
+        return $validator;
+    }
+
     public function validate()
     {
         $result = parent::validate();