typesense-php/tests/TestCase.php at 9ebff7d84357a0ada27a7dc6dffd941db27c1546 · tharropoulos/typesense-php · 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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
<?php

namespace Tests;

use PHPUnit\Framework\TestCase as BaseTestCase;
use Typesense\Client;
use Mockery;
use Typesense\ApiCall;
use Exception;

abstract class TestCase extends BaseTestCase
{
    private ?Client $typesenseClient = null;
    private $mockApiCall;

    protected function setUp(): void
    {
        $this->setUpTypesenseClient();
        $this->mockApiCall = Mockery::mock(ApiCall::class);
    }

    protected function tearDown(): void
    {
        $this->tearDownTypesense();
    }

    protected function client(): Client
    {
        return $this->typesenseClient;
    }

    protected function mockApiCall()
    {
        return $this->mockApiCall;
    }

    protected function getSchema(string $name): array
    {
        $path = __DIR__ . "/data/{$name}.schema.json";

        return $this->loadFromDataDir($path);
    }

    protected function getData(string $name): array
    {
        $path = __DIR__ . "/data/{$name}.data.json";

        return $this->loadFromDataDir($path);
    }

    private function loadFromDataDir(string $path): array
    {
        if (!file_exists($path)) {
            return [];
        }

        return json_decode(
            file_get_contents($path),
            true,
            512,
            JSON_THROW_ON_ERROR
        );
    }

    private function setUpTypesenseClient(): void
    {
        $this->typesenseClient = new Client([
            'api_key' => $_ENV['TYPESENSE_API_KEY'],
            'nodes' => [
                [
                    'host' => $_ENV['TYPESENSE_NODE_HOST'],
                    'port' => $_ENV['TYPESENSE_NODE_PORT'],
                    'protocol' => $_ENV['TYPESENSE_NODE_PROTOCOL']
                ],
            ]
        ]);
    }

    protected function setUpCollection(string $schema): void
    {
        $schema = $this->getSchema($schema);
        $this->typesenseClient->collections->create($schema);
    }

    protected function setUpDocuments(string $schema): void
    {
        $documents = $this->getData($schema);
        $this->typesenseClient->collections[$schema]->documents->import($documents);
    }

    protected function tearDownTypesense(): void
    {
        if ($this->typesenseClient === null) {
            return;
        }

        $collections = $this->typesenseClient->collections->retrieve();
        foreach ($collections as $collection) {
            $this->typesenseClient->collections[$collection['name']]->delete();
        }
    }

    protected function isV30OrAbove(): bool
    {
        try {
            $debug = $this->typesenseClient->debug->retrieve();
            $version = $debug['version'];

            if ($version === 'nightly') {
                return true;
            }

            if (preg_match('/^v(\d+)/', $version, $matches)) {
                $majorVersion = (int) $matches[1];
                return $majorVersion >= 30;
            }

            return false;
        } catch (Exception $e) {
            return false;
        }
    }
}