From 447ab8d3342a8f2213f1cbe4650d6d6bf0fc4d7f Mon Sep 17 00:00:00 2001
From: Matej Urbas <matej.urbas@gmail.com>
Date: Sat, 12 Apr 2014 20:16:58 +0100
Subject: [PATCH] Created unit tests for ParserWithExamples and
 FixedSetExampleSource.

---
 .../src/main/scala/sbt/complete/Parser.scala  |  2 +-
 .../sbt/complete/FixedSetExamplesTest.scala   | 26 ++++++
 .../sbt/complete/ParserWithExamplesTest.scala | 93 +++++++++++++++++++
 3 files changed, 120 insertions(+), 1 deletion(-)
 create mode 100644 util/complete/src/test/scala/sbt/complete/FixedSetExamplesTest.scala
 create mode 100644 util/complete/src/test/scala/sbt/complete/ParserWithExamplesTest.scala

diff --git a/util/complete/src/main/scala/sbt/complete/Parser.scala b/util/complete/src/main/scala/sbt/complete/Parser.scala
index 8a54b09fd..faea14d9a 100644
--- a/util/complete/src/main/scala/sbt/complete/Parser.scala
+++ b/util/complete/src/main/scala/sbt/complete/Parser.scala
@@ -752,7 +752,7 @@ private final class ParserWithExamples[T](delegate: Parser[T], exampleSource: Ex
 	}
 
 	private def isExampleValid(example: String): Boolean = {
-		apply(delegate)(example).resultEmpty.isFailure
+		apply(delegate)(example).resultEmpty.isValid
 	}
 }
 private final class StringLiteral(str: String, start: Int) extends ValidParser[String]
diff --git a/util/complete/src/test/scala/sbt/complete/FixedSetExamplesTest.scala b/util/complete/src/test/scala/sbt/complete/FixedSetExamplesTest.scala
new file mode 100644
index 000000000..b9a5b2de2
--- /dev/null
+++ b/util/complete/src/test/scala/sbt/complete/FixedSetExamplesTest.scala
@@ -0,0 +1,26 @@
+package sbt.complete
+
+import org.specs2.mutable.Specification
+import org.specs2.specification.Scope
+
+class FixedSetExamplesTest extends Specification {
+
+	"adding a prefix" should {
+		"produce a smaller set of examples with the prefix removed" in new examples {
+			fixedSetExamples.withAddedPrefix("f")() must containTheSameElementsAs(List("oo", "ool", "u"))
+			fixedSetExamples.withAddedPrefix("fo")() must containTheSameElementsAs(List("o", "ol"))
+			fixedSetExamples.withAddedPrefix("b")() must containTheSameElementsAs(List("ar"))
+		}
+	}
+
+	"without a prefix" should {
+		"produce the original set" in new examples {
+			fixedSetExamples() mustEqual exampleSet
+		}
+	}
+
+	trait examples extends Scope {
+		val exampleSet = List("foo", "bar", "fool", "fu")
+		val fixedSetExamples = FixedSetExamples(exampleSet)
+	}
+}
diff --git a/util/complete/src/test/scala/sbt/complete/ParserWithExamplesTest.scala b/util/complete/src/test/scala/sbt/complete/ParserWithExamplesTest.scala
new file mode 100644
index 000000000..664018f8f
--- /dev/null
+++ b/util/complete/src/test/scala/sbt/complete/ParserWithExamplesTest.scala
@@ -0,0 +1,93 @@
+package sbt.complete
+
+import org.specs2.mutable.Specification
+import org.specs2.specification.Scope
+import Completion._
+
+class ParserWithExamplesTest extends Specification {
+
+	"listing a limited number of completions" should {
+		"grab only the needed number of elements the iterable source of examples" in new parserWithLazyExamples {
+			parserWithExamples.completions(0)
+			examples.size shouldEqual maxNumberOfExamples
+		}
+	}
+
+	"listing only valid completions" should {
+		"remove invalid examples" in new parserWithValidExamples {
+			val validCompletions = Completions(Set(
+				suggestion("blue"),
+				suggestion("red")
+			))
+			parserWithExamples.completions(0) shouldEqual validCompletions
+		}
+	}
+
+	"listing completions in a derived parser" should {
+		"produce only examples that match the derivation" in new parserWithValidExamples {
+			val derivedCompletions = Completions(Set(
+				suggestion("lue")
+			))
+			parserWithExamples.derive('b').completions(0) shouldEqual derivedCompletions
+		}
+	}
+
+	"listing unfiltered completions" should {
+		"produce all examples" in new parserWithAllExamples {
+			val completions = Completions(examples.map(suggestion(_)).toSet)
+			parserWithExamples.completions(0) shouldEqual completions
+		}
+	}
+
+	"listing completions in a derived parser" should {
+		"produce only examples that match the derivation" in new parserWithAllExamples {
+			val derivedCompletions = Completions(Set(
+				suggestion("lue"),
+				suggestion("lock")
+			))
+			parserWithExamples.derive('b').completions(0) shouldEqual derivedCompletions
+		}
+	}
+
+	class parserWithLazyExamples extends parser(GrowableSourceOfExamples(), maxNumberOfExamples = 5, removeInvalidExamples = false)
+
+	class parserWithValidExamples extends parser(removeInvalidExamples = true)
+
+	class parserWithAllExamples extends parser(removeInvalidExamples = false)
+
+	case class parser(examples: Iterable[String] = Set("blue", "yellow", "greeen", "block", "red"),
+										maxNumberOfExamples: Int = 25,
+										removeInvalidExamples: Boolean) extends Scope {
+
+		import DefaultParsers._
+
+		val colorParser = "blue" | "green" | "black" | "red"
+		val parserWithExamples = new ParserWithExamples[String](
+			colorParser,
+			FixedSetExamples(examples),
+			maxNumberOfExamples,
+			removeInvalidExamples
+		)
+	}
+
+	case class GrowableSourceOfExamples() extends Iterable[String] {
+		var numberOfIteratedElements: Int = 0
+
+		override def iterator: Iterator[String] = {
+			new Iterator[String] {
+				var currentElement = 0
+
+				override def next(): String = {
+					currentElement += 1
+					numberOfIteratedElements = Math.max(currentElement, numberOfIteratedElements)
+					numberOfIteratedElements.toString
+				}
+
+				override def hasNext: Boolean = true
+			}
+		}
+
+		override def size: Int = numberOfIteratedElements
+	}
+
+}