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; }