diff --git a/tests/integration/API/JSON/v1/SetupTest.php b/tests/integration/API/JSON/v1/SetupTest.php index 0ccea31391..0e4e9631d2 100644 --- a/tests/integration/API/JSON/v1/SetupTest.php +++ b/tests/integration/API/JSON/v1/SetupTest.php @@ -6,6 +6,7 @@ use MailPoet\Config\Activator; use MailPoet\Config\Populator; use MailPoet\Models\Setting; use MailPoet\API\JSON\v1\Setup; +use MailPoet\Referrals\ReferralDetector; use MailPoet\WP\Functions as WPFunctions; use Helper\WordPressHooks as WPHooksHelper; use MailPoet\API\JSON\Response as APIResponse; @@ -25,7 +26,8 @@ class SetupTest extends \MailPoetTest { ]); $settings = new SettingsController(); - $populator = new Populator($settings, $wp, new Captcha()); + $referral_detector = new ReferralDetector($wp, $settings); + $populator = new Populator($settings, $wp, new Captcha(), $referral_detector); $router = new Setup($wp, new Activator($settings, $populator)); $response = $router->reset(); expect($response->status)->equals(APIResponse::STATUS_OK); diff --git a/tests/integration/Cron/Workers/SendingQueue/SendingQueueTest.php b/tests/integration/Cron/Workers/SendingQueue/SendingQueueTest.php index 5d9f7af34c..1b953b4b56 100644 --- a/tests/integration/Cron/Workers/SendingQueue/SendingQueueTest.php +++ b/tests/integration/Cron/Workers/SendingQueue/SendingQueueTest.php @@ -27,6 +27,7 @@ use MailPoet\Models\StatisticsNewsletters; use MailPoet\Models\Subscriber; use MailPoet\Models\SubscriberSegment; use MailPoet\Newsletter\Links\Links; +use MailPoet\Referrals\ReferralDetector; use MailPoet\Router\Endpoints\Track; use MailPoet\Router\Router; use MailPoet\Settings\SettingsController; @@ -48,7 +49,8 @@ class SendingQueueTest extends \MailPoetTest { $wp_users = get_users(); wp_set_current_user($wp_users[0]->ID); $this->settings = new SettingsController(); - $populator = new Populator($this->settings, WPFunctions::get(), new Captcha); + $referral_detector = new ReferralDetector(WPFunctions::get(), $this->settings); + $populator = new Populator($this->settings, WPFunctions::get(), new Captcha, $referral_detector); $populator->up(); $this->subscriber = Subscriber::create(); $this->subscriber->email = 'john@doe.com'; diff --git a/tests/integration/Cron/Workers/SendingQueue/Tasks/MailerTest.php b/tests/integration/Cron/Workers/SendingQueue/Tasks/MailerTest.php index 054bb0d4e4..142c2c4b5e 100644 --- a/tests/integration/Cron/Workers/SendingQueue/Tasks/MailerTest.php +++ b/tests/integration/Cron/Workers/SendingQueue/Tasks/MailerTest.php @@ -8,6 +8,7 @@ use MailPoet\Cron\Workers\SendingQueue\Tasks\Mailer as MailerTask; use MailPoet\Mailer\Mailer; use MailPoet\Models\Setting; use MailPoet\Models\Subscriber; +use MailPoet\Referrals\ReferralDetector; use MailPoet\Settings\SettingsController; use MailPoet\Subscription\Captcha; use MailPoet\WP\Functions as WPFunctions; @@ -26,7 +27,8 @@ class MailerTest extends \MailPoetTest { $wp_users = get_users(); wp_set_current_user($wp_users[0]->ID); $this->settings = new SettingsController(); - $populator = new Populator($this->settings, WPFunctions::get(), new Captcha); + $referral_detector = new ReferralDetector(WPFunctions::get(), $this->settings); + $populator = new Populator($this->settings, WPFunctions::get(), new Captcha, $referral_detector); $populator->up(); $this->mailer_task = new MailerTask(); $this->sender = $this->settings->get('sender'); diff --git a/tests/integration/Newsletter/ShortcodesTest.php b/tests/integration/Newsletter/ShortcodesTest.php index 2874b17fde..a4b054124f 100644 --- a/tests/integration/Newsletter/ShortcodesTest.php +++ b/tests/integration/Newsletter/ShortcodesTest.php @@ -9,6 +9,7 @@ use MailPoet\Models\Subscriber; use MailPoet\Models\SubscriberCustomField; use MailPoet\Newsletter\Shortcodes\Categories\Date; use MailPoet\Newsletter\Url as NewsletterUrl; +use MailPoet\Referrals\ReferralDetector; use MailPoet\Settings\SettingsController; use MailPoet\Subscription\Captcha; use MailPoet\Subscription\Url as SubscriptionUrl; @@ -26,7 +27,8 @@ class ShortcodesTest extends \MailPoetTest { function _before() { parent::_before(); $this->settings = new SettingsController(); - $populator = new Populator($this->settings, WPFunctions::get(), new Captcha); + $referral_detector = new ReferralDetector(WPFunctions::get(), $this->settings); + $populator = new Populator($this->settings, WPFunctions::get(), new Captcha, $referral_detector); $populator->up(); $this->WP_user = $this->_createWPUser(); $this->WP_post = $this->_createWPPost(); diff --git a/tests/integration/Subscription/UrlTest.php b/tests/integration/Subscription/UrlTest.php index 9f8e244a2c..29bac6792a 100644 --- a/tests/integration/Subscription/UrlTest.php +++ b/tests/integration/Subscription/UrlTest.php @@ -1,6 +1,7 @@ settings = new SettingsController; - $populator = new Populator($this->settings, WPFunctions::get(), new Captcha); + $referral_detector = new ReferralDetector(WPFunctions::get(), $this->settings); + $populator = new Populator($this->settings, WPFunctions::get(), new Captcha, $referral_detector); $populator->up(); }