test_chooser.py

import os, subprocess
import sublime, sublime_plugin, io

class TestChooserCommand(sublime_plugin.WindowCommand):

  def run(self, paths = [], name = ""):
    self.settings         = sublime.load_settings('TestChooser.sublime-settings')
    search_terms          = self.settings.get('search_terms')
    self.terminal         = self.settings.get('terminal')
    self.filepath         = self.window.active_view().file_name()
    self.filetype         = self.set_file_type()
    self.testable_lines   = ["Run'em All"]

    if self.filetype == 'unknown':
      self.exit_with_alert()

    else:
      line_number = 0

      for line in io.open(self.filepath, 'r'):
        line_number += 1
        for word in search_terms:
          if line.strip().startswith(word):
            self.testable_lines.append("%03d:%s" % (line_number, line))
            break

      chosen_test = self.choose_test()

  def choose_test(self):
    self.window.show_quick_panel(self.testable_lines, self.chose_item)

  def chose_item(self, chosen_test):
    if chosen_test != -1:
      formatted_line = self.format_line(self.testable_lines[chosen_test])

      if self.terminal == "iTerm":
        applescript = self.iterm_script()
        self.execute_cmd(applescript, formatted_line)
      else:
        applescript = self.terminal_script()
        self.execute_cmd(applescript, formatted_line)

    self.activate_sublime()

  def format_line(self, line):
    if line == "Run'em All":
      if 'rake' in self.filetype:
        return "%s SPEC='%s'" % (self.filetype, self.filepath)
      else:
        return "%s '%s'" % (self.filetype, self.filepath)
    else:
      items = line.split(":")
      if 'rake' in self.filetype:
        return "%s SPEC='%s:%s'" % (self.filetype, self.filepath, items[0])
      else:
        return "%s '%s:%s'" % (self.filetype, self.filepath, items[0])

  def execute_cmd(self, applescript, formatted_line):
    cmd = applescript.replace("$cmd", formatted_line)
    cmd = "osascript -e '%s'" % cmd
    os.system(cmd)

  def set_file_type(self):
    if 'feature' in self.filepath:
      return self.settings.get('cucumber_command')
    elif 'spec' in self.filepath:
      return self.settings.get('rspec_command')
    else:
      return 'unknown'

  def exit_with_alert(self):
    sublime.error_message('This is not a Cucumber or RSpec file.')

  def iterm_script(self):
    return """
      tell application "iTerm"
        tell current session of terminal 1
          write text "$cmd"
        end tell
      end tell
    """

  def terminal_script(self):
    return """
      tell application "Terminal"
        tell window 1
          do script "$cmd" in selected tab
        end tell
      end tell
    """

  def activate_sublime(self):
    subprocess.Popen("""osascript -e 'tell app "Sublime Text 2" to activate'""", shell=True)