matthandlersux · September 15, 2012 12:37 · Mar 2, 2012 · Apr 7, 2011
diff --git a/PrivateMethodCaller.scala b/PrivateMethodCaller.scala
@@ -14,7 +14,7 @@ class PrivateMethodCaller(x: AnyRef, methodName: String) {
 }
 
 class PrivateMethodExposer(x: AnyRef) {
-  def apply(method: Symbol): PrivateMethodCaller = new PrivateMethodCaller(x, method.name)
+  def apply(method: scala.Symbol): PrivateMethodCaller = new PrivateMethodCaller(x, method.name)
 }
 
 def p(x: AnyRef): PrivateMethodExposer = new PrivateMethodExposer(x)
diff --git a/PrivateMethodCaller.scala b/PrivateMethodCaller.scala
@@ -0,0 +1,20 @@
+// Usage:
+//   p(instance)('privateMethod)(arg1, arg2, arg3)
+
+class PrivateMethodCaller(x: AnyRef, methodName: String) {
+  def apply(_args: Any*): Any = {
+    val args = _args.map(_.asInstanceOf[AnyRef])
+    def _parents: Stream[Class[_]] = Stream(x.getClass) #::: _parents.map(_.getSuperclass)
+    val parents = _parents.takeWhile(_ != null).toList
+    val methods = parents.flatMap(_.getDeclaredMethods)
+    val method = methods.find(_.getName == methodName).getOrElse(throw new IllegalArgumentException("Method " + methodName + " not found"))
+    method.setAccessible(true)
+    method.invoke(x, args : _*)
+  }
+}
+
+class PrivateMethodExposer(x: AnyRef) {
+  def apply(method: Symbol): PrivateMethodCaller = new PrivateMethodCaller(x, method.name)
+}
+
+def p(x: AnyRef): PrivateMethodExposer = new PrivateMethodExposer(x)