<?php

namespace MailPoet\DynamicSegments\Persistence\Loading;

use MailPoet\DynamicSegments\Filters\UserRole;
use MailPoet\DynamicSegments\Mappers\DBMapper;
use MailPoet\Models\DynamicSegment;
use MailPoet\Models\DynamicSegmentFilter;

class LoaderTest extends \MailPoetTest {

  private $segments;

  /** @var Loader */
  private $loader;

  function _before() {
    $this->loader = new Loader(new DBMapper());
    \ORM::raw_execute('TRUNCATE ' . DynamicSegment::$_table);
    \ORM::raw_execute('TRUNCATE ' . DynamicSegmentFilter::$_table);
    $this->segments[] = DynamicSegment::createOrUpdate([
      'name' => 'segment 1',
      'description' => 'description',
    ]);
    $this->segments[] = DynamicSegment::createOrUpdate([
      'name' => 'segment 2',
      'description' => 'description',
    ]);
    $filter = new UserRole('Administrator', 'and');
    $filter_data = DynamicSegmentFilter::create();
    $filter_data->hydrate([
      'segment_id' => $this->segments[1]->id,
      'filter_data' => $filter->toArray(),
    ]);
    $filter_data->save();
    $filter = new UserRole('Editor', 'or');
    $filter_data = DynamicSegmentFilter::create();
    $filter_data->hydrate([
      'segment_id' => $this->segments[0]->id,
      'filter_data' => $filter->toArray(),
    ]);
    $filter_data->save();
  }

  function testItLoadsSegments() {
    $data = $this->loader->load();
    expect($data)->count(2);
    expect($data[0])->isInstanceOf('\MailPoet\Models\DynamicSegment');
    expect($data[1])->isInstanceOf('\MailPoet\Models\DynamicSegment');
  }

  function testItDoesNotLoadTrashedSegments() {
    $this->segments[0]->trash();
    $data = $this->loader->load();
    expect($data)->count(1);
    expect($data[0])->isInstanceOf('\MailPoet\Models\DynamicSegment');
    expect($data[0]->name)->equals('segment 2');
  }

  function testItPopulatesCommonData() {
    $data = $this->loader->load();
    expect($data[0]->name)->equals('segment 1');
    expect($data[1]->name)->equals('segment 2');
    expect($data[0]->description)->equals('description');
    expect($data[1]->description)->equals('description');
  }

  function testItPopulatesFilters() {
    $data = $this->loader->load();
    $filters0 = $data[0]->getFilters();
    $filters1 = $data[1]->getFilters();
    expect($filters0)->count(1);
    expect($filters1)->count(1);
    expect($filters0[0])->isInstanceOf('\MailPoet\DynamicSegments\Filters\UserRole');
    expect($filters1[0])->isInstanceOf('\MailPoet\DynamicSegments\Filters\UserRole');
    expect($filters0[0]->getRole())->equals('Editor');
    expect($filters1[0]->getRole())->equals('Administrator');
  }

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