123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120 |
- <?php
- namespace Test\Traits;
- use OCP\Http\Client\IClient;
- use OCP\Http\Client\IClientService;
- use OCP\Http\Client\IResponse;
- trait ClientServiceTrait {
-
- private $clientService;
-
- private $client;
- private $expectedGetRequests = [];
- private $expectedPostRequests = [];
-
- abstract protected function createMock(string $originalClassName);
-
- abstract public static function any();
- protected function setUpClientServiceTrait() {
- $this->clientService = $this->createMock(IClientService::class);
- $this->client = $this->createMock(IClient::class);
- $this->clientService->expects($this->any())
- ->method('newClient')
- ->willReturn($this->client);
- $this->client->expects($this->any())
- ->method('get')
- ->willReturnCallback(function ($url) {
- if (!isset($this->expectedGetRequests[$url])) {
- throw new \Exception('unexpected request: ' . $url);
- }
- $result = $this->expectedGetRequests[$url];
- if ($result instanceof \Exception) {
- throw $result;
- } else {
- $response = $this->createMock(IResponse::class);
- $response->expects($this->any())
- ->method('getBody')
- ->willReturn($result);
- return $response;
- }
- });
- $this->client->expects($this->any())
- ->method('post')
- ->willReturnCallback(function ($url) {
- if (!isset($this->expectedPostRequests[$url])) {
- throw new \Exception('unexpected request: ' . $url);
- }
- $result = $this->expectedPostRequests[$url];
- if ($result instanceof \Exception) {
- throw $result;
- } else {
- $response = $this->createMock(IResponse::class);
- $response->expects($this->any())
- ->method('getBody')
- ->willReturn($result);
- return $response;
- }
- });
- }
-
- protected function expectGetRequest($url, $result) {
- $this->expectedGetRequests[$url] = $result;
- }
-
- protected function expectPostRequest($url, $result) {
- $this->expectedPostRequests[$url] = $result;
- }
-
- protected function getClientService() {
- return $this->clientService;
- }
- }
|