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

Use Doctrine model in LinkTokensTest

Browse Source 
[MAILPOET-3269]
This commit is contained in:
Jan Lysý
2021-05-27 18:32:05 +02:00
committed by Veljko V
parent 136329b5a3
commit eb189809ae
1 changed files with 20 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
tests/integration/Subscribers/LinkTokensTest.php
View File

@@ -1,23 +1,26 @@
<?php <?php declare(strict_types = 1);
namespace MailPoet\Subscribers; namespace MailPoet\Subscribers;
use MailPoet\Models\Subscriber; use MailPoet\Entities\SubscriberEntity;
use MailPoetVendor\Idiorm\ORM;
class LinkTokensTest extends \MailPoetTest { class LinkTokensTest extends \MailPoetTest {
/** @var LinkTokens */ /** @var LinkTokens */
private $linkTokens; private $linkTokens;
/** @var SubscribersRepository */
private $subscribersRepository;
public function _before() { public function _before() {
parent::_before(); parent::_before();
$this->linkTokens = new LinkTokens; $this->subscribersRepository = $this->diContainer->get(SubscribersRepository::class);
$this->linkTokens = new LinkTokens($this->subscribersRepository);
} }
public function testItGeneratesSubscriberToken() { public function testItGeneratesSubscriberToken() {
$subscriber1 = $this->makeSubscriber(['email' => 'demo1@fake.loc']); $subscriber1 = $this->createSubscriber('demo1@fake.loc');
$subscriber2 = $this->makeSubscriber(['email' => 'demo2@fake.loc']); $subscriber2 = $this->createSubscriber('demo2@fake.loc');
$token1 = $this->linkTokens->getToken($subscriber1); $token1 = $this->linkTokens->getToken($subscriber1);
$token2 = $this->linkTokens->getToken($subscriber2); $token2 = $this->linkTokens->getToken($subscriber2);
expect(strlen($token1))->equals(6); expect(strlen($token1))->equals(6);
@@ -26,29 +29,30 @@ class LinkTokensTest extends \MailPoetTest {
} }
public function testItGetsSubscriberToken() { public function testItGetsSubscriberToken() {
$subscriber1 = $this->makeSubscriber(['email' => 'demo1@fake.loc', 'link_token' => 'already-existing-token']); $subscriber1 = $this->createSubscriber('demo1@fake.loc', 'already-existing-token');
$subscriber2 = $this->makeSubscriber(['email' => 'demo2@fake.loc']); $subscriber2 = $this->createSubscriber('demo2@fake.loc');
expect($this->linkTokens->getToken($subscriber1))->equals('already-existing-token'); expect($this->linkTokens->getToken($subscriber1))->equals('already-existing-token');
expect(strlen($this->linkTokens->getToken($subscriber2)))->equals(6); expect(strlen($this->linkTokens->getToken($subscriber2)))->equals(6);
} }
public function testItVerifiesSubscriberToken() { public function testItVerifiesSubscriberToken() {
$subscriber = $this->makeSubscriber([ $subscriber = $this->createSubscriber('demo@fake.loc');
'email' => 'demo@fake.loc',
]);
$token = $this->linkTokens->getToken($subscriber); $token = $this->linkTokens->getToken($subscriber);
expect($this->linkTokens->verifyToken($subscriber, $token))->true(); expect($this->linkTokens->verifyToken($subscriber, $token))->true();
expect($this->linkTokens->verifyToken($subscriber, 'faketoken'))->false(); expect($this->linkTokens->verifyToken($subscriber, 'faketoken'))->false();
} }
public function _after() { public function _after() {
ORM::raw_execute('TRUNCATE ' . Subscriber::$_table); parent::_after();
$this->truncateEntity(SubscriberEntity::class);
} }
private function makeSubscriber($data) { private function createSubscriber(string $email, ?string $linkToken = null): SubscriberEntity {
$subscriber = Subscriber::create(); $subscriber = new SubscriberEntity();
$subscriber->hydrate($data); $subscriber->setEmail($email);
$subscriber->save(); $subscriber->setLinkToken($linkToken);
$this->subscribersRepository->persist($subscriber);
$this->subscribersRepository->flush();
return $subscriber; return $subscriber;
} }
} }