test adjustments

heimrichhannot · Mar 1, 2024 · 9445ecb · 9445ecb
1 parent 1c480b4
commit 9445ecb
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/tests/EventListener/DcaField/DateAddedFieldListenerTest.php b/tests/EventListener/DcaField/DateAddedFieldListenerTest.php
@@ -4,13 +4,16 @@
 
 use Contao\DataContainer;
 use Contao\Model;
+use HeimrichHannot\TestUtilitiesBundle\Mock\ModelMockTrait;
 use HeimrichHannot\UtilsBundle\Dca\DateAddedField;
 use HeimrichHannot\UtilsBundle\EventListener\DcaField\DateAddedFieldListener;
 use PHPUnit\Framework\TestCase;
 use Psr\Container\ContainerInterface;
 
 class DateAddedFieldListenerTest extends TestCase
 {
+    use ModelMockTrait;
+
     /**
      * @runInSeparateProcess
      */
@@ -88,10 +91,7 @@ public function testOnLoadCallback()
             ->onlyMethods(['getModelInstance'])
             ->getMock();
 
-        $model = $this->getMockBuilder(Model::class)
-            ->disableOriginalConstructor()
-            ->onlyMethods(['save'])
-            ->getMock();
+        $model = $this->mockModelObject(Model::class);
 
         $model->dateAdded = 0;
 
@@ -169,10 +169,7 @@ public function testOnCopyCallback()
             }
         });
 
-        $model = $this->getMockBuilder(Model::class)
-            ->disableOriginalConstructor()
-            ->onlyMethods(['save'])
-            ->getMock();
+        $model = $this->mockModelObject(Model::class);
 
         $model->dateAdded = 0;