lib/java_autotest/test_runner.rb in java-autotest-0.0.1 vs lib/java_autotest/test_runner.rb in java-autotest-0.0.2

@@ -1,9 +1,40 @@
 class TestRunner
-  def self.run_test(test_class)
-		system("mvn -Dtest=#{test_class} test")
-	end
-	
-	def self.run_all_tests
-		system("mvn test")
-	end
+  attr_accessor :build_tool
+  ICON = File.expand_path(File.join(File.dirname(__FILE__), '..', '..', 'img', 'java_icon.png'))
+  Title = 'Java AutoTest'  
+
+  def self.valid_tools
+    %w(mvn gradle)
+  end
+
+  def initialize
+    @build_tool = discover_build_tool
+  end
+
+  def discover_build_tool
+    File.exists?("build.gradle") ? "gradle" : "mvn" 
+  end
+
+  def run_test(test_class)
+    command = ".single" if @build_tool == "gradle"
+    green = system("#{@build_tool} -Dtest#{command}=#{test_class} test")
+    notify "Test Failure: #{test_class}" unless green
+    green
+  end
+
+  def run_all_tests
+    green = system("#{@build_tool} test")
+    notify "Build Success" if green
+    notify "Build Broken" unless green 
+  end
+
+  def notify(message)
+    case RUBY_PLATFORM
+    when /darwin/
+      system "growlnotify -t '#{Title}' -m '#{message}' --image #{ICON}"
+    when /linux/
+      system "notify-send '#{Title}' '#{message}' --icon #{ICON}"
+    end
+  end
+
 end