tests/HTMLPurifier/ErrorCollectorEMock.php

   1 <?php
   2
   3 generate_mock_once('HTMLPurifier_ErrorCollector');
   4
   5 /**
   6  * Extended error collector mock that has the ability to expect context
   7  */
   8 class HTMLPurifier_ErrorCollectorEMock extends HTMLPurifier_ErrorCollectorMock
   9 {
  10
  11     private $_context;
  12     private $_expected_context = array();
  13     private $_expected_context_at = array();
  14
  15     public function prepare($context)
  16     {
  17         $this->_context = $context;
  18     }
  19
  20     public function expectContext($key, $value)
  21     {
  22         $this->_expected_context[$key] = $value;
  23     }
  24     public function expectContextAt($step, $key, $value)
  25     {
  26         $this->_expected_context_at[$step][$key] = $value;
  27     }
  28
  29     public function send($v1, $v2)
  30     {
  31         // test for context
  32         $context = SimpleTest::getContext();
  33         $test = $context->getTest();
  34         $mock = $this->mock;
  35
  36         foreach ($this->_expected_context as $key => $value) {
  37             $test->assertEqual($value, $this->_context->get($key));
  38         }
  39         $step = $mock->getCallCount('send');
  40         if (isset($this->_expected_context_at[$step])) {
  41             foreach ($this->_expected_context_at[$step] as $key => $value) {
  42                 $test->assertEqual($value, $this->_context->get($key));
  43             }
  44         }
  45         // boilerplate mock code, does not have return value or references
  46         $args = func_get_args();
  47         $mock->invoke('send', $args);
  48     }
  49
  50 }
  51
  52 // vim: et sw=4 sts=4