class PHPUnit_Framework_Constraint_Count

class PHPUnit_Framework_Constraint_Count extends PHPUnit_Framework_Constraint {

  /**
   * @var integer
   */
  protected $expectedCount = 0;

  /**
   * @param integer $expected
   */
  public function __construct($expected) {
    $this->expectedCount = $expected;
  }

  /**
   * Evaluates the constraint for parameter $other. Returns TRUE if the
   * constraint is met, FALSE otherwise.
   *
   * @param mixed $other
   * @return boolean
   */
  protected function matches($other) {
    return $this->expectedCount === $this
      ->getCountOf($other);
  }

  /**
   * @param mixed $other
   * @return boolean
   */
  protected function getCountOf($other) {
    if ($other instanceof Countable || is_array($other)) {
      return count($other);
    }
    else {
      if ($other instanceof Iterator) {
        return iterator_count($other);
      }
    }
  }

  /**
   * Returns the description of the failure
   *
   * The beginning of failure messages is "Failed asserting that" in most
   * cases. This method should return the second part of that sentence.
   *
   * @param  mixed $other Evaluated value or object.
   * @return string
   */
  protected function failureDescription($other) {
    return sprintf('actual size %d matches expected size %d', $this
      ->getCountOf($other), $this->expectedCount);
  }

  /**
   * @return string
   */
  public function toString() {
    return 'count matches ';
  }

}