From 83d238fed3e656cb915fb5f8ff4ef427eec48a6f Mon Sep 17 00:00:00 2001
From: Joshua Davey <josh@joshuadavey.com>
Date: Sun, 23 Oct 2011 15:13:13 -0500
Subject: [PATCH] Add basic test/unit support

---
 plugin/tslime.vim | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

diff --git a/plugin/tslime.vim b/plugin/tslime.vim
index 6746fba..ffb448d 100644
--- a/plugin/tslime.vim
+++ b/plugin/tslime.vim
@@ -73,17 +73,26 @@ function! s:first_readable_file(files) abort
   return files[0]
 endfunction
 
+function! s:prefix_for_test(file)
+  if a:file =~# '_spec.rb$'
+    return "rspec "
+  elseif a:file =~# '_test.rb$'
+    return "ruby -Itest "
+  endif
+  return ''
+endfunction
+
 function! s:SendAlternateToTmux() abort
   let current_file = expand("%")
-  if current_file =~# '_spec.rb$'
-    let command = "rspec ".current_file."\n"
+  if s:prefix_for_test(current_file) != ''
+    let command = s:prefix_for_test(current_file).current_file
   elseif exists('g:autoloaded_rails')
     let related_file = s:first_readable_file(rails#buffer().related())
-    let command = "rspec ".related_file."\n"
+    let command = s:prefix_for_test(related_file).related_file
   else
-    let command = "!!\n"
+    let command = "!!"
   endif
-  return Send_to_Tmux(command)
+  return Send_to_Tmux(command."\n")
 endfunction
 
 augroup tmux