From 13c105f6d9eea7934a3ed17e3d70eaebce25d38d Mon Sep 17 00:00:00 2001
From: Rodrigo Primo <rodrigosprimo@gmail.com>
Date: Wed, 25 Oct 2023 16:55:08 -0300
Subject: [PATCH] Add method to set unsubscribeToken to Newsletter factory

[MAILPOET-5680]
---
 mailpoet/tests/DataFactories/Newsletter.php | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/mailpoet/tests/DataFactories/Newsletter.php b/mailpoet/tests/DataFactories/Newsletter.php
index 8b9c338424..61e065ad21 100644
--- a/mailpoet/tests/DataFactories/Newsletter.php
+++ b/mailpoet/tests/DataFactories/Newsletter.php
@@ -173,6 +173,11 @@ class Newsletter {
     return $this;
   }
 
+  public function withUnsubscribeToken(string $unsubscribeToken) {
+    $this->data['unsubscribeToken'] = $unsubscribeToken;
+    return $this;
+  }
+
   /**
    * @return Newsletter
    */
@@ -417,6 +422,10 @@ class Newsletter {
     if (isset($this->data['ga_campaign'])) $newsletter->setGaCampaign($this->data['ga_campaign']);
     if (isset($this->data['wp_post_id'])) $newsletter->setWpPostId($this->data['wp_post_id']);
 
+    if (isset($this->data['unsubscribeToken'])) {
+      $newsletter->setUnsubscribeToken($this->data['unsubscribeToken']);
+    }
+
     return $newsletter;
   }