#124 Implemented helper to generate test state

src/Tests/ModelTestState.php

@@ -7,14 +7,15 @@
 
 class ModelTestState extends TableTestState
 {
-    public function __construct(string $modelClassName)
+    public function __construct(string $modelClassName, bool $globalExportMode = false)
     {
         $model = new $modelClassName();
 
         parent::__construct(
             tableName: $model->getTable(),
             jsonFields: $this->getModelJSONFields($model),
-            connectionName: $model->getConnectionName(),
+            globalExportMode: $globalExportMode,
+            connectionName: $model->getConnectionName($model),
         );
     }
 

src/Tests/TableTestState.php

@@ -14,15 +14,21 @@ class TableTestState extends Assert
     use FixturesTrait;
 
     protected string $tableName;
-    protected Collection $state;
-    protected ?string $connectionName;
     protected array $jsonFields;
+    protected ?string $connectionName;
+    protected bool $globalExportMode;
+    protected Collection $state;
 
-    public function __construct(string $tableName, array $jsonFields = [], ?string $connectionName = null)
-    {
+    public function __construct(
+        string $tableName,
+        array $jsonFields = [],
+        bool $globalExportMode = false,
+        ?string $connectionName = null,
+    ) {
         $this->tableName = $tableName;
         $this->jsonFields = $jsonFields;
         $this->connectionName = $connectionName ?? DB::getDefaultConnection();
+        $this->globalExportMode = $globalExportMode;
         $this->state = $this->getDataSet($tableName);
     }
 

tests/ModelTestStateTest.php

@@ -37,6 +37,28 @@ public function testInitialization()
         $this->assertEquals($originRecords, $state);
     }
 
+    public function testInitializationViaPrepareModelTestState()
+    {
+        $datasetMock = collect($this->getJsonFixture('initialization/dataset.json'));
+        $this->mockGettingDataset($datasetMock);
+
+        $mock = $this->getMockBuilder(TestCase::class)
+            ->disableOriginalConstructor()
+            ->getMockForAbstractClass();
+
+        $reflection = new ReflectionClass($mock);
+
+        $this->assertFalse($reflection->getProperty('globalExportMode')->getValue($mock));
+
+        $setGlobalExportMode = $reflection->getMethod('setGlobalExportMode');
+        $setGlobalExportMode->invoke($mock);
+
+        $prepareModelTestState = $reflection->getMethod('prepareModelTestState');
+        $prepareModelTestState->invoke($mock, TestModel::class);
+
+        $this->assertTrue($reflection->getProperty('globalExportMode')->getValue($mock));
+    }
+
     public function testAssertChangesEqualsFixture()
     {
         $initialDatasetMock = collect($this->getJsonFixture('changes_equals_fixture/initial_dataset.json'));

tests/TableTestStateTest.php

@@ -26,6 +26,28 @@ public function testInitialization()
         $this->assertEquals($originRecords, $state);
     }
 
+    public function testInitializationViaPrepareTableTestState()
+    {
+        $datasetMock = collect($this->getJsonFixture('initialization/dataset.json'));
+        $this->mockGettingDataset($datasetMock);
+
+        $mock = $this->getMockBuilder(TestCase::class)
+            ->disableOriginalConstructor()
+            ->getMockForAbstractClass();
+
+        $reflection = new ReflectionClass($mock);
+
+        $this->assertFalse($reflection->getProperty('globalExportMode')->getValue($mock));
+
+        $setGlobalExportMode = $reflection->getMethod('setGlobalExportMode');
+        $setGlobalExportMode->invoke($mock);
+
+        $prepareTableTestState = $reflection->getMethod('prepareTableTestState');
+        $prepareTableTestState->invoke($mock, 'test_models');
+
+        $this->assertTrue($reflection->getProperty('globalExportMode')->getValue($mock));
+    }
+
     public function testAssertChangesEqualsFixture()
     {
         $initialDatasetMock = collect($this->getJsonFixture('changes_equals_fixture/initial_dataset.json'));

tests/TestCase.php

@@ -112,4 +112,14 @@ protected function rollbackTransaction(): void
             $connection->disconnect();
         }
     }
+
+    protected function prepareModelTestState(string $modelClassName): ModelTestState
+    {
+        return new ModelTestState($modelClassName, $this->globalExportMode);
+    }
+
+    protected function prepareTableTestState(string $tableName, array $jsonFields = [], ?string $connectionName = null): TableTestState
+    {
+        return new TableTestState($tableName, $jsonFields, $this->globalExportMode, $connectionName);
+    }
 }