Add integration test for StatisticsFormsRepository

[MAILPOET-3032]

tests/integration/Statistics/StatisticsFormsRepositoryTest.php

@@ -0,0 +1,90 @@
+<?php declare(strict_types = 1);
+
+namespace MailPoet\Statistics;
+
+use MailPoet\Entities\FormEntity;
+use MailPoet\Entities\StatisticsFormEntity;
+use MailPoet\Entities\SubscriberEntity;
+
+class StatisticsFormsRepositoryTest extends \MailPoetTest {
+  /** @var StatisticsFormsRepository */
+  private $repository;
+
+  public function _before() {
+    parent::_before();
+    $this->cleanup();
+    $this->repository = $this->diContainer->get(StatisticsFormsRepository::class);
+  }
+
+  public function testItCanRecordStats(): void {
+    $form = $this->createForm();
+    $subscriber = $this->createSubscriber();
+    $record = $this->repository->record($form, $subscriber);
+    assert($record instanceof StatisticsFormEntity);
+    expect($record->getForm())->equals($form);
+    expect($record->getSubscriber())->equals($subscriber);
+    expect($this->repository->findOneBy(['form' => $form, 'subscriber' => $subscriber]))->isInstanceOf(StatisticsFormEntity::class);
+  }
+
+  public function testItDoesNotOverrideStats(): void {
+    $form = $this->createForm();
+    $subscriber = $this->createSubscriber();
+    $record = $this->repository->record($form, $subscriber);
+    assert($record instanceof StatisticsFormEntity);
+    expect($record->getForm())->equals($form);
+    expect($record->getSubscriber())->equals($subscriber);
+
+    $this->repository->record($form, $subscriber);
+    expect($this->repository->findAll())->count(1);
+  }
+
+  public function testItCanRecordMultipleStats(): void {
+    $form1 = $this->createForm();
+    $form2 = $this->createForm();
+    $subscriber1 = $this->createSubscriber();
+    $subscriber2 = $this->createSubscriber();
+    $this->repository->record($form1, $subscriber2);
+    $this->repository->record($form2, $subscriber2);
+    $this->repository->record($form1, $subscriber1);
+
+    expect($this->repository->findAll())->count(3);
+  }
+
+  public function testItCanReturnTheTotalSignupsOfAForm(): void {
+    $form1 = $this->createForm();
+    $form2 = $this->createForm();
+    $subscriber1 = $this->createSubscriber();
+    $subscriber2 = $this->createSubscriber();
+    // simulate 2 signups for form #1
+    $this->repository->record($form1, $subscriber2);
+    $this->repository->record($form1, $subscriber1);
+    // simulate 1 signup for form #2
+    $this->repository->record($form2, $subscriber2);
+
+    expect($this->repository->getTotalSignups((int)$form1->getId()))->equals(2);
+    expect($this->repository->getTotalSignups((int)$form2->getId()))->equals(1);
+    expect($this->repository->findAll())->count(3);
+  }
+
+  private function createSubscriber(): SubscriberEntity {
+    $subscriber = new SubscriberEntity();
+    $subscriber->setStatus(SubscriberEntity::STATUS_SUBSCRIBED);
+    $subscriber->setEmail('subscriber' . rand(0, 10000) . '@example.com');
+    $this->entityManager->persist($subscriber);
+    $this->entityManager->flush();
+    return $subscriber;
+  }
+
+  private function createForm(): FormEntity {
+    $form = new FormEntity('Form' . rand(0, 10000));
+    $this->entityManager->persist($form);
+    $this->entityManager->flush();
+    return $form;
+  }
+
+  private function cleanup(): void {
+    $this->truncateEntity(StatisticsFormEntity::class);
+    $this->truncateEntity(FormEntity::class);
+    $this->truncateEntity(SubscriberEntity::class);
+  }
+}