piratepoet/mailpoet/tests/integration/API/JSON/ResponseBuilders/DynamicSegmentsResponseBuilderTest.php

<?php declare(strict_types = 1);

namespace MailPoet\API\JSON\ResponseBuilders;

use MailPoet\Entities\DynamicSegmentFilterData;
use MailPoet\Entities\DynamicSegmentFilterEntity;
use MailPoet\Entities\SegmentEntity;
use MailPoet\Entities\SubscriberEntity;
use MailPoet\Entities\SubscriberSegmentEntity;
use MailPoet\Segments\DynamicSegments\Filters\UserRole;

class DynamicSegmentsResponseBuilderTest extends \MailPoetTest {
  public function _before() {
    parent::_before();
    $this->cleanup();
  }

  public function testItBuildsGetResponse() {
    $name = 'Response Listings Builder Test';
    $description = 'Testing description';
    $segment = $this->createDynamicSegmentEntity($name, $description);
    $this->addDynamicFilter($segment, 'editor');
    $this->entityManager->flush();

    /** @var DynamicSegmentsResponseBuilder $responseBuilder */
    $responseBuilder = $this->diContainer->get(DynamicSegmentsResponseBuilder::class);
    $response = $responseBuilder->build($segment);
    expect($response)->array();
    expect($response['id'])->equals($segment->getId());
    expect($response['name'])->equals($name);
    expect($response['description'])->equals($description);
    expect($response['type'])->equals(SegmentEntity::TYPE_DYNAMIC);
    expect($response)->hasKey('created_at');
    expect($response)->hasKey('updated_at');
    expect($response)->hasKey('deleted_at');
    expect($response['filters_connect'])->equals(DynamicSegmentFilterData::CONNECT_TYPE_AND);
    expect($response['filters'])->array();
    expect($response['filters'])->count(1);
    expect($response['filters'][0]['segmentType'])->equals(DynamicSegmentFilterData::TYPE_USER_ROLE);
    expect($response['filters'][0]['wordpressRole'])->equals(['editor']);
    expect($response['filters'][0]['action'])->equals(UserRole::TYPE);
  }

  public function testItBuildsGetResponseWithTwoFilters() {
    $name = 'Response Listings Builder Test';
    $description = 'Testing description';
    $segment = $this->createDynamicSegmentEntity($name, $description);
    $this->addDynamicFilter($segment, 'editor');
    $this->addDynamicFilter($segment, 'administrator');
    $this->entityManager->flush();

    /** @var DynamicSegmentsResponseBuilder $responseBuilder */
    $responseBuilder = $this->diContainer->get(DynamicSegmentsResponseBuilder::class);
    $response = $responseBuilder->build($segment);
    expect($response)->array();
    expect($response['id'])->equals($segment->getId());
    expect($response['name'])->equals($name);
    expect($response['description'])->equals($description);
    expect($response['type'])->equals(SegmentEntity::TYPE_DYNAMIC);
    expect($response)->hasKey('created_at');
    expect($response)->hasKey('updated_at');
    expect($response)->hasKey('deleted_at');
    expect($response['filters_connect'])->equals(DynamicSegmentFilterData::CONNECT_TYPE_AND);
    expect($response['filters'])->array();
    expect($response['filters'])->count(2);
    expect($response['filters'][0]['segmentType'])->equals(DynamicSegmentFilterData::TYPE_USER_ROLE);
    expect($response['filters'][0]['wordpressRole'])->equals(['editor']);
    expect($response['filters'][0]['action'])->equals(UserRole::TYPE);
    expect($response['filters'][0]['connect'])->equals(DynamicSegmentFilterData::CONNECT_TYPE_AND);
    expect($response['filters'][1]['segmentType'])->equals(DynamicSegmentFilterData::TYPE_USER_ROLE);
    expect($response['filters'][1]['wordpressRole'])->equals(['administrator']);
    expect($response['filters'][1]['action'])->equals(UserRole::TYPE);
    expect($response['filters'][1]['connect'])->equals(DynamicSegmentFilterData::CONNECT_TYPE_AND);
  }

  public function testItBuildsListingsResponse() {
    $name = 'Response Listings Builder Test';
    $description = 'Testing description';
    $wpUserEmail = 'editor1@example.com';

    $this->tester->deleteWordPressUser($wpUserEmail);
    $this->tester->createWordPressUser($wpUserEmail, 'editor');
    $wpUserSubscriber = $this->entityManager
      ->getRepository(SubscriberEntity::class)
      ->findOneBy(['email' => $wpUserEmail]);
    $this->assertInstanceOf(SubscriberEntity::class, $wpUserSubscriber);
    $wpUserSubscriber->setStatus(SubscriberEntity::STATUS_SUBSCRIBED);
    $segment = $this->createDynamicSegmentEntity($name, $description);
    $this->addDynamicFilter($segment, 'editor');
    $this->entityManager->flush();

    /** @var DynamicSegmentsResponseBuilder $responseBuilder */
    $responseBuilder = $this->diContainer->get(DynamicSegmentsResponseBuilder::class);
    $response = $responseBuilder->buildForListing([$segment]);
    expect($response)->array();
    expect($response[0]['name'])->equals($name);
    expect($response[0]['description'])->equals($description);
    expect($response[0]['type'])->equals(SegmentEntity::TYPE_DYNAMIC);
    expect($response[0]['subscribers_url'])->startsWith('http');
    expect($response[0]['count_all'])->equals(1);
    expect($response[0]['count_subscribed'])->equals(1);

    $this->tester->deleteWordPressUser($wpUserEmail);
  }

  private function createDynamicSegmentEntity(string $name, string $description): SegmentEntity {
    $segment = new SegmentEntity($name, SegmentEntity::TYPE_DYNAMIC, $description);
    $this->entityManager->persist($segment);
    return $segment;
  }

  private function addDynamicFilter(SegmentEntity $segment, string $wordpressRole): SegmentEntity {
    $dynamicFilter = new DynamicSegmentFilterEntity($segment, new DynamicSegmentFilterData(DynamicSegmentFilterData::TYPE_USER_ROLE, UserRole::TYPE, [
      'wordpressRole' => $wordpressRole,
      'connect' => DynamicSegmentFilterData::CONNECT_TYPE_AND,
    ]));
    $segment->getDynamicFilters()->add($dynamicFilter);
    $this->entityManager->persist($dynamicFilter);
    return $segment;
  }

  private function cleanup() {
    $this->truncateEntity(SegmentEntity::class);
    $this->truncateEntity(SubscriberEntity::class);
    $this->truncateEntity(SubscriberSegmentEntity::class);
    $this->truncateEntity(DynamicSegmentFilterEntity::class);
  }

  public function _after() {
    parent::_after();
    $this->cleanup();
  }
}