From 1b3ccc12957a0edd00279b311ed7e6164a3d41ad Mon Sep 17 00:00:00 2001
From: Joachim Noreiko <joachim@107701.no-reply.drupal.org>
Date: Tue, 9 Jul 2024 14:49:19 +0100
Subject: [PATCH] Added test trait for creating directory fields.

---
 .../Traits/DirectoryFieldsCreationTrait.php   | 30 +++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 tests/src/Traits/DirectoryFieldsCreationTrait.php

diff --git a/tests/src/Traits/DirectoryFieldsCreationTrait.php b/tests/src/Traits/DirectoryFieldsCreationTrait.php
new file mode 100644
index 0000000..c13ff4f
--- /dev/null
+++ b/tests/src/Traits/DirectoryFieldsCreationTrait.php
@@ -0,0 +1,30 @@
+<?php
+
+namespace Drupal\Tests\localgov_directories\Traits;
+
+/**
+ * Trait for setting up directory fields.
+ *
+ * Expects the methods from \Drupal\Tests\field\Traits\EntityReferenceTestTrait
+ * to be present.
+ */
+trait DirectoryFieldsCreationTrait {
+
+  /**
+   * Creates a directory channel selection field.
+   *
+   * @param string $node_type
+   *   The ID of node type to create the field on.
+   */
+  public function createDirectoryChannelsSelectionField(string $node_type): void {
+    $handler_settings = [
+      'sort' => [
+        'field' => 'title',
+        'direction' => 'DESC',
+      ],
+    ];
+    $this->createEntityReferenceField('node', $node_type, 'localgov_directory_channels', $this->randomString(), 'node', 'localgov_directories_channels_selection', $handler_settings);
+  }
+
+}
+