1 files changed, 93 insertions, 0 deletions
diff --git a/src/partest-extras/scala/org/scalacheck/ScalaCheckFramework.scala b/src/partest-extras/scala/org/scalacheck/ScalaCheckFramework.scala
new file mode 100644
index 0000000000..754b67764d
--- /dev/null
+++ b/src/partest-extras/scala/org/scalacheck/ScalaCheckFramework.scala
@@ -0,0 +1,93 @@
+/*-------------------------------------------------------------------------*\
+**  ScalaCheck                                                             **
+**  Copyright (c) 2007-2014 Rickard Nilsson. All rights reserved.          **
+**  http://www.scalacheck.org                                              **
+**                                                                         **
+**  This software is released under the terms of the Revised BSD License.  **
+**  There is NO WARRANTY. See the file LICENSE for the full text.          **
+\*------------------------------------------------------------------------ */
+
+package org.scalacheck
+
+import util.Pretty
+
+import org.scalatools.testing._
+
+class ScalaCheckFramework extends Framework {
+
+  private def mkFP(mod: Boolean, cname: String) =
+    new SubclassFingerprint {
+      val superClassName = cname
+      val isModule = mod
+    }
+
+  val name = "ScalaCheck"
+
+  val tests = Array[Fingerprint](
+    mkFP(true, "org.scalacheck.Properties"),
+    mkFP(false, "org.scalacheck.Prop"),
+    mkFP(false, "org.scalacheck.Properties"),
+    mkFP(true, "org.scalacheck.Prop")
+  )
+
+  def testRunner(loader: ClassLoader,  loggers: Array[Logger]) = new Runner2 {
+
+    private def asEvent(nr: (String, Test.Result)) = nr match {
+      case (n: String, r: Test.Result) => new Event {
+        val testName = n
+        val description = n
+        val result = r.status match {
+          case Test.Passed => Result.Success
+          case _:Test.Proved => Result.Success
+          case _:Test.Failed => Result.Failure
+          case Test.Exhausted => Result.Skipped
+          case _:Test.PropException | _:Test.GenException => Result.Error
+        }
+        val error = r.status match {
+          case Test.PropException(_, e, _) => e
+          case _:Test.Failed => new Exception(Pretty.pretty(r,Pretty.Params(0)))
+          case _ => null
+        }
+      }
+    }
+
+    def run(testClassName: String, fingerprint: Fingerprint, handler: EventHandler, args: Array[String]) {
+
+      val testCallback = new Test.TestCallback {
+        override def onPropEval(n: String, w: Int, s: Int, d: Int) = {}
+
+        override def onTestResult(n: String, r: Test.Result) = {
+          for (l <- loggers) {
+            import Pretty._
+            val verbosityOpts = Set("-verbosity", "-v")
+            val verbosity = args.grouped(2).filter(twos => verbosityOpts(twos.head)).toSeq.headOption.map(_.last).map(_.toInt).getOrElse(0)
+            l.info(
+              (if (r.passed) "+ " else "! ") + n + ": " + pretty(r, Params(verbosity))
+            )
+          }
+          handler.handle(asEvent((n,r)))
+        }
+      }
+
+      val prms = Test.parseParams(args) match {
+        case Some(params) =>
+          params.withTestCallback(testCallback).withCustomClassLoader(Some(loader))
+        // TODO: Maybe handle this a bit better than throwing exception?
+        case None => throw new Exception()
+      }
+
+      fingerprint match {
+        case fp: SubclassFingerprint =>
+          val obj =
+            if(fp.isModule) Class.forName(testClassName + "$", true, loader).getField("MODULE$").get(null)
+            else Class.forName(testClassName, true, loader).newInstance
+          if(obj.isInstanceOf[Properties])
+            Test.checkProperties(prms, obj.asInstanceOf[Properties])
+          else
+            handler.handle(asEvent((testClassName, Test.check(prms, obj.asInstanceOf[Prop]))))
+      }
+    }
+
+  }
+
+}