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

Merge pull request #1705 from mailpoet/open-stats-page-test

Browse Source 
Add methods for sent newsletters to data factory [MQ-162]
This commit is contained in:
M. Shull
2018-12-20 12:11:12 -05:00
committed by GitHub
parent da27680224 0d5e094481
commit 92f2424d9d
1 changed files with 26 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
tests/DataFactories/Newsletter.php
View File

@@ -3,6 +3,8 @@ namespace MailPoet\Test\DataFactories;
use Carbon\Carbon;
use MailPoet\Models\NewsletterSegment;
use MailPoet\Models\ScheduledTask;
use MailPoet\Tasks\Sending as SendingTask;
class Newsletter {
@@ -24,6 +26,7 @@ class Newsletter {
];
$this->options = [];
$this->segments = [];
$this->queue_options = [];
$this->loadBodyFrom('newsletterWithALC.json');
}
@@ -48,6 +51,11 @@ class Newsletter {
return $this;
}
public function withSentStatus() {
$this->data['status'] = 'sent';
return $this;
}
public function withImmediateSendingSettings() {
$this->withOptions([
8 => 'immediately', # intervalType
@@ -120,6 +128,16 @@ class Newsletter {
return $this;
}
public function withSendingQueue(array $options = []) {
$this->queue_options = [
'status' => ScheduledTask::STATUS_COMPLETED,
'count_processed' => 1,
'count_total' => 1,
];
$this->queue_options = array_merge($this->queue_options, $options);
return $this;
}
/**
* @return \MailPoet\Models\Newsletter
*/
@@ -140,6 +158,14 @@ class Newsletter {
'segment_id' => $segment_id,
]);
}
if($this->queue_options) {
$sending_task = SendingTask::create();
$sending_task->newsletter_id = $newsletter->id;
$sending_task->status = $this->queue_options['status'];
$sending_task->count_processed = $this->queue_options['count_processed'];
$sending_task->count_total = $this->queue_options['count_total'];
$sending_task->save();
}
return $newsletter;
}
}