Fix node and schedule unit tests

This commit is contained in:
Dane Everitt 2020-10-05 21:31:39 -07:00
parent 0c2bd416ee
commit b9eb87deaa
No known key found for this signature in database
GPG key ID: EEA66103B3D71F53
2 changed files with 3 additions and 12 deletions

View file

@ -1,11 +1,4 @@
<?php <?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 Tests\Unit\Services\Nodes; namespace Tests\Unit\Services\Nodes;
@ -90,7 +83,7 @@ class NodeDeletionServiceTest extends TestCase
*/ */
public function testModelCanBePassedToFunctionInPlaceOfNodeId() public function testModelCanBePassedToFunctionInPlaceOfNodeId()
{ {
$node = factory(Node::class)->make(); $node = factory(Node::class)->make(['id' => 123]);
$this->serverRepository->shouldReceive('setColumns')->with('id')->once()->andReturnSelf() $this->serverRepository->shouldReceive('setColumns')->with('id')->once()->andReturnSelf()
->shouldReceive('findCountWhere')->with([['node_id', '=', $node->id]])->once()->andReturn(0); ->shouldReceive('findCountWhere')->with([['node_id', '=', $node->id]])->once()->andReturn(0);

View file

@ -47,7 +47,7 @@ class ProcessScheduleServiceTest extends TestCase
*/ */
public function testScheduleIsUpdatedAndRun() public function testScheduleIsUpdatedAndRun()
{ {
$model = factory(Schedule::class)->make(); $model = factory(Schedule::class)->make(['id' => 123]);
$model->setRelation('tasks', collect([$task = factory(Task::class)->make([ $model->setRelation('tasks', collect([$task = factory(Task::class)->make([
'sequence_id' => 1, 'sequence_id' => 1,
])])); ])]));
@ -65,14 +65,12 @@ class ProcessScheduleServiceTest extends TestCase
$this->dispatcher->shouldReceive('dispatch')->with(m::on(function ($class) use ($model, $task) { $this->dispatcher->shouldReceive('dispatch')->with(m::on(function ($class) use ($model, $task) {
$this->assertInstanceOf(RunTaskJob::class, $class); $this->assertInstanceOf(RunTaskJob::class, $class);
$this->assertSame($task->time_offset, $class->delay); $this->assertSame($task->time_offset, $class->delay);
$this->assertSame($task->id, $class->task); $this->assertSame($task->id, $class->task->id);
$this->assertSame($model->id, $class->schedule);
return true; return true;
}))->once(); }))->once();
$this->getService()->handle($model); $this->getService()->handle($model);
$this->assertTrue(true);
} }
/** /**