From 5b4adb286e24f538bcd1cab17d17268feffe01e8 Mon Sep 17 00:00:00 2001
From: Andrew Aitken-Fincham <andrew@silverstripe.com>
Date: Fri, 26 May 2017 17:40:25 +0100
Subject: [PATCH] add object traits

---
 src/EnvironmentCheckSuite.php | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/EnvironmentCheckSuite.php b/src/EnvironmentCheckSuite.php
index 0e7950e..4807ef4 100644
--- a/src/EnvironmentCheckSuite.php
+++ b/src/EnvironmentCheckSuite.php
@@ -4,10 +4,12 @@ namespace SilverStripe\EnvironmentCheck;
 
 use Exception;
 use InvalidArgumentException;
+use SilverStripe\Core\Config\Configurable;
+use SilverStripe\Core\Extensible;
+use SilverStripe\Core\Injector\Injectable;
 use SilverStripe\Core\Injector\Injector;
 use SilverStripe\EnvironmentCheck\EnvironmentCheck;
 use SilverStripe\ORM\ArrayList;
-use SilverStripe\ORM\DataObject;
 use SilverStripe\View\ArrayData;
 use SilverStripe\View\ViewableData;
 
@@ -32,8 +34,11 @@ use SilverStripe\View\ViewableData;
  *
  * @package environmentcheck
  */
-class EnvironmentCheckSuite extends DataObject
+class EnvironmentCheckSuite
 {
+    use Configurable;
+    use Injectable;
+    use Extensible;
     /**
      * Name of this suite.
      *
@@ -72,8 +77,7 @@ class EnvironmentCheckSuite extends DataObject
      */
     public function __construct($suiteName)
     {
-        parent::__construct();
-
+        $this->constructExtensions();
         if (empty($this->config()->registered_suites[$suiteName])) {
             // Not registered via config system, but it still may be configured later via self::register.
             return;