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

Remove more uses of the old models inside FormsTest

Browse Source 
[MAILPOET-3036]
This commit is contained in:
Rodrigo Primo
2021-03-30 15:12:53 -03:00
committed by Veljko V
parent 4b5f044f6b
commit faf714e832
1 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
tests/integration/API/JSON/v1/FormsTest.php
View File

@ -10,7 +10,7 @@ use MailPoet\Entities\SegmentEntity;
use MailPoet\Form\FormsRepository;
use MailPoet\Form\PreviewPage;
use MailPoet\Models\Form;
use MailPoet\Models\Segment;
use MailPoet\Segments\SegmentsRepository;
use MailPoet\WP\Functions as WPFunctions;
class FormsTest extends \MailPoetTest {
@ -35,8 +35,8 @@ class FormsTest extends \MailPoetTest {
$this->form1 = $this->createForm('Form 1');
$this->form2 = $this->createForm('Form 2');
$this->form3 = $this->createForm('Form 3');
Segment::createOrUpdate(['name' => 'Segment 1']);
Segment::createOrUpdate(['name' => 'Segment 2']);
$this->createSegment('Segment 1');
$this->createSegment('Segment 2');
}
public function testItCanGetAForm() {
@ -194,10 +194,10 @@ class FormsTest extends \MailPoetTest {
public function testItCanRestoreAForm() {
$this->form1->setDeletedAt(new \DateTime());
$this->formsRepository->flush();
$this->entityManager->refresh($this->form1);
$trashedForm = Form::findOne($this->form1->getId());
assert($trashedForm instanceof Form);
expect($trashedForm->deletedAt)->notNull();
assert($this->form1 instanceof FormEntity);
expect($this->form1)->notNull();
$response = $this->endpoint->restore(['id' => $this->form1->getId()]);
expect($response->status)->equals(APIResponse::STATUS_OK);
@ -301,6 +301,13 @@ class FormsTest extends \MailPoetTest {
return $form;
}
private function createSegment(string $name) {
$segmentsRepository = ContainerWrapper::getInstance()->get(SegmentsRepository::class);
$segment = new SegmentEntity($name, SegmentEntity::TYPE_DEFAULT, 'Some description');
$segmentsRepository->persist($segment);
$segmentsRepository->flush();
}
private function reloadForm(int $id): Form {
$reloaded = Form::findOne($id);
assert($reloaded instanceof Form);