Update unit tests

silverstripe · Sep 9, 2022 · 9529cba · 9529cba
1 parent 4023774
commit 9529cba
Showing 1 changed file with 29 additions and 3 deletions.
diff --git a/tests/DataObject/DataObjectDocumentTest.php b/tests/DataObject/DataObjectDocumentTest.php
@@ -185,6 +185,7 @@ public function testSubsiteDataObjectShouldIndex(): void
         // Mocked indexes:
         //  - index0: allows all data without subsite filter
         //  - index1: for subsite2 and Page class and Data object that does not implement subsite
+        //  - index2: for subsite2 and Data object that does not implement subsite
         $config->set(
             'indexes',
             [
@@ -199,6 +200,13 @@ public function testSubsiteDataObjectShouldIndex(): void
                     'includeClasses' => [
                         Page::class => true,
                         DataObjectFake::class => true,
+                        DataObjectFakeVersioned::class => true,
+                    ],
+                ],
+                'index2' => [
+                    'subsite_id' => $subsite2->ID,
+                    'includeClasses' => [
+                        DataObjectFake::class => true,
                     ],
                 ],
             ]
@@ -239,14 +247,32 @@ public function testSubsiteDataObjectShouldIndex(): void
         $this->assertTrue($doc3->shouldIndex());
 
         // Get an object without subsite filter
-        $object = $this->objFromFixture(DataObjectFake::class, 'one');
+        $object1 = $this->objFromFixture(DataObjectFake::class, 'one');
 
         // Prepare object for index
-        $doc4 = DataObjectDocument::create($object);
+        $doc4 = DataObjectDocument::create($object1);
 
         // Assert that the object without subsite ID and that it can be indexed, because configuration allows it
-        $this->assertNull($object->SubsiteID);
+        $this->assertNull($object1->SubsiteID);
         $this->assertTrue($doc4->shouldIndex());
+        $this->assertArrayHasKey('index1', $doc4->getIndexes());
+        $this->assertArrayHasKey('index2', $doc4->getIndexes());
+        $this->assertArrayNotHasKey('index0', $doc4->getIndexes());
+
+        // Get an object without subsite filter
+        $object2 = $this->objFromFixture(DataObjectFakeVersioned::class, 'one');
+        $object2->publishRecursive();
+        $object2 = $this->objFromFixture(DataObjectFakeVersioned::class, 'one');
+
+        // Prepare object for index
+        $doc5 = DataObjectDocument::create($object2);
+
+        // Assert that the object without subsite ID and that it can be indexed, because configuration allows it
+        $this->assertNull($object2->SubsiteID);
+        $this->assertTrue($doc5->shouldIndex());
+        $this->assertArrayHasKey('index1', $doc5->getIndexes());
+        $this->assertArrayNotHasKey('index2', $doc5->getIndexes());
+        $this->assertArrayNotHasKey('index0', $doc5->getIndexes());
     }
 
     public function testMarkIndexed(): void