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: 
<?php
namespace Guzzle\Tests\Plugin\Md5;
use Guzzle\Http\EntityBody;
use Guzzle\Http\Message\RequestFactory;
use Guzzle\Http\Message\Response;
use Guzzle\Plugin\Md5\Md5ValidatorPlugin;
class Md5ValidatorPluginTest extends \Guzzle\Tests\GuzzleTestCase
{
    public function testValidatesMd5()
    {
        $plugin = new Md5ValidatorPlugin();
        $request = RequestFactory::getInstance()->create('GET', 'http://www.test.com/');
        $request->getEventDispatcher()->addSubscriber($plugin);
        $body = 'abc';
        $hash = md5($body);
        $response = new Response(200, array(
            'Content-MD5' => $hash,
            'Content-Length' => 3
        ), 'abc');
        $request->dispatch('request.complete', array(
            'response' => $response
        ));
        
        $response->removeHeader('Content-MD5');
        $request->dispatch('request.complete', array(
            'response' => $response
        ));
    }
    
    public function testThrowsExceptionOnInvalidMd5()
    {
        $plugin = new Md5ValidatorPlugin();
        $request = RequestFactory::getInstance()->create('GET', 'http://www.test.com/');
        $request->getEventDispatcher()->addSubscriber($plugin);
        $request->dispatch('request.complete', array(
            'response' => new Response(200, array(
                'Content-MD5' => 'foobar',
                'Content-Length' => 3
            ), 'abc')
        ));
    }
    public function testSkipsWhenContentLengthIsTooLarge()
    {
        $plugin = new Md5ValidatorPlugin(false, 1);
        $request = RequestFactory::getInstance()->create('GET', 'http://www.test.com/');
        $request->getEventDispatcher()->addSubscriber($plugin);
        $request->dispatch('request.complete', array(
            'response' => new Response(200, array(
                'Content-MD5' => 'foobar',
                'Content-Length' => 3
            ), 'abc')
        ));
    }
    public function testProperlyValidatesWhenUsingContentEncoding()
    {
        $plugin = new Md5ValidatorPlugin(true);
        $request = RequestFactory::getInstance()->create('GET', 'http://www.test.com/');
        $request->getEventDispatcher()->addSubscriber($plugin);
        
        
        
        $body = EntityBody::factory('abc');
        $body->compress();
        $hash = $body->getContentMd5();
        $body->uncompress();
        $response = new Response(200, array(
            'Content-MD5' => $hash,
            'Content-Encoding' => 'gzip'
        ), 'abc');
        $request->dispatch('request.complete', array(
            'response' => $response
        ));
        $this->assertEquals('abc', $response->getBody(true));
        
        $response = new Response(200, array(
            'Content-MD5' => $hash,
            'Content-Encoding' => 'foobar'
        ), 'abc');
        $request->dispatch('request.complete', array(
            'response' => $response
        ));
        
        $body->compress('bzip2.compress');
        $response = new Response(200, array(
            'Content-MD5' => $body->getContentMd5(),
            'Content-Encoding' => 'compress'
        ), 'abc');
        $request->dispatch('request.complete', array(
            'response' => $response
        ));
        
        $request->getEventDispatcher()->removeSubscriber($plugin);
        $plugin = new Md5ValidatorPlugin(false);
        $request->getEventDispatcher()->addSubscriber($plugin);
        $request->dispatch('request.complete', array(
            'response' => $response
        ));
    }
}