Test out the query tagging for IP range statements.

Bit of formatting, and using constants.
discoverygarden · Mar 9, 2024 · 19abd03 · 19abd03
1 parent 19bb47c
commit 19abd03
Showing 1 changed file with 27 additions and 2 deletions.
diff --git a/tests/src/Kernel/IpRangeEmbargoTest.php b/tests/src/Kernel/IpRangeEmbargoTest.php
@@ -5,6 +5,7 @@
 use Drupal\embargo\EmbargoInterface;
 use Drupal\embargo\IpRangeInterface;
 use Drupal\node\NodeInterface;
+use Drupal\Tests\islandora_test_support\Traits\DatabaseQueryTestTraits;
 
 /**
  * Test IpRange embargo.
@@ -13,6 +14,8 @@
  */
 class IpRangeEmbargoTest extends EmbargoKernelTestBase {
 
+  use DatabaseQueryTestTraits;
+
   /**
    * Embargo for test.
    *
@@ -105,8 +108,16 @@ public function setUp(): void {
     $this->embargoedNodeWithDifferentIpRange = $this->createNode();
     $this->currentIpRangeEntity = $this->createIpRangeEntity($this->ipRange);
     $this->embargoWithoutIpRange = $this->createEmbargo($this->embargoedNodeWithoutIpRange);
-    $this->embargoWithCurrentIpRange = $this->createEmbargo($this->embargoedNodeWithCurrentIpRange, 1, $this->currentIpRangeEntity);
-    $this->embargoWithDifferentIpRange = $this->createEmbargo($this->embargoedNodeWithDifferentIpRange, 1, $this->createIpRangeEntity('0.0.0.0.1/29'));
+    $this->embargoWithCurrentIpRange = $this->createEmbargo(
+      $this->embargoedNodeWithCurrentIpRange,
+      EmbargoInterface::EMBARGO_TYPE_NODE,
+      $this->currentIpRangeEntity,
+    );
+    $this->embargoWithDifferentIpRange = $this->createEmbargo(
+      $this->embargoedNodeWithDifferentIpRange,
+      EmbargoInterface::EMBARGO_TYPE_NODE,
+      $this->createIpRangeEntity('0.0.0.1/29'),
+    );
   }
 
   /**
@@ -127,4 +138,18 @@ public function testIpRangeEmbargoNodeAccess() {
     $this->assertTrue($this->embargoedNodeWithCurrentIpRange->access('view', $this->user));
   }
 
+  /**
+   * Test IP range query tagging.
+   */
+  public function testIpRangeQueryTagging() {
+    $results = $this->generateNodeSelectAccessQuery($this->user)->execute()->fetchAll();
+
+    $ids = array_column($results, 'nid');
+
+    $this->assertContains($this->nonEmbargoedNode->id(), $ids, 'non-embargoed node present');
+    $this->assertNotContains($this->embargoedNodeWithoutIpRange->id(), $ids, 'generally embargoed node absent');
+    $this->assertNotContains($this->embargoedNodeWithDifferentIpRange->id(), $ids, 'node exempted to other ranges absent');
+    $this->assertContains($this->embargoedNodeWithCurrentIpRange->id(), $ids, 'node exempted to our range present');
+  }
+
 }