*
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */
namespace Symfony\Component\HttpKernel\Tests\HttpCache;
use PHPUnit\Framework\TestCase;
use Symfony\Component\HttpFoundation\Request;
use Symfony\Component\HttpFoundation\Response;
use Symfony\Component\HttpKernel\HttpCache\Esi;
class EsiTest extends TestCase
{
    public function testHasSurrogateEsiCapability()
    {
        $esi = new Esi();
        $request = Request::create('/');
        $request->headers->set('Surrogate-Capability', 'abc="ESI/1.0"');
        $this->assertTrue($esi->hasSurrogateCapability($request));
        $request = Request::create('/');
        $request->headers->set('Surrogate-Capability', 'foobar');
        $this->assertFalse($esi->hasSurrogateCapability($request));
        $request = Request::create('/');
        $this->assertFalse($esi->hasSurrogateCapability($request));
    }
    public function testAddSurrogateEsiCapability()
    {
        $esi = new Esi();
        $request = Request::create('/');
        $esi->addSurrogateCapability($request);
        $this->assertEquals('symfony="ESI/1.0"', $request->headers->get('Surrogate-Capability'));
        $esi->addSurrogateCapability($request);
        $this->assertEquals('symfony="ESI/1.0", symfony="ESI/1.0"', $request->headers->get('Surrogate-Capability'));
    }
    public function testAddSurrogateControl()
    {
        $esi = new Esi();
        $response = new Response('foo ');
        $esi->addSurrogateControl($response);
        $this->assertEquals('content="ESI/1.0"', $response->headers->get('Surrogate-Control'));
        $response = new Response('foo');
        $esi->addSurrogateControl($response);
        $this->assertEquals('', $response->headers->get('Surrogate-Control'));
    }
    public function testNeedsEsiParsing()
    {
        $esi = new Esi();
        $response = new Response();
        $response->headers->set('Surrogate-Control', 'content="ESI/1.0"');
        $this->assertTrue($esi->needsParsing($response));
        $response = new Response();
        $this->assertFalse($esi->needsParsing($response));
    }
    public function testRenderIncludeTag()
    {
        $esi = new Esi();
        $this->assertEquals('', $esi->renderIncludeTag('/', '/alt', true));
        $this->assertEquals('', $esi->renderIncludeTag('/', '/alt', false));
        $this->assertEquals('', $esi->renderIncludeTag('/'));
        $this->assertEquals(''."\n".'', $esi->renderIncludeTag('/', '/alt', true, 'some comment'));
    }
    public function testProcessDoesNothingIfContentTypeIsNotHtml()
    {
        $esi = new Esi();
        $request = Request::create('/');
        $response = new Response();
        $response->headers->set('Content-Type', 'text/plain');
        $esi->process($request, $response);
        $this->assertFalse($response->headers->has('x-body-eval'));
    }
    public function testMultilineEsiRemoveTagsAreRemoved()
    {
        $esi = new Esi();
        $request = Request::create('/');
        $response = new Response(' www.example.com  Keep this'."\n www.example.com  And this");
        $esi->process($request, $response);
        $this->assertEquals(' Keep this And this', $response->getContent());
    }
    public function testCommentTagsAreRemoved()
    {
        $esi = new Esi();
        $request = Request::create('/');
        $response = new Response(' Keep this');
        $esi->process($request, $response);
        $this->assertEquals(' Keep this', $response->getContent());
    }
    public function testProcess()
    {
        $esi = new Esi();
        $request = Request::create('/');
        $response = new Response('foo ');
        $esi->process($request, $response);
        $this->assertEquals('foo surrogate->handle($this, \'...\', \'alt\', true) ?>'."\n", $response->getContent());
        $this->assertEquals('ESI', $response->headers->get('x-body-eval'));
        $response = new Response('foo ');
        $esi->process($request, $response);
        $this->assertEquals('foo surrogate->handle($this, \'foo\\\'\', \'bar\\\'\', true) ?>'."\n", $response->getContent());
        $response = new Response('foo ');
        $esi->process($request, $response);
        $this->assertEquals('foo surrogate->handle($this, \'...\', \'\', false) ?>'."\n", $response->getContent());
        $response = new Response('foo ');
        $esi->process($request, $response);
        $this->assertEquals('foo surrogate->handle($this, \'...\', \'\', false) ?>'."\n", $response->getContent());
    }
    public function testProcessEscapesPhpTags()
    {
        $esi = new Esi();
        $request = Request::create('/');
        $response = new Response('');
        $esi->process($request, $response);
        $this->assertEquals('php   cript language=php>', $response->getContent());
    }
    /**
     * @expectedException \RuntimeException
     */
    public function testProcessWhenNoSrcInAnEsi()
    {
        $esi = new Esi();
        $request = Request::create('/');
        $response = new Response('foo ');
        $esi->process($request, $response);
    }
    public function testProcessRemoveSurrogateControlHeader()
    {
        $esi = new Esi();
        $request = Request::create('/');
        $response = new Response('foo ');
        $response->headers->set('Surrogate-Control', 'content="ESI/1.0"');
        $esi->process($request, $response);
        $this->assertEquals('ESI', $response->headers->get('x-body-eval'));
        $response->headers->set('Surrogate-Control', 'no-store, content="ESI/1.0"');
        $esi->process($request, $response);
        $this->assertEquals('ESI', $response->headers->get('x-body-eval'));
        $this->assertEquals('no-store', $response->headers->get('surrogate-control'));
        $response->headers->set('Surrogate-Control', 'content="ESI/1.0", no-store');
        $esi->process($request, $response);
        $this->assertEquals('ESI', $response->headers->get('x-body-eval'));
        $this->assertEquals('no-store', $response->headers->get('surrogate-control'));
    }
    public function testHandle()
    {
        $esi = new Esi();
        $cache = $this->getCache(Request::create('/'), new Response('foo'));
        $this->assertEquals('foo', $esi->handle($cache, '/', '/alt', true));
    }
    /**
     * @expectedException \RuntimeException
     */
    public function testHandleWhenResponseIsNot200()
    {
        $esi = new Esi();
        $response = new Response('foo');
        $response->setStatusCode(404);
        $cache = $this->getCache(Request::create('/'), $response);
        $esi->handle($cache, '/', '/alt', false);
    }
    public function testHandleWhenResponseIsNot200AndErrorsAreIgnored()
    {
        $esi = new Esi();
        $response = new Response('foo');
        $response->setStatusCode(404);
        $cache = $this->getCache(Request::create('/'), $response);
        $this->assertEquals('', $esi->handle($cache, '/', '/alt', true));
    }
    public function testHandleWhenResponseIsNot200AndAltIsPresent()
    {
        $esi = new Esi();
        $response1 = new Response('foo');
        $response1->setStatusCode(404);
        $response2 = new Response('bar');
        $cache = $this->getCache(Request::create('/'), array($response1, $response2));
        $this->assertEquals('bar', $esi->handle($cache, '/', '/alt', false));
    }
    protected function getCache($request, $response)
    {
        $cache = $this->getMockBuilder('Symfony\Component\HttpKernel\HttpCache\HttpCache')->setMethods(array('getRequest', 'handle'))->disableOriginalConstructor()->getMock();
        $cache->expects($this->any())
              ->method('getRequest')
              ->will($this->returnValue($request))
        ;
        if (\is_array($response)) {
            $cache->expects($this->any())
                  ->method('handle')
                  ->will(\call_user_func_array(array($this, 'onConsecutiveCalls'), $response))
            ;
        } else {
            $cache->expects($this->any())
                  ->method('handle')
                  ->will($this->returnValue($response))
            ;
        }
        return $cache;
    }
}