Add tests [MAILPOET-940]

2017-07-12 18:09:05 +03:00
parent 55d3b67a2a
commit acbe2e383a
4 changed files with 159 additions and 1 deletions
--- a/tests/unit/Models/ScheduledTaskSubscriberTest.php
+++ b/tests/unit/Models/ScheduledTaskSubscriberTest.php
@ -0,0 +1,75 @@
+<?php
+namespace MailPoet\Test\Models;
+
+use MailPoet\Models\ScheduledTaskSubscriber;
+
+class ScheduledTaskSubscriberTest extends \MailPoetTest {
+  function _before() {
+    $this->task_id = 123;
+    $this->subscriber_id = 456;
+    $this->task_subscriber = ScheduledTaskSubscriber::createOrUpdate(array(
+      'task_id' => $this->task_id,
+      'subscriber_id' => $this->subscriber_id
+    ));
+  }
+
+  function testItCanBeCreated() {
+    expect($this->task_subscriber->task_id)->equals($this->task_id);
+    expect($this->task_subscriber->subscriber_id)->equals($this->subscriber_id);
+    expect($this->task_subscriber->processed)->equals(ScheduledTaskSubscriber::STATUS_TO_PROCESS);
+  }
+
+  function testItCanBeUpdated() {
+    $task_subscriber = ScheduledTaskSubscriber::createOrUpdate(array(
+      'task_id' => $this->task_id,
+      'subscriber_id' => $this->subscriber_id,
+      'processed' => ScheduledTaskSubscriber::STATUS_PROCESSED
+    ));
+    expect($task_subscriber->processed)->equals(ScheduledTaskSubscriber::STATUS_PROCESSED);
+  }
+
+  function testItCanAddMultipleSubscribers() {
+    ScheduledTaskSubscriber::deleteMany();
+    $subscriber_ids = array(321, 654, 987);
+    ScheduledTaskSubscriber::addSubscribers($this->task_id, $subscriber_ids);
+    $task_subscribers = ScheduledTaskSubscriber::where('task_id', $this->task_id)
+      ->orderByAsc('subscriber_id')
+      ->findMany();
+    expect(count($task_subscribers))->equals(count($subscriber_ids));
+    expect($task_subscribers[0]->subscriber_id)->equals($subscriber_ids[0]);
+    expect($task_subscribers[1]->subscriber_id)->equals($subscriber_ids[1]);
+    expect($task_subscribers[2]->subscriber_id)->equals($subscriber_ids[2]);
+  }
+
+  function testItCanGetToProcessCount() {
+    $count = ScheduledTaskSubscriber::getToProcessCount($this->task_id);
+    expect($count)->equals(1);
+    $this->task_subscriber->processed = ScheduledTaskSubscriber::STATUS_PROCESSED;
+    $this->task_subscriber->save();
+    $count = ScheduledTaskSubscriber::getToProcessCount($this->task_id);
+    expect($count)->equals(0);
+  }
+
+  function testItCanGetProcessedCount() {
+    $count = ScheduledTaskSubscriber::getProcessedCount($this->task_id);
+    expect($count)->equals(0);
+    $this->task_subscriber->processed = ScheduledTaskSubscriber::STATUS_PROCESSED;
+    $this->task_subscriber->save();
+    $count = ScheduledTaskSubscriber::getProcessedCount($this->task_id);
+    expect($count)->equals(1);
+  }
+
+  function testItCanGetTotalCount() {
+    ScheduledTaskSubscriber::createOrUpdate(array(
+      'task_id' => $this->task_id,
+      'subscriber_id' => 555,
+      'processed' => ScheduledTaskSubscriber::STATUS_PROCESSED
+    ));
+    $count = ScheduledTaskSubscriber::getTotalCount($this->task_id);
+    expect($count)->equals(2);
+  }
+
+  function _after() {
+    \ORM::raw_execute('TRUNCATE ' . ScheduledTaskSubscriber::$_table);
+  }
+}