<?php

namespace MailPoet\Subscribers;

use MailPoet\Models\Subscriber;

class SourceTest extends \MailPoetTest {

  function testItDoesntOverrideSource() {
    $subscriber = Subscriber::createOrUpdate(array(
      'source' => Source::FORM,
    ));
    $updated_subscriber = Source::setSource($subscriber, Source::API);
    expect($updated_subscriber->source)->equals(Source::FORM);
  }

  function testItDoesntAllowInvalidSource() {
    $subscriber = Subscriber::createOrUpdate(array(
      'source' => Source::UNKNOWN,
    ));
    $this->setExpectedException('\InvalidArgumentException');
    Source::setSource($subscriber, 'invalid source');
  }

  function testItWorksWhenNoSourceIsSet() {
    $subscriber = Subscriber::createOrUpdate(array());
    $updated_subscriber = Source::setSource($subscriber, Source::FORM);
    expect($updated_subscriber->source)->equals(Source::FORM);
  }

  function testItWorks() {
    $subscriber = Subscriber::createOrUpdate(array(
      'source' => Source::UNKNOWN,
    ));
    $updated_subscriber = Source::setSource($subscriber, Source::FORM);
    expect($updated_subscriber->source)->equals(Source::FORM);
  }

}