Apply safe to load association trait to all toOne associations

[MAILPOET-2818]

lib/Entities/StatisticsClickEntity.php

@@ -4,6 +4,7 @@ namespace MailPoet\Entities;
 
 use MailPoet\Doctrine\EntityTraits\AutoincrementedIdTrait;
 use MailPoet\Doctrine\EntityTraits\CreatedAtTrait;
+use MailPoet\Doctrine\EntityTraits\SafeToOneAssociationLoadTrait;
 use MailPoet\Doctrine\EntityTraits\UpdatedAtTrait;
 use MailPoetVendor\Doctrine\ORM\Mapping as ORM;
 
@@ -15,6 +16,7 @@ class StatisticsClickEntity {
   use AutoincrementedIdTrait;
   use CreatedAtTrait;
   use UpdatedAtTrait;
+  use SafeToOneAssociationLoadTrait;
 
   /**
    * @ORM\ManyToOne(targetEntity="MailPoet\Entities\NewsletterEntity")
@@ -48,4 +50,27 @@ class StatisticsClickEntity {
    */
   private $count;
 
+  /**
+   * @return NewsletterEntity|null
+   */
+  public function getNewsletter() {
+    $this->safelyLoadToOneAssociation('newsletter');
+    return $this->newsletter;
+  }
+
+  /**
+   * @return SendingQueueEntity|null
+   */
+  public function getQueue() {
+    $this->safelyLoadToOneAssociation('queue');
+    return $this->queue;
+  }
+
+  /**
+   * @return NewsletterLinkEntity
+   */
+  public function getLink() {
+    $this->safelyLoadToOneAssociation('link');
+    return $this->link;
+  }
 }