api/tests/Jobs/ProcessMediaWikiJobsJobTest.php at 488d04184da878384619a4e95f62e8c8098e96a4 · wbstack/api · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
<?php

namespace Tests\Jobs;

use App\Jobs\ProcessMediaWikiJobsJob;
use App\Services\MediaWikiHostResolver;
use GuzzleHttp\Handler\MockHandler;
use GuzzleHttp\HandlerStack;
use GuzzleHttp\Psr7\Response;
use Http\Adapter\Guzzle7\Client as GuzzleClient;
use Illuminate\Contracts\Queue\Job;
use Illuminate\Foundation\Testing\RefreshDatabase;
use Maclof\Kubernetes\Client;
use Tests\TestCase;

class ProcessMediaWikiJobsJobTest extends TestCase {
    use RefreshDatabase;

    public function testJobFailOnNoMediaWikiPod() {

        $mockResolver = $this->createMock(MediaWikiHostResolver::class);

        $mockJob = $this->createMock(Job::class);
        $mockJob->expects($this->once())->method('fail');

        $job = new ProcessMediaWikiJobsJob('test.wikibase.cloud');
        $job->setJob($mockJob);

        $mock = new MockHandler([
            new Response(200, [], json_encode(['items' => []])),
            new Response(200, [], json_encode(['items' => []])),
        ]);

        $handlerStack = HandlerStack::create($mock);
        $mockGuzzle = GuzzleClient::createWithConfig([
            'handler' => $handlerStack,
            'verify' => '/var/run/secrets/kubernetes.io/serviceaccount/ca.crt',
        ]);

        $job->handle(new Client([
            'master' => 'https://kubernetes.default.svc',
            'token' => '/var/run/secrets/kubernetes.io/serviceaccount/token',
        ], null, $mockGuzzle), $mockResolver);
    }

    public function testJobDoesNotFail() {
        $mockResolver = $this->createMock(MediaWikiHostResolver::class);
        $mockJob = $this->createMock(Job::class);
        $mockJob->expects($this->never())->method('fail');

        $job = new ProcessMediaWikiJobsJob('test.wikibase.cloud');
        $job->setJob($mockJob);

        $mock = new MockHandler([
            new Response(200, [], json_encode(['items' => [
                [
                    'kind' => 'Service',
                    'spec' => [
                        'selector' => [
                            'app.kubernetes.io/component' => 'app-backend',
                            'app.kubernetes.io/instance' => 'mediawiki-143',
                            'app.kubernetes.io/name' => 'mediawiki',
                        ],
                    ],
                ],
            ]])),
            new Response(200, [], json_encode(['items' => [
                [
                    'kind' => 'Pod',
                    'spec' => [
                        'containers' => [
                            [
                                'image' => 'helloworld',
                                'env' => [
                                    'SOMETHING' => 'something',
                                ],
                            ],
                        ],
                    ],
                ],
            ]])),
            new Response(200, [], json_encode(['items' => []])),
            new Response(201, [], json_encode([
                'metadata' => [
                    'name' => 'some-job-name',
                ],
            ])),
        ]);

        $handlerStack = HandlerStack::create($mock);
        $mockGuzzle = GuzzleClient::createWithConfig([
            'handler' => $handlerStack,
            'verify' => '/var/run/secrets/kubernetes.io/serviceaccount/ca.crt',
        ]);

        $job->handle(
            new Client([
                'master' => 'https://kubernetes.default.svc',
                'token' => '/var/run/secrets/kubernetes.io/serviceaccount/token',
            ], null, $mockGuzzle),
            $mockResolver
        );
    }
}