<?php

/**

 * Zend Framework

 *

 * LICENSE

 *

 * This source file is subject to the new BSD license that is bundled

 * with this package in the file LICENSE.txt.

 * It is also available through the world-wide-web at this URL:

 * http://framework.zend.com/license/new-bsd

 * If you did not receive a copy of the license and are unable to

 * obtain it through the world-wide-web, please send an email

 * to license@zend.com so we can send you a copy immediately.

 *

 * @category   Zend

 * @package    Zend_Controller

 * @copyright  Copyright (c) 2005-2010 Zend Technologies USA Inc. (http://www.zend.com)

 * @license    http://framework.zend.com/license/new-bsd     New BSD License

 * @version    $Id: HttpTestCase.php 20096 2010-01-06 02:05:09Z bkarwin $

 */

/**

 * @see Zend_Controller_Response_Http

 */

require_once 'Zend/Controller/Response/Http.php';

/**

 * Zend_Controller_Response_HttpTestCase

 *

 * @uses Zend_Controller_Response_Http

 * @package Zend_Controller

 * @subpackage Response

 */

class Zend_Controller_Response_HttpTestCase extends Zend_Controller_Response_Http

{

    /**

     * "send" headers by returning array of all headers that would be sent

     *

     * @return array

     */

    public function sendHeaders()

    {

        $headers = array();

        foreach ($this->_headersRaw as $header) {

            $headers[] = $header;

        }

        foreach ($this->_headers as $header) {

            $name = $header['name'];

            $key  = strtolower($name);

            if (array_key_exists($name, $headers)) {

                if ($header['replace']) {

                    $headers[$key] = $header['name'] . ': ' . $header['value'];

                }

            } else {

                $headers[$key] = $header['name'] . ': ' . $header['value'];

            }

        }

        return $headers;

    }

    /**

     * Can we send headers?

     *

     * @param  bool $throw

     * @return void

     */

    public function canSendHeaders($throw = false)

    {

        return true;

    }

    /**

     * Return the concatenated body segments

     *

     * @return string

     */

    public function outputBody()

    {

        $fullContent = '';

        foreach ($this->_body as $content) {

            $fullContent .= $content;

        }

        return $fullContent;

    }

    /**

     * Get body and/or body segments

     *

     * @param  bool|string $spec

     * @return string|array|null

     */

    public function getBody($spec = false)

    {

        if (false === $spec) {

            return $this->outputBody();

        } elseif (true === $spec) {

            return $this->_body;

        } elseif (is_string($spec) && isset($this->_body[$spec])) {

            return $this->_body[$spec];

        }

        return null;

    }

    /**

     * "send" Response

     *

     * Concats all response headers, and then final body (separated by two

     * newlines)

     *

     * @return string

     */

    public function sendResponse()

    {

        $headers = $this->sendHeaders();

        $content = implode("\n", $headers) . "\n\n";

        if ($this->isException() && $this->renderExceptions()) {

            $exceptions = '';

            foreach ($this->getException() as $e) {

                $exceptions .= $e->__toString() . "\n";

            }

            $content .= $exceptions;

        } else {

            $content .= $this->outputBody();

        }

        return $content;

    }

}