Use Doctrine model in LinkTokensTest

[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

View File

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