sublime-phpunit.py

import os
import sys
import shlex
import ntpath
import subprocess
import sublime
import sublime_plugin

class PhpunitTestCommand(sublime_plugin.WindowCommand):
    def get_paths(self):
        file_name = self.window.active_view().file_name()
        phpunit_config_path = self.find_phpunit_config(file_name)

        directory = os.path.dirname(os.path.realpath(file_name))

        file_name = file_name.replace(' ', '\ ')
        phpunit_config_path = phpunit_config_path.replace(' ', '\ ')

        active_view = self.window.active_view()

        return file_name, phpunit_config_path, active_view, directory

    def get_current_function(self, view):
        sel = view.sel()[0]
        function_regions = view.find_by_selector('entity.name.function')
        cf = None
        for r in reversed(function_regions):
            if r.a < sel.a:
                cf = view.substr(r)
                break
        return cf

    def find_phpunit_config(self, file_name):
        phpunit_config_path = file_name
        found = False
        while found == False:
            phpunit_config_path = os.path.abspath(os.path.join(phpunit_config_path, os.pardir))
            found = os.path.isfile(phpunit_config_path + '/phpunit.xml') or phpunit_config_path == '/'
        return phpunit_config_path

    def run_in_terminal(self, command):
        settings = sublime.load_settings("Preferences.sublime-settings")
        terminal_setting = settings.get('phpunit-sublime-terminal', 'Terminal')

        osascript_command = 'osascript '

        if terminal_setting == 'iTerm':
            osascript_command += '"' + os.path.dirname(os.path.realpath(__file__)) + '/open_iterm.applescript"'
            osascript_command += ' "' + command + '"'
        else:
            osascript_command += '"' + os.path.dirname(os.path.realpath(__file__)) + '/run_command.applescript"'
            osascript_command += ' "' + command + '"'
            osascript_command += ' "PHPUnit Tests"'

        os.system(osascript_command)

class RunPhpunitTestCommand(PhpunitTestCommand):

    def run(self, *args, **kwargs):
        file_name, phpunit_config_path, active_view, directory = self.get_paths()

        self.run_in_terminal('cd ' + phpunit_config_path + ' && phpunit ' + file_name)

class RunAllPhpunitTestsCommand(PhpunitTestCommand):

    def run(self, *args, **kwargs):
        file_name, phpunit_config_path, active_view, directory = self.get_paths()

        self.run_in_terminal('cd ' + phpunit_config_path + ' && phpunit')


class RunSinglePhpunitTestCommand(PhpunitTestCommand):

    def run(self, *args, **kwargs):
        file_name, phpunit_config_path, active_view, directory = self.get_paths()

        current_function = self.get_current_function(active_view)

        self.run_in_terminal('cd ' + phpunit_config_path + ' && phpunit ' + file_name + " --filter '/::" + current_function + "$/'")

class RunPhpunitTestsInDirCommand(PhpunitTestCommand):

    def run(self, *args, **kwargs):
        file_name, phpunit_config_path, active_view, directory = self.get_paths()

        self.run_in_terminal('cd ' + phpunit_config_path + ' && phpunit ' + directory)

class FindMatchingTestCommand(sublime_plugin.WindowCommand):

    def path_leaf(self, path):
        head, tail = ntpath.split(path)
        return tail or ntpath.basename(head)

    def run(self, *args, **kwargs):
        file_name = self.window.active_view().file_name()
        file_name = self.path_leaf(file_name)
        file_name = file_name[0:file_name.find('.')]
        tab_target = 0

        if 'Test' not in file_name:
            file_name = file_name + 'Test'
        else:
            # Strip 'Test' and add '.' to force matching the non-test file
            file_name = file_name[0:file_name.find('Test')] + '.'
            tab_target = 1

        # Big dirty macro-ish hack. Eventually I should just open the file in some sort of
        # logical way.
        self.window.run_command("set_layout", {"cells": [[0, 0, 1, 1], [1, 0, 2, 1]], "cols": [0.0, 0.5, 1.0], "rows": [0.0, 1.0]})
        self.window.run_command("focus_group", {"group": tab_target})
        self.window.run_command("show_overlay", {"overlay": "goto", "text": file_name, "show_files": "true"})
        self.window.run_command("move", {"by": "lines", "forward": False})

        # This is a dirty hack to get it to switch files... Can't simulate 'Enter'
        # but triggering the overlay again to close it seems to have the same effect.
        self.window.run_command("show_overlay", {"overlay": "goto", "show_files": "true"})
        self.window.run_command("focus_group", {"group": 0})
        self.window.run_command("focus_group", {"group": tab_target})