piratepoet/tests/unit/Models/SubscriberCest.php

<?php
use MailPoet\Models\CustomField;
use MailPoet\Models\Segment;
use MailPoet\Models\Subscriber;
use MailPoet\Models\SubscriberCustomField;
use MailPoet\Models\SubscriberSegment;

class SubscriberCest {

  function _before() {
    $this->data = array(
      'first_name' => 'John',
      'last_name' => 'Mailer',
      'email' => 'john@mailpoet.com'
    );
    $this->subscriber = Subscriber::create();
    $this->subscriber->hydrate($this->data);
    $this->saved = $this->subscriber->save();
  }

  function itCanBeCreated() {
    expect($this->saved)->equals(true);
  }

  function itHasFirstName() {
    $subscriber =
      Subscriber::where('email', $this->data['email'])
        ->findOne();
    expect($subscriber->first_name)
      ->equals($this->data['first_name']);
  }

  function itHasLastName() {
    $subscriber =
      Subscriber::where('email', $this->data['email'])
        ->findOne();
    expect($subscriber->last_name)
      ->equals($this->data['last_name']);
  }

  function itHasEmail() {
    $subscriber =
      Subscriber::where('email', $this->data['email'])
        ->findOne();
    expect($subscriber->email)
      ->equals($this->data['email']);
  }

  function emailMustBeUnique() {
    $conflict_subscriber = Subscriber::create();
    $conflict_subscriber->hydrate($this->data);
    $saved = $conflict_subscriber->save();
    expect($saved)->notEquals(true);
  }

  function itHasStatus() {
    $subscriber =
      Subscriber::where('email', $this->data['email'])
        ->findOne();
    expect($subscriber->status)->equals('unconfirmed');
  }

  function itCanChangeStatus() {
    $subscriber = Subscriber::where('email', $this->data['email'])
      ->findOne();
    $subscriber->status = 'subscribed';
    expect($subscriber->save())->equals(true);
    $subscriber_updated = Subscriber::where(
      'email',
      $this->data['email']
    )
      ->findOne();
    expect($subscriber_updated->status)->equals('subscribed');
  }

  function itHasSearchFilter() {
    $subscriber = Subscriber::filter('search', 'john')
      ->findOne();
    expect($subscriber->first_name)->equals($this->data['first_name']);
    $subscriber = Subscriber::filter('search', 'mailer')
      ->findOne();
    expect($subscriber->last_name)->equals($this->data['last_name']);
    $subscriber = Subscriber::filter('search', 'mailpoet')
      ->findOne();
    expect($subscriber->email)->equals($this->data['email']);
  }

  function itHasGroupFilter() {
    $subscribers = Subscriber::filter('groupBy', 'unconfirmed')
      ->findMany();
    foreach ($subscribers as $subscriber) {
      expect($subscriber->status)->equals('unconfirmed');
    }
    $subscribers = Subscriber::filter('groupBy', 'subscribed')
      ->findMany();
    foreach ($subscribers as $subscriber) {
      expect($subscriber->status)->equals('subscribed');
    }
    $subscribers = Subscriber::filter('groupBy', 'unsubscribed')
      ->findMany();
    foreach ($subscribers as $subscriber) {
      expect($subscriber->status)->equals('unsubscribed');
    }
  }

  function itCanHaveSegment() {
    $segmentData = array(
      'name' => 'some name'
    );
    $segment = Segment::create();
    $segment->hydrate($segmentData);
    $segment->save();
    $association = SubscriberSegment::create();
    $association->subscriber_id = $this->subscriber->id;
    $association->segment_id = $segment->id;
    $association->save();
    $subscriber = Subscriber::findOne($this->subscriber->id);
    $subscriberSegment = $subscriber->segments()
      ->findOne();
    expect($subscriberSegment->id)->equals($segment->id);
  }
  
  function itCanHaveCustomField() {
    $customFieldData = array(
      'name' => 'city'
    );
    $customField = CustomField::create();
    $customField->hydrate($customFieldData);
    $customField->save();
    $association = SubscriberCustomField::create();
    $association->subscriber_id = $this->subscriber->id;
    $association->custom_field_id = $customField->id;
    $association->save();
    $subscriber = Subscriber::findOne($this->subscriber->id);
    $subscriberCustomField = $subscriber->customFields()
      ->findOne();
    expect($subscriberCustomField->id)->equals($customField->id);
  }

  function itCanCreateOrUpdate() {
    $data = array(
      'email' => 'john.doe@mailpoet.com',
      'first_name' => 'John',
      'last_name' => 'Doe'
    );
    $result = Subscriber::createOrUpdate($data);
    expect($result)->equals(true);
    $record = Subscriber::where('email', $data['email'])
      ->findOne();
    expect($record->first_name)->equals($data['first_name']);
    expect($record->last_name)->equals($data['last_name']);
    $record->last_name = 'Mailer';
    $result = Subscriber::createOrUpdate($record->asArray());
    expect($result)->equals(true);
    $record = Subscriber::where('email', $data['email'])
      ->findOne();
    expect($record->last_name)->equals('Mailer');
  }

  function _after() {
    ORM::forTable(Subscriber::$_table)
      ->deleteMany();
    ORM::forTable(Segment::$_table)
      ->deleteMany();
    ORM::forTable(SubscriberSegment::$_table)
      ->deleteMany();
    ORM::forTable(CustomField::$_table)
      ->deleteMany();
    ORM::forTable(SubscriberCustomField::$_table)
      ->deleteMany();
  }
}