tests/HTMLPurifier/ComplexHarness.php

   1 <?php
   2
   3 /**
   4  * General-purpose test-harness that makes testing functions that require
   5  * configuration and context objects easier when those two parameters are
   6  * meaningless.  See HTMLPurifier_ChildDefTest for a good example of usage.
   7  */
   8 class HTMLPurifier_ComplexHarness extends HTMLPurifier_Harness
   9 {
  10
  11     /**
  12      * Instance of the object that will execute the method
  13      */
  14     protected $obj;
  15
  16     /**
  17      * Name of the function to be executed
  18      */
  19     protected $func;
  20
  21     /**
  22      * Whether or not the method deals in tokens. If set to true, assertResult()
  23      * will transparently convert HTML to and back from tokens.
  24      */
  25     protected $to_tokens = false;
  26
  27     /**
  28      * Whether or not to convert tokens back into HTML before performing
  29      * equality check, has no effect on bools.
  30      */
  31     protected $to_html = false;
  32
  33     /**
  34      * Instance of an HTMLPurifier_Lexer implementation.
  35      */
  36     protected $lexer;
  37
  38     public function __construct() {
  39         $this->lexer     = new HTMLPurifier_Lexer_DirectLex();
  40         parent::__construct();
  41     }
  42
  43     /**
  44      * Asserts a specific result from a one parameter + config/context function
  45      * @param $input Input parameter
  46      * @param $expect Expectation
  47      * @param $config Configuration array in form of Ns.Directive => Value.
  48      *                Has no effect if $this->config is set.
  49      * @param $context_array Context array in form of Key => Value or an actual
  50      *                       context object.
  51      */
  52     protected function assertResult($input, $expect = true) {
  53
  54         if ($this->to_tokens && is_string($input)) {
  55             // $func may cause $input to change, so "clone" another copy
  56             // to sacrifice
  57             $input   = $this->tokenize($temp = $input);
  58             $input_c = $this->tokenize($temp);
  59         } else {
  60             $input_c = $input;
  61         }
  62
  63         // call the function
  64         $func = $this->func;
  65         $result = $this->obj->$func($input_c, $this->config, $this->context);
  66
  67         // test a bool result
  68         if (is_bool($result)) {
  69             $this->assertIdentical($expect, $result);
  70             return;
  71         } elseif (is_bool($expect)) {
  72             $expect = $input;
  73         }
  74
  75         if ($this->to_html) {
  76             $result = $this->generate($result);
  77             if (is_array($expect)) {
  78                 $expect = $this->generate($expect);
  79             }
  80         }
  81         $this->assertIdentical($expect, $result);
  82
  83         if ($expect !== $result) {
  84             echo '<pre>' . var_dump($result) . '</pre>';
  85         }
  86
  87     }
  88
  89     /**
  90      * Tokenize HTML into tokens, uses member variables for common variables
  91      */
  92     protected function tokenize($html) {
  93         return $this->lexer->tokenizeHTML($html, $this->config, $this->context);
  94     }
  95
  96     /**
  97      * Generate textual HTML from tokens
  98      */
  99     protected function generate($tokens) {
 100         $generator = new HTMLPurifier_Generator($this->config, $this->context);
 101         return $generator->generateFromTokens($tokens);
 102     }
 103
 104 }
 105
 106 // vim: et sw=4 sts=4