From e24aa82d9549f2c9bc99aeebd1d607bb0fdfb444 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jan=20Lys=C3=BD?= <lysyjan87@gmail.com>
Date: Wed, 30 Sep 2020 08:03:23 +0200
Subject: [PATCH] Add integration test on SegmentsResponseBuilder
 [MAILPOET-3166]

---
 .../SegmentsResponseBuilderTest.php           | 32 +++++++++++++++++++
 1 file changed, 32 insertions(+)
 create mode 100644 tests/integration/API/JSON/ResponseBuilders/SegmentsResponseBuilderTest.php

diff --git a/tests/integration/API/JSON/ResponseBuilders/SegmentsResponseBuilderTest.php b/tests/integration/API/JSON/ResponseBuilders/SegmentsResponseBuilderTest.php
new file mode 100644
index 0000000000..724636992c
--- /dev/null
+++ b/tests/integration/API/JSON/ResponseBuilders/SegmentsResponseBuilderTest.php
@@ -0,0 +1,32 @@
+<?php
+
+namespace MailPoet\API\JSON\ResponseBuilders;
+
+use MailPoet\DI\ContainerWrapper;
+use MailPoet\Entities\SegmentEntity;
+use MailPoetVendor\Doctrine\ORM\EntityManager;
+
+class SegmentsResponseBuilderTest extends \MailPoetTest {
+  public function testItBuildsResponse() {
+    $name = 'Response Builder Test';
+    $description = 'Testing description';
+
+    $di = ContainerWrapper::getInstance();
+    $em = $di->get(EntityManager::class);
+    $segment = new SegmentEntity($name, SegmentEntity::TYPE_DEFAULT, $description);
+    $em->persist($segment);
+    $em->flush();
+    $responseBuilder = new SegmentsResponseBuilder();
+    $response = $responseBuilder->build($segment);
+
+    expect($response['name'])->equals($name);
+    expect($response['type'])->equals(SegmentEntity::TYPE_DEFAULT);
+    expect($response['description'])->equals($description);
+    expect($response)->hasKey('id');
+    expect($response)->hasKey('created_at');
+    expect($response)->hasKey('updated_at');
+    expect($response)->hasKey('deleted_at');
+    $em->remove($segment);
+    $em->flush();
+  }
+}