UltiSnips/test.py

280 lines
7.1 KiB
Python
Raw Normal View History

2009-06-23 08:45:04 -04:00
#!/usr/bin/env python
# encoding: utf-8
#
import os
import tempfile
2009-06-23 08:45:04 -04:00
import unittest
import time
2009-06-23 08:45:04 -04:00
class _VimTest(unittest.TestCase):
def type(self, str):
"""
Send the keystrokes to vim via screen. Pause after each tab, so
expansion can take place
"""
def _send(s):
2009-07-01 09:20:40 -04:00
os.system("screen -x %s -X stuff '%s'" % (self.session,s))
2009-06-23 08:45:04 -04:00
splits = str.split('\t')
for w in splits[:-1]:
_send(w + '\t')
_send(splits[-1])
def escape(self):
self.type("\x1b")
2009-06-23 08:45:04 -04:00
def setUp(self):
self.escape()
self.type(":py PySnipSnippets.reset()\n")
for sv,content in self.snippets:
self.type(''':py << EOF
PySnipSnippets.add_snippet("%s","""%s""")
EOF
''' % (sv,content))
2009-07-01 09:20:40 -04:00
# Clear the buffer
self.type("bggVGd")
2009-07-01 09:20:40 -04:00
if not self.interrupt:
# Enter insert mode
self.type("i")
2009-07-01 09:20:40 -04:00
# Execute the command
self.cmd()
2009-07-01 09:20:40 -04:00
handle, fn = tempfile.mkstemp(prefix="PySnipEmuTest",suffix=".txt")
os.close(handle)
2009-06-23 08:45:04 -04:00
2009-07-01 09:20:40 -04:00
self.escape()
self.type(":w! %s\n" % fn)
2009-06-23 08:45:04 -04:00
2009-07-01 09:20:40 -04:00
# Give screen a chance to send the cmd and vim to write the file
time.sleep(.01)
2009-07-01 09:20:40 -04:00
# Read the output, chop the trailing newline
self.output = open(fn,"r").read()[:-1]
2009-06-23 08:45:04 -04:00
def cmd(self):
"""Overwrite these in the children"""
pass
##################
# Simple Expands #
##################
class _SimpleExpands(_VimTest):
snippets = (
("hallo", "Hallo Welt!"),
)
2009-06-23 08:45:04 -04:00
class SimpleExpand_ExceptCorrectResult(_SimpleExpands):
def cmd(self):
self.type("hallo\t")
2009-06-23 08:45:04 -04:00
def runTest(self):
self.assertEqual(self.output,"Hallo Welt!")
2009-06-23 08:45:04 -04:00
class SimpleExpandTypeAfterExpand_ExceptCorrectResult(_SimpleExpands):
def cmd(self):
self.type("hallo\tand again")
2009-06-23 08:45:04 -04:00
def runTest(self):
self.assertEqual(self.output,"Hallo Welt!and again")
2009-06-23 08:45:04 -04:00
class SimpleExpandTypeAfterExpand1_ExceptCorrectResult(_SimpleExpands):
def cmd(self):
self.type("na du hallo\tand again")
2009-06-23 08:45:04 -04:00
def runTest(self):
self.assertEqual(self.output,"na du Hallo Welt!and again")
2009-06-23 08:45:04 -04:00
class DoNotExpandAfterSpace_ExceptCorrectResult(_SimpleExpands):
def cmd(self):
self.type("hallo \t")
2009-06-23 08:45:04 -04:00
def runTest(self):
self.assertEqual(self.output,"hallo ")
2009-06-23 08:45:04 -04:00
class ExpandInTheMiddleOfLine_ExceptCorrectResult(_SimpleExpands):
def cmd(self):
self.type("Wie hallo gehts?")
self.escape()
self.type("bhi\t")
2009-06-23 08:45:04 -04:00
def runTest(self):
self.assertEqual(self.output,"Wie Hallo Welt! gehts?")
class MultilineExpand_ExceptCorrectResult(_VimTest):
snippets = (
("hallo", "Hallo Welt!\nUnd Wie gehts?"),
)
2009-06-23 08:45:04 -04:00
def cmd(self):
self.type("Wie hallo gehts?")
self.escape()
self.type("bhi\t")
2009-06-23 08:45:04 -04:00
def runTest(self):
self.assertEqual(self.output, "Wie Hallo Welt!\nUnd Wie gehts? gehts?")
class MultilineExpandTestTyping_ExceptCorrectResult(_VimTest):
snippets = (
("hallo", "Hallo Welt!\nUnd Wie gehts?"),
)
def cmd(self):
self.type("Wie hallo gehts?")
self.escape()
self.type("bhi\tHuiui!")
def runTest(self):
self.assertEqual(self.output,
"Wie Hallo Welt!\nUnd Wie gehts?Huiui! gehts?")
2009-06-23 08:45:04 -04:00
############
# TabStops #
############
class ExitTabStop_ExceptCorrectResult(_VimTest):
snippets = (
("echo", "$0 run"),
)
2009-06-23 08:45:04 -04:00
def cmd(self):
self.type("echo\ttest")
2009-06-23 08:45:04 -04:00
def runTest(self):
self.assertEqual(self.output,"test run")
class TextTabStopNoReplace_ExceptCorrectResult(_VimTest):
snippets = (
("echo", "echo ${1:Hallo}"),
)
def cmd(self):
self.type("echo\t")
def runTest(self):
self.assertEqual(self.output,"echo Hallo")
class TextTabStopSimpleReplace_ExceptCorrectResult(_VimTest):
snippets = (
("hallo", "hallo ${0:End} ${1:Beginning}"),
)
def cmd(self):
self.type("hallo\tna\tDu Nase")
def runTest(self):
self.assertEqual(self.output,"hallo Du Nase na")
2009-07-01 09:20:40 -04:00
class TextTabStopSimpleReplace_ExceptCorrectResult(_VimTest):
snippets = (
("hallo", "hallo ${0:End} ${1:Beginning}"),
)
def cmd(self):
self.type("hallo\tna\tDu Nase")
def runTest(self):
self.assertEqual(self.output,"hallo Du Nase na")
# TODO: multiline mirrors
class TextTabStopSimpleMirror_ExceptCorrectResult(_VimTest):
snippets = (
("test", "$1\n$1"),
)
def cmd(self):
self.type("test\thallo")
def runTest(self):
self.assertEqual(self.output,"hallo\nhallo")
class TextTabStopSimpleMirrorDelete_ExceptCorrectResult(_VimTest):
snippets = (
("test", "$1\n$1"),
)
def cmd(self):
self.type("test\thallo")
self.type("\b\b")
def runTest(self):
self.assertEqual(self.output,"hal\nhal")
class TextTabStopSimpleMirrorSameLine_ExceptCorrectResult(_VimTest):
snippets = (
("test", "$1 $1"),
)
def cmd(self):
self.type("test\thallo")
def runTest(self):
self.assertEqual(self.output,"hal\nhal")
class TextTabStopSimpleMirrorDeleteSomeEnterSome_ExceptCorrectResult(_VimTest):
snippets = (
("test", "$1\n$1"),
)
def cmd(self):
self.type("test\thallo\b\bhups")
def runTest(self):
self.assertEqual(self.output,"halhups\nhalhups")
class TextTabStopMirrorMoreInvolved_ExceptCorrectResult(_VimTest):
snippets = (
("for", "for(size_t ${2:i} = 0; $2 < ${1:count}; ${3:++$2})\n{\n\t${0:/* code */}\n}"),
)
def cmd(self):
self.type("for\t")
def runTest(self):
self.assertEqual(self.output,"hallo Du Nase na")
2009-06-23 08:45:04 -04:00
if __name__ == '__main__':
import sys
2009-07-01 09:20:40 -04:00
import optparse
def parse_args():
p = optparse.OptionParser("%prog [OPTIONS] <test case names to run>")
p.set_defaults(session="vim", interrupt=False)
p.add_option("-s", "--session", dest="session", metavar="SESSION",
help="send commands to screen session SESSION [%default]")
p.add_option("-i", "--interrupt", dest="interrupt",
action="store_true",
help="Stop after defining the snippet. This allows the user" \
"to interactively test the snippet in vim. You must give exactly" \
"one test case on the cmdline. The test will always fail."
)
2009-07-01 09:20:40 -04:00
o, args = p.parse_args()
return o, args
options,selected_tests = parse_args()
# The next line doesn't work in python 2.3
all_test_suites = unittest.TestLoader().loadTestsFromModule(__import__("test"))
# Inform all test case which screen session to use
suite = unittest.TestSuite()
2009-07-01 09:20:40 -04:00
for s in all_test_suites:
for test in s:
test.session = options.session
test.interrupt = options.interrupt
if len(selected_tests):
id = test.id().split('.')[1]
if id not in selected_tests:
continue
suite.addTest(test)
res = unittest.TextTestRunner().run(suite)
2009-07-01 09:20:40 -04:00