<?php

namespace ZendDiagnosticsTest;

use Guzzle\Http\Client as Guzzle3Client;
use Guzzle\Http\Message\Response as Guzzle3Response;
use GuzzleHttp\Client as Guzzle4And5Client;
use GuzzleHttp\Message\Response as Guzzle4And5Response;
use Guzzle\Plugin\Mock\MockPlugin;
use GuzzleHttp\Stream\Stream;
use GuzzleHttp\Subscriber\Mock;
use ZendDiagnostics\Check\GuzzleHttpService;

class GuzzleHttpServiceTest extends \PHPUnit_Framework_TestCase
{
    /**
     * @dataProvider checkProvider
     */
    public function testGuzzle3Check($content, $actualContent, $actualStatusCode, $resultClass, $method = 'GET', $body = null)
    {
        $check = new GuzzleHttpService(
            'http://www.example.com/foobar',
            array(),
            array(),
            '200',
            $content,
            $this->getMockGuzzle3Client($actualStatusCode, $actualContent),
            $method,
            $body
        );
        $result = $check->check();

        $this->assertInstanceOf($resultClass, $result);
    }

    /**
     * @dataProvider checkProvider
     */
    public function testGuzzle4And5Check($content, $actualContent, $actualStatusCode, $resultClass, $method = 'GET', $body = null)
    {
        if (!class_exists('GuzzleHttp\Client')) {
            $this->markTestSkipped('guzzlehttp/guzzle not installed.');
        }

        $check = new GuzzleHttpService(
            'http://www.example.com/foobar',
            array(),
            array(),
            '200',
            $content,
            $this->getMockGuzzle4And5Client($actualStatusCode, $actualContent),
            $method,
            $body
        );
        $result = $check->check();

        $this->assertInstanceOf($resultClass, $result);
    }

    /**
     * @expectedException \InvalidArgumentException
     */
    public function testInvalidClient()
    {
        $check = new GuzzleHttpService('http://example.com', array(), array(), 200, null, 'not guzzle');
    }

    public function checkProvider()
    {
        return array(
            array(null, null, 200, 'ZendDiagnostics\Result\SuccessInterface'),
            array(null, null, 200, 'ZendDiagnostics\Result\SuccessInterface', 'POST', array('key' => 'value')),
            array(null, null, 200, 'ZendDiagnostics\Result\SuccessInterface', 'PUT'),
            array(null, null, 404, 'ZendDiagnostics\Result\FailureInterface'),
            array(null, null, 404, 'ZendDiagnostics\Result\FailureInterface', 'POST', array('key' => 'value')),
            array(null, null, 404, 'ZendDiagnostics\Result\FailureInterface', 'PUT'),
            array('foo', 'foobar', 200, 'ZendDiagnostics\Result\SuccessInterface'),
            array('foo', 'foobar', 200, 'ZendDiagnostics\Result\SuccessInterface', 'POST', array('key' => 'value')),
            array('foo', 'foobar', 200, 'ZendDiagnostics\Result\SuccessInterface', 'PUT'),
            array('baz', 'foobar', 200, 'ZendDiagnostics\Result\FailureInterface'),
            array('baz', 'foobar', 200, 'ZendDiagnostics\Result\FailureInterface', 'POST', array('key' => 'value')),
            array('baz', 'foobar', 200, 'ZendDiagnostics\Result\FailureInterface', 'PUT'),
            array('baz', 'foobar', 500, 'ZendDiagnostics\Result\FailureInterface'),
        );
    }

    private function getMockGuzzle3Client($statusCode = 200, $content = null)
    {
        $plugin = new MockPlugin();
        $plugin->addResponse(new Guzzle3Response($statusCode, null, $content));

        $client = new Guzzle3Client();
        $client->addSubscriber($plugin);

        return $client;
    }

    private function getMockGuzzle4And5Client($statusCode = 200, $content = null)
    {
        $client = new Guzzle4And5Client();
        $client->getEmitter()->attach(new Mock(array(new Guzzle4And5Response($statusCode, array(), Stream::factory((string) $content)))));

        return $client;
    }
}