svn commit: r1131934 - /incubator/mesos/trunk/src/swig/java/TestFramework.java

[be...@apache.org]

Author: benh
Date: Sun Jun  5 06:38:18 2011
New Revision: 1131934

URL: http://svn.apache.org/viewvc?rev=1131934&view=rev
Log:
Making TestFramework optionally take num_tasks as command line argument.

Modified:
    incubator/mesos/trunk/src/swig/java/TestFramework.java

Modified: incubator/mesos/trunk/src/swig/java/TestFramework.java
URL: http://svn.apache.org/viewvc/incubator/mesos/trunk/src/swig/java/TestFramework.java?rev=1131934&r1=1131933&r2=1131934&view=diff
==============================================================================
--- incubator/mesos/trunk/src/swig/java/TestFramework.java (original)
+++ incubator/mesos/trunk/src/swig/java/TestFramework.java Sun Jun  5 06:38:18 2011
@@ -9,7 +9,13 @@ public class TestFramework {
   static class MyScheduler extends Scheduler {
     int launchedTasks = 0;
     int finishedTasks = 0;
-    final int totalTasks = 5;
+    int totalTasks = 5;
+
+    public MyScheduler() {}
+
+    public MyScheduler(int numTasks) {
+      totalTasks = numTasks;
+    }
 
     @Override
     public String getFrameworkName(SchedulerDriver d) {
@@ -79,6 +85,12 @@ public class TestFramework {
   }
 
   public static void main(String[] args) throws Exception {
-    new MesosSchedulerDriver(new MyScheduler(), args[0]).run();
+    if (args.length < 1 || args.length > 2) {
+      System.out.println("Invalid use: please specify a master");
+    } else if (args.length == 1) {
+      new MesosSchedulerDriver(new MyScheduler(),args[0]).run();
+    } else {
+      new MesosSchedulerDriver(new MyScheduler(Integer.parseInt(args[1])), args[0]).run();
+    }
   }
 }