<?php
namespace MailPoet\Test\Tasks\Subscribers;

use MailPoet\Models\ScheduledTaskSubscriber;
use MailPoet\Tasks\Subscribers\BatchIterator;

class BatchIteratorTest extends \MailPoetTest {
  function _before() {
    $this->task_id = 123;
    $this->batch_size = 2;
    $this->subscriber_count = 5;
    for($i = 0; $i < $this->subscriber_count; $i++) {
      ScheduledTaskSubscriber::createOrUpdate(array(
        'task_id' => $this->task_id,
        'subscriber_id' => $i + 1,
      ));
    }
    $this->iterator = new BatchIterator($this->task_id, $this->batch_size);
  }

  function testItFailsToConstructWithWrongArguments() {
    try {
      $iterator = new BatchIterator(0, 0);
      $this->fail('Exception was not thrown');
    } catch(\Exception $e) {
      // No exception handling necessary
    }
  }

  function testItConstructs() {
    $iterator = new BatchIterator(123, 456);
    expect_that($iterator instanceof BatchIterator);
  }

  function testItIterates() {
    $iterations = ceil($this->subscriber_count / $this->batch_size);
    $i = 0;
    foreach($this->iterator as $batch) {
      $i++;
      if($i < $iterations) {
        expect(count($batch))->equals($this->batch_size);
      } else {
        expect(count($batch))->lessOrEquals($this->batch_size);
      }
    }
    expect($i)->equals($iterations);
  }

  function testItCanBeCounted() {
    expect(count($this->iterator))->equals($this->subscriber_count);
  }

  function _after() {
    \ORM::raw_execute('TRUNCATE ' . ScheduledTaskSubscriber::$_table);
  }
}