com.github.loyada.dollarx.custommatchers.hamcrest.HasElementNTimesForDocument.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dollarx-scala Show documentation

There is a newer version: 0.35

package com.github.loyada.dollarx.custommatchers.hamcrest

import com.github.loyada.dollarx.RelationOperator._
import com.github.loyada.dollarx.util.StringUtil
import com.github.loyada.dollarx.{PathParsers, RelationOperator, Path}
import org.hamcrest.{Matcher, Description, TypeSafeMatcher}
import org.w3c.dom.Document


class HasElementNTimesForDocument(path: Path, nTimes: Int) {

    case class NTimesMatcherForDocument(relationOperator: RelationOperator = exactly) extends TypeSafeMatcher[Document] {
      private var foundNTimes: Int = 0

      def describeTo(description: Description) {
        description.appendText(s"document page contains ${StringUtil.wrap(path)}${opAsEnglish(relationOperator)}${nTimes} time${if (nTimes != 1) "s" else ""}")
      }

      protected override def describeMismatchSafely(document: Document, mismatchDescription: Description) {
        mismatchDescription.appendText(StringUtil.wrap(path) + " appears " + foundNTimes + " time" + (if (foundNTimes != 1) "s" else ""))
      }

      protected def matchesSafely(document: Document): Boolean = {
        foundNTimes = PathParsers.findAllByPath(document, path).getLength
        relationOperator match {
          case RelationOperator.orLess => foundNTimes <= nTimes
          case RelationOperator.orMore => foundNTimes >= nTimes
          case RelationOperator.exactly => foundNTimes == nTimes
        }
      }
    }

    def timesOrMore: Matcher[Document] = NTimesMatcherForDocument(orMore)
    def timesOrLess: Matcher[Document] = NTimesMatcherForDocument(orLess)
    def times: Matcher[Document] = NTimesMatcherForDocument()

}