Use Doctrine model in LinkTokensTest
[MAILPOET-3269]
This commit is contained in:
@@ -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;
|
||||
}
|
||||
}
|
||||
|
Reference in New Issue
Block a user