Cavemanon/piratepoet
7
Fork 0
Code Pull Requests Actions Releases 16 Activity

Refactor dynamic segments trash and restore

Browse Source 
[MAILPOET-3177]
This commit is contained in:
Rostislav Wolny
2021-03-04 15:58:26 +01:00
committed by Veljko V
parent de9b3689bd
commit e1eea57d32
2 changed files with 38 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
lib/API/JSON/v1/DynamicSegments.php
View File

@ -120,49 +120,45 @@ class DynamicSegments extends APIEndpoint {
}
public function trash($data = []) {
if (isset($data['id'])) {
$id = (int)$data['id'];
} else {
if (!isset($data['id'])) {
return $this->errorResponse([
Error::BAD_REQUEST => WPFunctions::get()->__('Missing mandatory argument `id`.', 'mailpoet'),
]);
}
try {
$segment = $this->dynamicSegmentsLoader->load($id);
$segment->trash();
return $this->successResponse(
$segment->asArray(),
['count' => 1]
);
} catch (\InvalidArgumentException $e) {
$segment = $this->getSegment($data);
if ($segment === null) {
return $this->errorResponse([
Error::NOT_FOUND => WPFunctions::get()->__('This segment does not exist.', 'mailpoet'),
]);
}
$this->segmentsRepository->bulkTrash([$segment->getId()], SegmentEntity::TYPE_DYNAMIC);
return $this->successResponse(
$this->segmentsResponseBuilder->build($segment),
['count' => 1]
);
}
public function restore($data = []) {
if (isset($data['id'])) {
$id = (int)$data['id'];
} else {
if (!isset($data['id'])) {
return $this->errorResponse([
Error::BAD_REQUEST => WPFunctions::get()->__('Missing mandatory argument `id`.', 'mailpoet'),
]);
}
try {
$segment = $this->dynamicSegmentsLoader->load($id);
$segment->restore();
return $this->successResponse(
$segment->asArray(),
['count' => 1]
);
} catch (\InvalidArgumentException $e) {
$segment = $this->getSegment($data);
if ($segment === null) {
return $this->errorResponse([
Error::NOT_FOUND => WPFunctions::get()->__('This segment does not exist.', 'mailpoet'),
]);
}
$this->segmentsRepository->bulkRestore([$segment->getId()], SegmentEntity::TYPE_DYNAMIC);
return $this->successResponse(
$this->segmentsResponseBuilder->build($segment),
['count' => 1]
);
}
public function delete($data = []) {
@ -210,4 +206,10 @@ class DynamicSegments extends APIEndpoint {
]);
}
}
private function getSegment(array $data): ?SegmentEntity {
return isset($data['id'])
? $this->segmentsRepository->findOneById((int)$data['id'])
: null;
}
}

50
tests/integration/API/JSON/v1/DynamicSegmentsTest.php
View File

@ -107,55 +107,33 @@ class DynamicSegmentsTest extends \MailPoetTest {
}
public function testItCanTrashASegment() {
DynamicSegment::deleteMany();
$dynamicSegment = DynamicSegment::createOrUpdate([
'name' => 'Trash test',
'description' => 'description',
]);
$loader = Stub::makeEmpty('\MailPoet\DynamicSegments\Persistence\Loading\SingleSegmentLoader', [
'load' => function () use($dynamicSegment) {
return $dynamicSegment;
},
]);
$dynamicSegment = $this->createDynamicSegmentEntity('Trash test', 'description');
$endpoint = new DynamicSegments($this->bulkAction, $this->listingHandler, $this->listingRepository, $this->responseBuilder, $this->segmentsRepository, $this->saveController, $loader);
$response = $endpoint->trash(['id' => $dynamicSegment->id]);
$endpoint = new DynamicSegments($this->bulkAction, $this->listingHandler, $this->listingRepository, $this->responseBuilder, $this->segmentsRepository, $this->saveController);
$response = $endpoint->trash(['id' => $dynamicSegment->getId()]);
expect($response->status)->equals(self::SUCCESS_RESPONSE_CODE);
expect($response->data)->equals($dynamicSegment->asArray());
expect($response->data['name'])->equals($dynamicSegment->getName());
expect($response->meta['count'])->equals(1);
$dynamicSegment = DynamicSegment::findOne($dynamicSegment->id);
assert($dynamicSegment instanceof DynamicSegment);
expect($dynamicSegment->deletedAt)->notNull();
$dynamicSegment->delete();
$this->entityManager->refresh($dynamicSegment);
assert($dynamicSegment instanceof SegmentEntity);
expect($dynamicSegment->getDeletedAt())->notNull();
}
public function testItCanRestoreASegment() {
DynamicSegment::deleteMany();
$dynamicSegment = DynamicSegment::createOrUpdate([
'name' => 'Restore test',
'description' => 'description',
]);
$loader = Stub::makeEmpty('\MailPoet\DynamicSegments\Persistence\Loading\SingleSegmentLoader', [
'load' => function () use($dynamicSegment) {
return $dynamicSegment;
},
]);
$dynamicSegment = $this->createDynamicSegmentEntity('Trash test', 'description');
$endpoint = new DynamicSegments($this->bulkAction, $this->listingHandler, $this->listingRepository, $this->responseBuilder, $this->segmentsRepository, $this->saveController, $loader);
$response = $endpoint->restore(['id' => $dynamicSegment->id]);
$endpoint = new DynamicSegments($this->bulkAction, $this->listingHandler, $this->listingRepository, $this->responseBuilder, $this->segmentsRepository, $this->saveController);
$response = $endpoint->restore(['id' => $dynamicSegment->getId()]);
expect($response->status)->equals(self::SUCCESS_RESPONSE_CODE);
expect($response->data)->equals($dynamicSegment->asArray());
expect($response->data['name'])->equals($dynamicSegment->getName());
expect($response->meta['count'])->equals(1);
$dynamicSegment = DynamicSegment::findOne($dynamicSegment->id);
assert($dynamicSegment instanceof DynamicSegment);
expect($dynamicSegment->deletedAt)->equals(null);
$dynamicSegment->delete();
$this->entityManager->refresh($dynamicSegment);
assert($dynamicSegment instanceof SegmentEntity);
expect($dynamicSegment->getDeletedAt())->null();
}
public function testItCanDeleteASegment() {