Merge pull request #980 from creative-commoners/pulls/8/test-generated-columns

MNT Add tests for generated columns

Author: emteknetnz

tests/php/Extension/FluentExtensionTest.php

@@ -20,6 +20,7 @@
 use TractorCow\Fluent\Tests\Extension\FluentExtensionTest\UnlocalisedChild;
 use TractorCow\Fluent\Tests\Extension\Stub\FluentStubObject;
 use PHPUnit\Framework\Attributes\DataProvider;
+use TractorCow\Fluent\Tests\Extension\FluentExtensionTest\TestGeneratedColumns;
 
 class FluentExtensionTest extends SapphireTest
 {
@@ -33,6 +34,7 @@ class FluentExtensionTest extends SapphireTest
         UnlocalisedChild::class,
         TestRelationPage::class,
         TestModel::class,
+        TestGeneratedColumns::class,
     ];
 
     protected static $required_extensions = [
@@ -510,6 +512,45 @@ public static function sortRecordProvider()
         ];
     }
 
+    public function testLocalisedGeneratedColumns(): void
+    {
+        // Setup first localisation
+        $recordID = FluentState::singleton()->withState(function (FluentState $state): int {
+            $state->setLocale('en_US');
+
+            $record = new TestGeneratedColumns();
+            $record->BaseField = 'EN copy test';
+            return $record->write();
+        });
+
+        // Setup second localisation and make sure we have two localisations to work with
+        FluentState::singleton()->withState(function (FluentState $state) use ($recordID): void {
+            $state->setLocale('de_DE');
+
+            $record = TestGeneratedColumns::get()->byID($recordID);
+            $record->BaseField = 'DE copy test';
+            $record->write();
+        });
+
+        // Check the values of the generated columns in en_US
+        FluentState::singleton()->withState(function (FluentState $state) use ($recordID): void {
+            $state->setLocale('en_US');
+
+            $record = TestGeneratedColumns::get()->byID($recordID);
+            $this->assertSame('EN copy test_virtual', $record->GeneratedField1);
+            $this->assertSame('EN copy test_stored', $record->GeneratedField2);
+        });
+
+        // Check the values of the generated columns in de_DE
+        FluentState::singleton()->withState(function (FluentState $state) use ($recordID): void {
+            $state->setLocale('de_DE');
+
+            $record = TestGeneratedColumns::get()->byID($recordID);
+            $this->assertSame('DE copy test_virtual', $record->GeneratedField1);
+            $this->assertSame('DE copy test_stored', $record->GeneratedField2);
+        });
+    }
+
     /**
      * Get a Locale field value directly from a record's localised database table, skipping the ORM
      *

tests/php/Extension/FluentExtensionTest/TestGeneratedColumns.php

@@ -0,0 +1,31 @@
+<?php
+
+namespace TractorCow\Fluent\Tests\Extension\FluentExtensionTest;
+
+use SilverStripe\Dev\TestOnly;
+use SilverStripe\ORM\DataObject;
+use TractorCow\Fluent\Extension\FluentExtension;
+
+/**
+ * @mixin FluentExtension
+ */
+class TestGeneratedColumns extends DataObject implements TestOnly
+{
+    private static string $table_name = 'FluentExtensionTest_TestGeneratedColumns';
+
+    private static array $db = [
+        'BaseField' => 'Varchar(255)',
+        'GeneratedField1' => 'Generated("Varchar(255)", "CONCAT(\\"BaseField\\", \'_virtual\')", "VIRTUAL")',
+        'GeneratedField2' => 'Generated("Varchar(255)", "CONCAT(\\"BaseField\\", \'_stored\')", "STORED")',
+    ];
+
+    private static array $extensions = [
+        FluentExtension::class,
+    ];
+
+    private static $translate = [
+        'BaseField',
+        'GeneratedField1',
+        'GeneratedField2',
+    ];
+}