Remove AspectMock from CronHelperTest [MAILPOET-2464]

This commit is contained in:
wxa
2019-10-29 13:02:52 +03:00
committed by Jack Kitterhing
parent 2ed473f3a0
commit 8f4321487b
2 changed files with 19 additions and 13 deletions

View File

@@ -94,13 +94,17 @@ class CronHelper {
return Security::generateRandomString(); return Security::generateRandomString();
} }
static function pingDaemon() { static function pingDaemon($wp = null) {
if (is_null($wp)) {
$wp = new WPFunctions();
}
$url = self::getCronUrl( $url = self::getCronUrl(
CronDaemonEndpoint::ACTION_PING_RESPONSE CronDaemonEndpoint::ACTION_PING_RESPONSE,
false,
$wp
); );
$result = self::queryCronUrl($url); $result = self::queryCronUrl($url);
if (is_wp_error($result)) return $result->get_error_message(); if (is_wp_error($result)) return $result->get_error_message();
$wp = new WPFunctions();
$response = $wp->wpRemoteRetrieveBody($result); $response = $wp->wpRemoteRetrieveBody($result);
$response = substr(trim($response), -strlen(DaemonHttpRunner::PING_SUCCESS_RESPONSE)) === DaemonHttpRunner::PING_SUCCESS_RESPONSE ? $response = substr(trim($response), -strlen(DaemonHttpRunner::PING_SUCCESS_RESPONSE)) === DaemonHttpRunner::PING_SUCCESS_RESPONSE ?
DaemonHttpRunner::PING_SUCCESS_RESPONSE : DaemonHttpRunner::PING_SUCCESS_RESPONSE :
@@ -112,7 +116,10 @@ class CronHelper {
return $response === DaemonHttpRunner::PING_SUCCESS_RESPONSE; return $response === DaemonHttpRunner::PING_SUCCESS_RESPONSE;
} }
static function accessDaemon($token) { static function accessDaemon($token, $wp = null) {
if (is_null($wp)) {
$wp = new WPFunctions();
}
$data = ['token' => $token]; $data = ['token' => $token];
$url = self::getCronUrl( $url = self::getCronUrl(
CronDaemonEndpoint::ACTION_RUN, CronDaemonEndpoint::ACTION_RUN,
@@ -124,8 +131,7 @@ class CronHelper {
} }
$daemon['run_accessed_at'] = time(); $daemon['run_accessed_at'] = time();
self::saveDaemon($daemon); self::saveDaemon($daemon);
$result = self::queryCronUrl($url); $result = self::queryCronUrl($url, $wp);
$wp = new WPFunctions();
return $wp->wpRemoteRetrieveBody($result); return $wp->wpRemoteRetrieveBody($result);
} }

View File

@@ -2,7 +2,6 @@
namespace MailPoet\Test\Cron; namespace MailPoet\Test\Cron;
use AspectMock\Test as Mock;
use Codeception\Stub; use Codeception\Stub;
use MailPoet\Cron\CronHelper; use MailPoet\Cron\CronHelper;
use MailPoet\Cron\DaemonHttpRunner; use MailPoet\Cron\DaemonHttpRunner;
@@ -102,8 +101,10 @@ class CronHelperTest extends \MailPoetTest {
$daemon $daemon
); );
$time = time(); $time = time();
Mock::double('MailPoet\Cron\CronHelper', ['queryCronUrl' => []]); $wp = Stub::make(new WPFunctions, [
CronHelper::accessDaemon('some_token'); 'wpRemotePost' => [],
]);
CronHelper::accessDaemon('some_token', $wp);
$updated_daemon = CronHelper::getDaemon(); $updated_daemon = CronHelper::getDaemon();
expect($updated_daemon['run_accessed_at'])->greaterOrEquals($time); expect($updated_daemon['run_accessed_at'])->greaterOrEquals($time);
expect($updated_daemon['run_accessed_at'])->lessThan($time + 2); expect($updated_daemon['run_accessed_at'])->lessThan($time + 2);
@@ -300,10 +301,10 @@ class CronHelperTest extends \MailPoetTest {
} }
function testItReturnsErrorMessageAsPingResponseWhenCronUrlCannotBeAccessed() { function testItReturnsErrorMessageAsPingResponseWhenCronUrlCannotBeAccessed() {
Mock::double('MailPoet\Cron\CronHelper', [ $wp = Stub::make(new WPFunctions, [
'getSiteUrl' => false, 'applyFilters' => false,
]); ]);
expect(CronHelper::pingDaemon())->equals('A valid URL was not provided.'); expect(CronHelper::pingDaemon($wp))->equals('A valid URL was not provided.');
} }
function testItPingsDaemon() { function testItPingsDaemon() {
@@ -318,7 +319,6 @@ class CronHelperTest extends \MailPoetTest {
} }
function _after() { function _after() {
Mock::clean();
\ORM::raw_execute('TRUNCATE ' . Setting::$_table); \ORM::raw_execute('TRUNCATE ' . Setting::$_table);
} }