<?php

require_once 'HTMLPurifier/ErrorCollector.php';

generate_mock_once('HTMLPurifier_ErrorCollector');

/**
 * Extended error collector mock that has the ability to expect context
 */
class HTMLPurifier_ErrorCollectorEMock extends HTMLPurifier_ErrorCollectorMock
{
    
    private $_context;
    private $_expected_context = array();
    private $_expected_context_at = array();
    
    public function prepare(&$context) {
        $this->_context =& $context;
    }
    
    public function expectContext($key, $value) {
        $this->_expected_context[$key] = $value;
    }
    public function expectContextAt($step, $key, $value) {
        $this->_expected_context_at[$step][$key] = $value;
    }
    
    public function send($v1, $v2) {
        // test for context
        $context =& SimpleTest::getContext();
        $test =& $context->getTest();
        
        // compat
        if (empty($this->_mock)) {
            $mock =& $this;
        } else {
            $mock =& $this->_mock;
        }
        
        foreach ($this->_expected_context as $key => $value) {
            $test->assertEqual($value, $this->_context->get($key));
        }
        $step = $mock->getCallCount('send');
        if (isset($this->_expected_context_at[$step])) {
            foreach ($this->_expected_context_at[$step] as $key => $value) {
                $test->assertEqual($value, $this->_context->get($key));
            }
        }
        // boilerplate mock code, does not have return value or references
        $args = func_get_args();
        $mock->_invoke('send', $args);
    }
    
}