Cavemanon/piratepoet
7
Fork 0
Code Pull Requests Actions Releases 16 Activity

Refactor tracking into doctrine

Browse Source 
[MAILPOET-2993]
This commit is contained in:
Pavel Dohnal
2020-11-02 08:45:47 +01:00
committed by Veljko V
parent 4134e6e882
commit 7d538abad8
7 changed files with 183 additions and 107 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lib/Newsletter/Shortcodes/Categories/Link.php
View File

@ -2,6 +2,7 @@
namespace MailPoet\Newsletter\Shortcodes\Categories; namespace MailPoet\Newsletter\Shortcodes\Categories;
use MailPoet\Entities\SendingQueueEntity;
use MailPoet\Newsletter\Url as NewsletterUrl; use MailPoet\Newsletter\Url as NewsletterUrl;
use MailPoet\Settings\SettingsController; use MailPoet\Settings\SettingsController;
use MailPoet\Subscription\SubscriptionUrlFactory; use MailPoet\Subscription\SubscriptionUrlFactory;
@ -128,6 +129,9 @@ class Link {
if ($queue instanceof Sending) { if ($queue instanceof Sending) {
return (int)$queue->id; return (int)$queue->id;
} }
if ($queue instanceof SendingQueueEntity) {
return $queue->getId();
}
return null; return null;
} }
} }

2
lib/Newsletter/Shortcodes/Shortcodes.php
View File

@ -5,11 +5,11 @@ namespace MailPoet\Newsletter\Shortcodes;
use MailPoet\WP\Functions as WPFunctions; use MailPoet\WP\Functions as WPFunctions;
class Shortcodes { class Shortcodes {
const SHORTCODE_CATEGORY_NAMESPACE = 'MailPoet\Newsletter\Shortcodes\Categories\\';
public $newsletter; public $newsletter;
public $subscriber; public $subscriber;
public $queue; public $queue;
public $wpUserPreview; public $wpUserPreview;
const SHORTCODE_CATEGORY_NAMESPACE = 'MailPoet\Newsletter\Shortcodes\Categories\\';
public function __construct( public function __construct(
$newsletter = false, $newsletter = false,

60
lib/Router/Endpoints/Track.php
View File

@ -3,15 +3,16 @@
namespace MailPoet\Router\Endpoints; namespace MailPoet\Router\Endpoints;
use MailPoet\Config\AccessControl; use MailPoet\Config\AccessControl;
use MailPoet\Models\Newsletter; use MailPoet\Cron\Workers\StatsNotifications\NewsletterLinkRepository;
use MailPoet\Models\NewsletterLink;
use MailPoet\Models\SendingQueue; use MailPoet\Models\SendingQueue;
use MailPoet\Models\Subscriber; use MailPoet\Models\Subscriber;
use MailPoet\Newsletter\Links\Links; use MailPoet\Newsletter\Links\Links;
use MailPoet\Newsletter\NewslettersRepository;
use MailPoet\Newsletter\Sending\SendingQueuesRepository;
use MailPoet\Statistics\Track\Clicks; use MailPoet\Statistics\Track\Clicks;
use MailPoet\Statistics\Track\Opens; use MailPoet\Statistics\Track\Opens;
use MailPoet\Subscribers\LinkTokens; use MailPoet\Subscribers\LinkTokens;
use MailPoet\Tasks\Sending as SendingTask; use MailPoet\Subscribers\SubscribersRepository;
use MailPoet\WP\Functions as WPFunctions; use MailPoet\WP\Functions as WPFunctions;
class Track { class Track {
@ -35,10 +36,34 @@ class Track {
/** @var LinkTokens */ /** @var LinkTokens */
private $linkTokens; private $linkTokens;
public function __construct(Clicks $clicks, Opens $opens, LinkTokens $linkTokens) { /** @var SendingQueuesRepository */
private $sendingQueuesRepository;
/** @var SubscribersRepository */
private $subscribersRepository;
/** @var NewslettersRepository */
private $newslettersRepository;
/** @var NewsletterLinkRepository */
private $newsletterLinkRepository;
public function __construct(
Clicks $clicks,
Opens $opens,
SendingQueuesRepository $sendingQueuesRepository,
SubscribersRepository $subscribersRepository,
NewslettersRepository $newslettersRepository,
NewsletterLinkRepository $newsletterLinkRepository,
LinkTokens $linkTokens
) {
$this->clicks = $clicks; $this->clicks = $clicks;
$this->opens = $opens; $this->opens = $opens;
$this->linkTokens = $linkTokens; $this->linkTokens = $linkTokens;
$this->sendingQueuesRepository = $sendingQueuesRepository;
$this->subscribersRepository = $subscribersRepository;
$this->newslettersRepository = $newslettersRepository;
$this->newsletterLinkRepository = $newsletterLinkRepository;
} }
public function click($data) { public function click($data) {
@ -57,34 +82,33 @@ class Track {
) { ) {
return false; return false;
} }
$data->queue = SendingQueue::findOne($data->queue_id); // phpcs:ignore Squiz.NamingConventions.ValidVariableName.NotCamelCaps $data->queue = $this->sendingQueuesRepository->findOneById($data->queue_id);// phpcs:ignore Squiz.NamingConventions.ValidVariableName.NotCamelCaps
if ($data->queue instanceof SendingQueue) { $data->subscriber = $this->subscribersRepository->findOneById($data->subscriber_id); // phpcs:ignore Squiz.NamingConventions.ValidVariableName.NotCamelCaps
$data->queue = SendingTask::createFromQueue($data->queue); $data->newsletter = $this->newslettersRepository->findOneById($data->newsletter_id); // phpcs:ignore Squiz.NamingConventions.ValidVariableName.NotCamelCaps
}
$data->subscriber = Subscriber::findOne($data->subscriber_id) ?: null; // phpcs:ignore Squiz.NamingConventions.ValidVariableName.NotCamelCaps
$data->newsletter = (!empty($data->queue->newsletter_id)) ? // phpcs:ignore Squiz.NamingConventions.ValidVariableName.NotCamelCaps
Newsletter::findOne($data->queue->newsletter_id) : // phpcs:ignore Squiz.NamingConventions.ValidVariableName.NotCamelCaps
false;
if (!empty($data->link_hash)) { // phpcs:ignore Squiz.NamingConventions.ValidVariableName.NotCamelCaps if (!empty($data->link_hash)) { // phpcs:ignore Squiz.NamingConventions.ValidVariableName.NotCamelCaps
$data->link = NewsletterLink::where('hash', $data->link_hash) // phpcs:ignore Squiz.NamingConventions.ValidVariableName.NotCamelCaps $data->link = $this->newsletterLinkRepository->findOneBy([
->where('queue_id', $data->queue_id) // phpcs:ignore Squiz.NamingConventions.ValidVariableName.NotCamelCaps 'hash' => $data->link_hash, // phpcs:ignore Squiz.NamingConventions.ValidVariableName.NotCamelCaps
->findOne(); 'queue' => $data->queue_id, // phpcs:ignore Squiz.NamingConventions.ValidVariableName.NotCamelCaps
]);
} }
return $this->_validateTrackData($data); return $this->_validateTrackData($data);
} }
public function _validateTrackData($data) { public function _validateTrackData($data) {
if (!$data->subscriber || !$data->queue || !$data->newsletter) return false; if (!$data->subscriber || !$data->queue || !$data->newsletter) return false;
$subscriberTokenMatch = $this->linkTokens->verifyToken($data->subscriber, $data->subscriber_token); // phpcs:ignore Squiz.NamingConventions.ValidVariableName.NotCamelCaps $subscriberModel = Subscriber::findOne($data->subscriber->getId());
$subscriberTokenMatch = $this->linkTokens->verifyToken($subscriberModel, $data->subscriber_token); // phpcs:ignore Squiz.NamingConventions.ValidVariableName.NotCamelCaps
if (!$subscriberTokenMatch) { if (!$subscriberTokenMatch) {
$this->terminate(403); $this->terminate(403);
} }
// return if this is a WP user previewing the newsletter // return if this is a WP user previewing the newsletter
if ($data->subscriber->isWPUser() && $data->preview) { if ($subscriberModel->isWPUser() && $data->preview) {
return $data; return $data;
} }
// check if the newsletter was sent to the subscriber // check if the newsletter was sent to the subscriber
return ($data->queue->isSubscriberProcessed($data->subscriber->id)) ? $queue = SendingQueue::findOne($data->queue_id); // phpcs:ignore Squiz.NamingConventions.ValidVariableName.NotCamelCaps
if (!$queue instanceof SendingQueue) return false;
return ($queue->isSubscriberProcessed($data->subscriber->getId())) ?
$data : $data :
false; false;
} }

28
lib/Statistics/Track/Clicks.php
View File

@ -2,6 +2,10 @@
namespace MailPoet\Statistics\Track; namespace MailPoet\Statistics\Track;
use MailPoet\Entities\NewsletterEntity;
use MailPoet\Entities\NewsletterLinkEntity;
use MailPoet\Entities\SendingQueueEntity;
use MailPoet\Entities\SubscriberEntity;
use MailPoet\Models\StatisticsClicks; use MailPoet\Models\StatisticsClicks;
use MailPoet\Newsletter\Shortcodes\Categories\Link; use MailPoet\Newsletter\Shortcodes\Categories\Link;
use MailPoet\Newsletter\Shortcodes\Shortcodes; use MailPoet\Newsletter\Shortcodes\Shortcodes;
@ -35,19 +39,23 @@ class Clicks {
if (!$data || empty($data->link)) { if (!$data || empty($data->link)) {
return $this->abort(); return $this->abort();
} }
/** @var SubscriberEntity $subscriber */
$subscriber = $data->subscriber; $subscriber = $data->subscriber;
/** @var SendingQueueEntity $queue */
$queue = $data->queue; $queue = $data->queue;
/** @var NewsletterEntity $newsletter */
$newsletter = $data->newsletter; $newsletter = $data->newsletter;
/** @var NewsletterLinkEntity $link */
$link = $data->link; $link = $data->link;
$wpUserPreview = ($data->preview && $subscriber->isWPUser()); $wpUserPreview = ($data->preview && ($subscriber->getWpUserId() > 0));
// log statistics only if the action did not come from // log statistics only if the action did not come from
// a WP user previewing the newsletter // a WP user previewing the newsletter
if (!$wpUserPreview) { if (!$wpUserPreview) {
$statisticsClicks = StatisticsClicks::createOrUpdateClickCount( $statisticsClicks = StatisticsClicks::createOrUpdateClickCount(
$link->id, $link->getId(),
$subscriber->id, $subscriber->getId(),
$newsletter->id, $newsletter->getId(),
$queue->id $queue->getId()
); );
$this->sendRevenueCookie($statisticsClicks); $this->sendRevenueCookie($statisticsClicks);
$this->sendAbandonedCartCookie($subscriber); $this->sendAbandonedCartCookie($subscriber);
@ -55,7 +63,7 @@ class Clicks {
$openEvent = new Opens(); $openEvent = new Opens();
$openEvent->track($data, $displayImage = false); $openEvent->track($data, $displayImage = false);
} }
$url = $this->processUrl($link->url, $newsletter, $subscriber, $queue, $wpUserPreview); $url = $this->processUrl($link->getUrl(), $newsletter, $subscriber, $queue, $wpUserPreview);
$this->redirectToUrl($url); $this->redirectToUrl($url);
} }
@ -90,7 +98,13 @@ class Clicks {
} }
} }
public function processUrl($url, $newsletter, $subscriber, $queue, $wpUserPreview) { public function processUrl(
string $url,
NewsletterEntity $newsletter,
SubscriberEntity $subscriber,
SendingQueueEntity $queue,
bool $wpUserPreview
) {
if (preg_match('/\[link:(?P<action>.*?)\]/', $url, $shortcode)) { if (preg_match('/\[link:(?P<action>.*?)\]/', $url, $shortcode)) {
if (!$shortcode['action']) $this->abort(); if (!$shortcode['action']) $this->abort();
$url = Link::processShortcodeAction( $url = Link::processShortcodeAction(

14
lib/Statistics/Track/Opens.php
View File

@ -2,6 +2,9 @@
namespace MailPoet\Statistics\Track; namespace MailPoet\Statistics\Track;
use MailPoet\Entities\NewsletterEntity;
use MailPoet\Entities\SendingQueueEntity;
use MailPoet\Entities\SubscriberEntity;
use MailPoet\Models\StatisticsOpens; use MailPoet\Models\StatisticsOpens;
class Opens { class Opens {
@ -9,17 +12,20 @@ class Opens {
if (!$data) { if (!$data) {
return $this->returnResponse($displayImage); return $this->returnResponse($displayImage);
} }
/** @var SubscriberEntity $subscriber */
$subscriber = $data->subscriber; $subscriber = $data->subscriber;
/** @var SendingQueueEntity $queue */
$queue = $data->queue; $queue = $data->queue;
/** @var NewsletterEntity $newsletter */
$newsletter = $data->newsletter; $newsletter = $data->newsletter;
$wpUserPreview = ($data->preview && $subscriber->isWPUser()); $wpUserPreview = ($data->preview && ($subscriber->getWpUserId() > 0));
// log statistics only if the action did not come from // log statistics only if the action did not come from
// a WP user previewing the newsletter // a WP user previewing the newsletter
if (!$wpUserPreview) { if (!$wpUserPreview) {
StatisticsOpens::getOrCreate( StatisticsOpens::getOrCreate(
$subscriber->id, $subscriber->getId(),
$newsletter->id, $newsletter->getId(),
$queue->id $queue->getId()
); );
} }
return $this->returnResponse($displayImage); return $this->returnResponse($displayImage);

102
tests/integration/Router/Endpoints/TrackTest.php
View File

@ -3,20 +3,19 @@
namespace MailPoet\Test\Router\Endpoints; namespace MailPoet\Test\Router\Endpoints;
use Codeception\Stub; use Codeception\Stub;
use MailPoet\Entities\NewsletterEntity;
use MailPoet\Entities\NewsletterLinkEntity;
use MailPoet\Entities\ScheduledTaskEntity;
use MailPoet\Entities\ScheduledTaskSubscriberEntity;
use MailPoet\Entities\SendingQueueEntity;
use MailPoet\Entities\SubscriberEntity;
use MailPoet\Models\Newsletter; use MailPoet\Models\Newsletter;
use MailPoet\Models\NewsletterLink; use MailPoet\Models\NewsletterLink;
use MailPoet\Models\ScheduledTask;
use MailPoet\Models\ScheduledTaskSubscriber;
use MailPoet\Models\SendingQueue; use MailPoet\Models\SendingQueue;
use MailPoet\Models\Subscriber; use MailPoet\Models\Subscriber;
use MailPoet\Router\Endpoints\Track; use MailPoet\Router\Endpoints\Track;
use MailPoet\Settings\SettingsController;
use MailPoet\Statistics\Track\Clicks;
use MailPoet\Statistics\Track\Opens;
use MailPoet\Subscribers\LinkTokens; use MailPoet\Subscribers\LinkTokens;
use MailPoet\Tasks\Sending as SendingTask; use MailPoet\Tasks\Sending as SendingTask;
use MailPoet\Util\Cookies;
use MailPoetVendor\Idiorm\ORM;
class TrackTest extends \MailPoetTest { class TrackTest extends \MailPoetTest {
public $track; public $track;
@ -29,40 +28,48 @@ class TrackTest extends \MailPoetTest {
public function _before() { public function _before() {
parent::_before(); parent::_before();
// create newsletter // create newsletter
$newsletter = Newsletter::create(); $newsletter = new NewsletterEntity();
$newsletter->type = 'type'; $newsletter->setType('type');
$this->newsletter = $newsletter->save(); $newsletter->setSubject('Subject');
$this->newsletter = $newsletter;
$this->entityManager->persist($newsletter);
// create subscriber // create subscriber
$subscriber = Subscriber::create(); $subscriber = new SubscriberEntity();
$subscriber->email = 'test@example.com'; $subscriber->setEmail('test@example.com');
$subscriber->firstName = 'First'; $subscriber->setFirstName('First');
$subscriber->lastName = 'Last'; $subscriber->setLastName('Last');
$this->subscriber = $subscriber->save(); $subscriber->setLinkToken('token');
$this->subscriber = $subscriber;
$this->entityManager->persist($subscriber);
// create queue // create queue
$queue = SendingTask::create(); $task = new ScheduledTaskEntity();
$queue->newsletterId = $newsletter->id; $task->setType('sending');
$queue->setSubscribers([$subscriber->id]); $this->entityManager->persist($task);
$queue->updateProcessedSubscribers([$subscriber->id]); $queue = new SendingQueueEntity();
$this->queue = $queue->save(); $queue->setTask($task);
$queue->setNewsletter($newsletter);
$this->queue = $queue;
$this->entityManager->persist($queue);
// create link // create link
$link = NewsletterLink::create(); $link = new NewsletterLinkEntity($newsletter, $queue, 'url', 'hash');
$link->hash = 'hash'; $this->link = $link;
$link->url = 'url'; $this->entityManager->persist($link);
$link->newsletterId = $newsletter->id; $this->entityManager->flush();
$link->queueId = $queue->id; $subscriberModel = Subscriber::findOne($subscriber->getId());
$this->link = $link->save();
$linkTokens = new LinkTokens; $linkTokens = new LinkTokens;
// build track data // build track data
$this->trackData = [ $this->trackData = [
'queue_id' => $queue->id, 'queue_id' => $queue->getId(),
'subscriber_id' => $subscriber->id, 'subscriber_id' => $subscriber->getId(),
'newsletter_id' => $newsletter->id, 'newsletter_id' => $newsletter->getId(),
'subscriber_token' => $linkTokens->getToken($subscriber), 'subscriber_token' => $linkTokens->getToken($subscriberModel),
'link_hash' => $link->hash, 'link_hash' => $link->getHash(),
'preview' => false, 'preview' => false,
]; ];
$queue = SendingTask::createFromQueue(SendingQueue::findOne($queue->getId()));
$queue->updateProcessedSubscribers([$subscriberModel->id]);
// instantiate class // instantiate class
$this->track = new Track(new Clicks(SettingsController::getInstance(), new Cookies()), new Opens(), new LinkTokens()); $this->track = $this->diContainer->get(Track::class);
} }
public function testItReturnsFalseWhenTrackDataIsMissing() { public function testItReturnsFalseWhenTrackDataIsMissing() {
@ -89,7 +96,8 @@ class TrackTest extends \MailPoetTest {
'newsletter' => $this->newsletter, 'newsletter' => $this->newsletter,
] ]
); );
$data->subscriber->email = 'random@email.com'; $data->subscriber->setEmail('random@email.com');
$this->entityManager->flush();
$track = Stub::make(Track::class, [ $track = Stub::make(Track::class, [
'linkTokens' => new LinkTokens, 'linkTokens' => new LinkTokens,
'terminate' => function($code) { 'terminate' => function($code) {
@ -106,9 +114,17 @@ class TrackTest extends \MailPoetTest {
'queue' => $this->queue, 'queue' => $this->queue,
'subscriber' => $this->subscriber, 'subscriber' => $this->subscriber,
'newsletter' => $this->newsletter, 'newsletter' => $this->newsletter,
'subscriber_token' => $this->subscriber->getLinkToken(),
] ]
); );
$data->subscriber->id = 99; $subscriber = new SubscriberEntity();
$subscriber->setEmail('test1@example.com');
$subscriber->setFirstName('First');
$subscriber->setLastName('Last');
$subscriber->setLinkToken($this->subscriber->getLinkToken());
$this->entityManager->persist($subscriber);
$this->entityManager->flush();
$data->subscriber->setId($subscriber->getId());
expect($this->track->_validateTrackData($data))->false(); expect($this->track->_validateTrackData($data))->false();
} }
@ -121,7 +137,8 @@ class TrackTest extends \MailPoetTest {
'newsletter' => $this->newsletter, 'newsletter' => $this->newsletter,
] ]
); );
$data->subscriber->wp_user_id = 99; $this->subscriber->setWpUserId(99);
$this->entityManager->flush();
$data->preview = true; $data->preview = true;
expect($this->track->_validateTrackData($data))->equals($data); expect($this->track->_validateTrackData($data))->equals($data);
} }
@ -138,7 +155,7 @@ class TrackTest extends \MailPoetTest {
$data = $this->trackData; $data = $this->trackData;
$data['newsletter_id'] = false; $data['newsletter_id'] = false;
$processedData = $this->track->_processTrackData($data); $processedData = $this->track->_processTrackData($data);
expect($processedData->newsletter->id)->equals($this->newsletter->id); expect($processedData->newsletter->getId())->equals($this->newsletter->getId());
} }
public function testItProcessesTrackData() { public function testItProcessesTrackData() {
@ -179,12 +196,11 @@ class TrackTest extends \MailPoetTest {
} }
public function _after() { public function _after() {
ORM::raw_execute('TRUNCATE ' . Newsletter::$_table); $this->truncateEntity(NewsletterEntity::class);
ORM::raw_execute('TRUNCATE ' . Subscriber::$_table); $this->truncateEntity(SubscriberEntity::class);
ORM::raw_execute('TRUNCATE ' . NewsletterLink::$_table); $this->truncateEntity(NewsletterLinkEntity::class);
ORM::raw_execute('TRUNCATE ' . ScheduledTask::$_table); $this->truncateEntity(ScheduledTaskEntity::class);
ORM::raw_execute('TRUNCATE ' . ScheduledTaskSubscriber::$_table); $this->truncateEntity(ScheduledTaskSubscriberEntity::class);
ORM::raw_execute('TRUNCATE ' . SendingQueue::$_table); $this->truncateEntity(SendingQueueEntity::class);
ORM::raw_execute('TRUNCATE ' . SendingQueue::$_table);
} }
} }

80
tests/integration/Statistics/Track/ClicksTest.php
View File

@ -4,9 +4,13 @@ namespace MailPoet\Test\Statistics\Track;
use Codeception\Stub; use Codeception\Stub;
use Codeception\Stub\Expected; use Codeception\Stub\Expected;
use MailPoet\Models\Newsletter; use MailPoet\Entities\NewsletterEntity;
use MailPoet\Models\NewsletterLink; use MailPoet\Entities\NewsletterLinkEntity;
use MailPoet\Models\ScheduledTask; use MailPoet\Entities\ScheduledTaskEntity;
use MailPoet\Entities\SendingQueueEntity;
use MailPoet\Entities\StatisticsClickEntity;
use MailPoet\Entities\StatisticsOpenEntity;
use MailPoet\Entities\SubscriberEntity;
use MailPoet\Models\SendingQueue; use MailPoet\Models\SendingQueue;
use MailPoet\Models\StatisticsClicks; use MailPoet\Models\StatisticsClicks;
use MailPoet\Models\StatisticsOpens; use MailPoet\Models\StatisticsOpens;
@ -16,7 +20,6 @@ use MailPoet\Statistics\Track\Clicks;
use MailPoet\Subscribers\LinkTokens; use MailPoet\Subscribers\LinkTokens;
use MailPoet\Tasks\Sending as SendingTask; use MailPoet\Tasks\Sending as SendingTask;
use MailPoet\Util\Cookies; use MailPoet\Util\Cookies;
use MailPoetVendor\Idiorm\ORM;
class ClicksTest extends \MailPoetTest { class ClicksTest extends \MailPoetTest {
public $trackData; public $trackData;
@ -32,40 +35,48 @@ class ClicksTest extends \MailPoetTest {
public function _before() { public function _before() {
parent::_before(); parent::_before();
$this->cleanup();
// create newsletter // create newsletter
$newsletter = Newsletter::create(); $newsletter = new NewsletterEntity();
$newsletter->type = 'type'; $newsletter->setType('type');
$newsletter->subject = 'Subject'; $newsletter->setSubject('Subject');
$this->newsletter = $newsletter->save(); $this->newsletter = $newsletter;
$this->entityManager->persist($newsletter);
// create subscriber // create subscriber
$subscriber = Subscriber::create(); $subscriber = new SubscriberEntity();
$subscriber->email = 'test@example.com'; $subscriber->setEmail('test@example.com');
$subscriber->firstName = 'First'; $subscriber->setFirstName('First');
$subscriber->lastName = 'Last'; $subscriber->setLastName('Last');
$this->subscriber = $subscriber->save(); $this->subscriber = $subscriber;
$this->entityManager->persist($subscriber);
// create queue // create queue
$queue = SendingTask::create(); $task = new ScheduledTaskEntity();
$queue->newsletterId = $newsletter->id; $task->setType('sending');
$queue->setSubscribers([$subscriber->id]); $this->entityManager->persist($task);
$queue->updateProcessedSubscribers([$subscriber->id]); $queue = new SendingQueueEntity();
$this->queue = $queue->save(); $queue->setTask($task);
$queue->setNewsletter($newsletter);
$this->queue = $queue;
$this->entityManager->persist($queue);
// create link // create link
$link = NewsletterLink::create(); $link = new NewsletterLinkEntity($newsletter, $queue, 'url', 'hash');
$link->hash = 'hash'; $this->link = $link;
$link->url = 'url'; $this->entityManager->persist($link);
$link->newsletterId = $newsletter->id; $this->entityManager->flush();
$link->queueId = $queue->id; $subscriberModel = Subscriber::findOne($subscriber->getId());
$this->link = $link->save();
$linkTokens = new LinkTokens; $linkTokens = new LinkTokens;
// build track data // build track data
$this->trackData = (object)[ $this->trackData = (object)[
'queue' => $queue, 'queue' => $queue,
'subscriber' => $subscriber, 'subscriber' => $subscriber,
'newsletter' => $newsletter, 'newsletter' => $newsletter,
'subscriber_token' => $linkTokens->getToken($subscriber), 'subscriber_token' => $linkTokens->getToken($subscriberModel),
'link' => $link, 'link' => $link,
'preview' => false, 'preview' => false,
]; ];
$queue = SendingTask::createFromQueue(SendingQueue::findOne($queue->getId()));
$queue->updateProcessedSubscribers([$subscriberModel->id]);
// instantiate class // instantiate class
$this->settingsController = Stub::makeEmpty(SettingsController::class, [ $this->settingsController = Stub::makeEmpty(SettingsController::class, [
'get' => false, 'get' => false,
@ -88,7 +99,8 @@ class ClicksTest extends \MailPoetTest {
public function testItDoesNotTrackEventsFromWpUserWhenPreviewIsEnabled() { public function testItDoesNotTrackEventsFromWpUserWhenPreviewIsEnabled() {
$data = $this->trackData; $data = $this->trackData;
$data->subscriber->wp_user_id = 99; $this->subscriber->setWpUserId(99);
$this->entityManager->flush();
$data->preview = true; $data->preview = true;
$clicks = Stub::construct($this->clicks, [$this->settingsController, new Cookies()], [ $clicks = Stub::construct($this->clicks, [$this->settingsController, new Cookies()], [
'redirectToUrl' => null, 'redirectToUrl' => null,
@ -183,13 +195,13 @@ class ClicksTest extends \MailPoetTest {
expect($link)->equals('http://example.com/?email=test@example.com&newsletter_subject=Subject'); expect($link)->equals('http://example.com/?email=test@example.com&newsletter_subject=Subject');
} }
public function _after() { public function cleanup() {
ORM::raw_execute('TRUNCATE ' . Newsletter::$_table); $this->truncateEntity(NewsletterEntity::class);
ORM::raw_execute('TRUNCATE ' . Subscriber::$_table); $this->truncateEntity(SubscriberEntity::class);
ORM::raw_execute('TRUNCATE ' . NewsletterLink::$_table); $this->truncateEntity(NewsletterLinkEntity::class);
ORM::raw_execute('TRUNCATE ' . ScheduledTask::$_table); $this->truncateEntity(ScheduledTaskEntity::class);
ORM::raw_execute('TRUNCATE ' . SendingQueue::$_table); $this->truncateEntity(SendingQueueEntity::class);
ORM::raw_execute('TRUNCATE ' . StatisticsOpens::$_table); $this->truncateEntity(StatisticsOpenEntity::class);
ORM::raw_execute('TRUNCATE ' . StatisticsClicks::$_table); $this->truncateEntity(StatisticsClickEntity::class);
} }
} }