StarMC/Astral-nook
StarMC/Astral-nook
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add unit tests for RunTaskJob.

Browse source
This commit is contained in:
Dane Everitt 2017-09-30 12:40:07 -05:00
parent 238ce435d6
commit d5bf8734ef
No known key found for this signature in database
GPG key ID: EEA66103B3D71F53
4 changed files with 219 additions and 99 deletions
Download patch file Download diff file Expand all files Collapse all files

3
app/Jobs/Schedule/RunTaskJob.php
View file

@ -61,7 +61,8 @@ class RunTaskJob extends Job implements ShouldQueue
*/
public function __construct($task, $schedule)
{
Assert::integerish($task, 'First argument passed to constructor must be numeric, received %s.');
Assert::integerish($task, 'First argument passed to constructor must be integer, received %s.');
Assert::integerish($schedule, 'Second argument passed to constructor must be integer, received %s.');
$this->queue = app()->make('config')->get('pterodactyl.queues.standard');
$this->task = $task;

98
app/Jobs/SendScheduledTask.php
View file

@ -1,98 +0,0 @@
<?php
/**
* Pterodactyl - Panel
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* This software is licensed under the terms of the MIT license.
* https://opensource.org/licenses/MIT
*/
namespace Pterodactyl\Jobs;
use Cron;
use Carbon;
use Pterodactyl\Models\Task;
use Pterodactyl\Models\TaskLog;
use Illuminate\Queue\SerializesModels;
use Illuminate\Queue\InteractsWithQueue;
use Illuminate\Contracts\Queue\ShouldQueue;
use Pterodactyl\Repositories\old_Daemon\PowerRepository;
use Pterodactyl\Repositories\old_Daemon\CommandRepository;
class SendScheduledTask extends Job implements ShouldQueue
{
use InteractsWithQueue, SerializesModels;
/**
* @var \Pterodactyl\Models\Task
*/
protected $task;
/**
* Create a new job instance.
*/
public function __construct(Task $task)
{
$this->task = $task;
$this->task->queued = true;
$this->task->save();
}
/**
* Execute the job.
*/
public function handle()
{
$time = Carbon::now();
$log = new TaskLog;
if ($this->attempts() >= 1) {
// Just delete the job, we will attempt it again later anyways.
$this->delete();
}
try {
if ($this->task->action === 'command') {
$repo = new CommandRepository($this->task->server, $this->task->user);
$response = $repo->send($this->task->data);
} elseif ($this->task->action === 'power') {
$repo = new PowerRepository($this->task->server, $this->task->user);
$response = $repo->do($this->task->data);
} else {
throw new \Exception('Task type provided was not valid.');
}
$log->fill([
'task_id' => $this->task->id,
'run_time' => $time,
'run_status' => 0,
'response' => $response,
]);
} catch (\Exception $ex) {
$log->fill([
'task_id' => $this->task->id,
'run_time' => $time,
'run_status' => 1,
'response' => $ex->getMessage(),
]);
} finally {
$cron = Cron::factory(sprintf(
'%s %s %s %s %s %s',
$this->task->minute,
$this->task->hour,
$this->task->day_of_month,
$this->task->month,
$this->task->day_of_week,
$this->task->year
));
$this->task->fill([
'last_run' => $time,
'next_run' => $cron->getNextRunDate(),
'queued' => false,
]);
$this->task->save();
$log->save();
}
}
}