UltiSnips/test.py

523 lines
18 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):
text_before = " --- some text before --- "
text_after = " --- some text after --- "
2009-07-01 11:11:33 -04:00
def send(self, s):
os.system("screen -x %s -X stuff '%s'" % (self.session,s))
def type(self, str):
"""
2009-07-01 11:11:33 -04:00
Send the keystrokes to vim via screen. Pause after each char, so
vim can handle this
"""
2009-07-01 11:11:33 -04:00
for c in str:
self.send(c)
2009-06-23 08:45:04 -04:00
def check_output(self):
wanted = self.text_before + '\n\n' + self.wanted + \
'\n\n' + self.text_after
self.assertEqual(self.output, wanted)
def escape(self):
self.type("\x1b")
2009-06-23 08:45:04 -04:00
def setUp(self):
self.escape()
2009-07-01 11:11:33 -04:00
self.send(":py PySnipSnippets.reset()\n")
if not isinstance(self.snippets[0],tuple):
self.snippets = ( self.snippets, )
for sv,content in self.snippets:
2009-07-01 11:11:33 -04:00
self.send(''':py << EOF
PySnipSnippets.add_snippet("%s","""%s""")
EOF
''' % (sv,content))
2009-07-01 11:11:33 -04:00
# Clear the buffer
2009-07-01 11:11:33 -04:00
self.send("bggVGd")
2009-07-01 09:20:40 -04:00
if not self.interrupt:
# Enter insert mode
2009-07-01 11:11:33 -04:00
self.send("i")
self.send(self.text_before + '\n\n')
self.send('\n\n' + self.text_after)
# Go to the middle of the buffer
self.escape()
self.send("ggjji")
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()
2009-07-01 11:11:33 -04:00
self.send(":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(.10)
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
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):
wanted = "Hallo Welt!"
def cmd(self): self.type("hallo\t")
def runTest(self): self.check_output()
2009-06-23 08:45:04 -04:00
class SimpleExpandTypeAfterExpand_ExceptCorrectResult(_SimpleExpands):
wanted = "Hallo Welt!and again"
def cmd(self): self.type("hallo\tand again")
def runTest(self): self.check_output()
2009-06-23 08:45:04 -04:00
2009-07-03 04:59:55 -04:00
class SimpleExpandTypeAndDelete_ExceptCorrectResult(_SimpleExpands):
wanted = "na du Hallo Welt!and blub"
def cmd(self): self.type("na du hallo\tand again\b\b\b\b\bblub")
def runTest(self): self.check_output()
2009-06-23 08:45:04 -04:00
class DoNotExpandAfterSpace_ExceptCorrectResult(_SimpleExpands):
wanted = "hallo "
def cmd(self): self.type("hallo \t")
def runTest(self): self.check_output()
2009-06-23 08:45:04 -04:00
class ExpandInTheMiddleOfLine_ExceptCorrectResult(_SimpleExpands):
wanted = "Wie Hallo Welt! gehts?"
2009-06-23 08:45:04 -04:00
def cmd(self):
self.type("Wie hallo gehts?")
self.escape()
self.type("bhi\t")
def runTest(self): self.check_output()
2009-06-23 08:45:04 -04:00
class MultilineExpand_ExceptCorrectResult(_VimTest):
wanted = "Wie Hallo Welt!\nUnd Wie gehts? gehts?"
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")
def runTest(self): self.check_output()
class MultilineExpandTestTyping_ExceptCorrectResult(_VimTest):
snippets = ("hallo", "Hallo Welt!\nUnd Wie gehts?")
wanted = "Wie Hallo Welt!\nUnd Wie gehts?Huiui! gehts?"
def cmd(self):
self.type("Wie hallo gehts?")
self.escape()
self.type("bhi\tHuiui!")
def runTest(self): self.check_output()
2009-06-23 08:45:04 -04:00
############
# TabStops #
############
class TabStopSimpleReplace_ExceptCorrectResult(_VimTest):
snippets = ("hallo", "hallo ${0:End} ${1:Beginning}")
wanted = "hallo Du Nase na"
def cmd(self):
self.type("hallo\tna\tDu Nase")
def runTest(self): self.check_output()
class TabStopSimpleReplaceSurrounded_ExceptCorrectResult(_VimTest):
snippets = ("hallo", "hallo ${0:End} a small feed")
wanted = "hallo Nase a small feed"
2009-06-23 08:45:04 -04:00
def cmd(self):
self.type("hallo\tNase")
def runTest(self): self.check_output()
class TabStopSimpleReplaceSurrounded1_ExceptCorrectResult(_VimTest):
snippets = ("hallo", "hallo $0 a small feed")
wanted = "hallo Nase a small feed"
def cmd(self):
self.type("hallo\tNase")
def runTest(self): self.check_output()
2009-06-23 08:45:04 -04:00
class ExitTabStop_ExceptCorrectResult(_VimTest):
snippets = ("echo", "$0 run")
wanted = "test run"
def cmd(self):
self.type("echo\ttest")
def runTest(self): self.check_output()
class TabStopNoReplace_ExceptCorrectResult(_VimTest):
snippets = ("echo", "echo ${1:Hallo}")
wanted = "echo Hallo"
def cmd(self):
self.type("echo\t")
def runTest(self): self.check_output()
# TODO: multiline tabstops, maybe?
class TabStopWithOneChar_ExceptCorrectResult(_VimTest):
snippets = ("hallo", "nothing ${1:i} hups")
wanted = "nothing ship hups"
def cmd(self):
self.type("hallo\tship")
def runTest(self): self.check_output()
class TabStopTestJumping_ExceptCorrectResult(_VimTest):
snippets = ("hallo", "hallo ${2:End} mitte ${1:Beginning}")
wanted = "hallo TestHi mitte Beginning"
def cmd(self):
self.type("hallo\t\tTest\tHi")
def runTest(self): self.check_output()
class TabStopTestJumping2_ExceptCorrectResult(_VimTest):
snippets = ("hallo", "hallo $0 $1")
wanted = "hallo TestHi "
def cmd(self):
self.type("hallo\t\tTest\tHi")
def runTest(self): self.check_output()
2009-07-01 09:20:40 -04:00
class TabStopTestBackwardJumping_ExceptCorrectResult(_VimTest):
snippets = ("hallo", "hallo ${0:End} mitte${1:Beginning}")
wanted = "hallo Blah mitteLets replace it again"
2009-07-01 09:20:40 -04:00
def cmd(self):
self.type(
"hallo\tSomelengthy Text\tHi+Lets replace it again\tBlah\t++\t")
def runTest(self): self.check_output()
class TabStopTestBackwardJumping2_ExceptCorrectResult(_VimTest):
snippets = ("hallo", "hallo $0 $1")
wanted = "hallo Blah Lets replace it again"
def cmd(self):
self.type(
"hallo\tSomelengthy Text\tHi+Lets replace it again\tBlah\t++\t")
def runTest(self): self.check_output()
class TabStopTestMultilineExpand_ExceptCorrectResult(_VimTest):
snippets = ("hallo", "hallo $0\nnice $1 work\n$3 $2\nSeem to work")
wanted = "test hallo one more\nnice world work\n" \
"test try\nSeem to work World"
def cmd(self):
self.type("test hallo World")
self.escape()
2009-07-02 09:41:58 -04:00
self.type("02f i\t")
self.type("world\ttry\ttest\tone more\t\t")
def runTest(self): self.check_output()
2009-07-03 04:59:55 -04:00
# # TODO: pasting with <C-R> while mirroring
# ###########
# # MIRRORS #
# ###########
class TextTabStopTextAfterTab_ExceptCorrectResult(_VimTest):
snippets = ("test", "$1 Hinten\n$1")
wanted = "hallo Hinten\nhallo"
def cmd(self):
self.type("test\thallo")
def runTest(self): self.check_output()
class TextTabStopTextBeforeTab_ExceptCorrectResult(_VimTest):
snippets = ("test", "Vorne $1\n$1")
wanted = "Vorne hallo\nhallo"
def cmd(self):
self.type("test\thallo")
def runTest(self): self.check_output()
class TextTabStopTextSurroundedTab_ExceptCorrectResult(_VimTest):
snippets = ("test", "Vorne $1 Hinten\n$1")
wanted = "Vorne hallo test Hinten\nhallo test"
def cmd(self):
self.type("test\thallo test")
def runTest(self): self.check_output()
class TextTabStopTextBeforeMirror_ExceptCorrectResult(_VimTest):
snippets = ("test", "$1\nVorne $1")
wanted = "hallo\nVorne hallo"
def cmd(self):
self.type("test\thallo")
def runTest(self): self.check_output()
class TextTabStopAfterMirror_ExceptCorrectResult(_VimTest):
snippets = ("test", "$1\n$1 Hinten")
wanted = "hallo\nhallo Hinten"
def cmd(self):
self.type("test\thallo")
def runTest(self): self.check_output()
class TextTabStopSurroundMirror_ExceptCorrectResult(_VimTest):
snippets = ("test", "$1\nVorne $1 Hinten")
wanted = "hallo welt\nVorne hallo welt Hinten"
def cmd(self):
self.type("test\thallo welt")
def runTest(self): self.check_output()
class TextTabStopAllSurrounded_ExceptCorrectResult(_VimTest):
snippets = ("test", "ObenVorne $1 ObenHinten\nVorne $1 Hinten")
wanted = "ObenVorne hallo welt ObenHinten\nVorne hallo welt Hinten"
def cmd(self):
self.type("test\thallo welt")
def runTest(self): self.check_output()
class MirrorBeforeTabstopLeave_ExceptCorrectResult(_VimTest):
snippets = ("test", "$1 ${1:this is it} $1")
wanted = "this is it this is it this is it"
def cmd(self):
self.type("test\t")
def runTest(self): self.check_output()
class MirrorBeforeTabstopOverwrite_ExceptCorrectResult(_VimTest):
snippets = ("test", "$1 ${1:this is it} $1")
wanted = "a a a"
def cmd(self):
self.type("test\ta")
def runTest(self): self.check_output()
class TextTabStopSimpleMirrorMultiline_ExceptCorrectResult(_VimTest):
snippets = ("test", "$1\n$1")
wanted = "hallo\nhallo"
def cmd(self):
self.type("test\thallo")
def runTest(self): self.check_output()
class SimpleMirrorMultilineMany_ExceptCorrectResult(_VimTest):
snippets = ("test", " $1\n$1\na$1b\n$1\ntest $1 mich")
wanted = " hallo\nhallo\nahallob\nhallo\ntest hallo mich"
def cmd(self):
self.type("test\thallo")
def runTest(self): self.check_output()
class MultilineTabStopSimpleMirrorMultiline_ExceptCorrectResult(_VimTest):
snippets = ("test", "$1\n\n$1\n\n$1")
wanted = "hallo Du\nHi\n\nhallo Du\nHi\n\nhallo Du\nHi"
def cmd(self):
self.type("test\thallo Du\nHi")
def runTest(self): self.check_output()
class MultilineTabStopSimpleMirrorMultiline1_ExceptCorrectResult(_VimTest):
snippets = ("test", "$1\n$1\n$1")
wanted = "hallo Du\nHi\nhallo Du\nHi\nhallo Du\nHi"
def cmd(self):
self.type("test\thallo Du\nHi")
def runTest(self): self.check_output()
# TODO: Multiline delete over line endings
class MultilineTabStopSimpleMirrorDeleteInLine_ExceptCorrectResult(_VimTest):
snippets = ("test", "$1\n$1\n$1")
wanted = "hallo Du\nAch Blah\nhallo Du\nAch Blah\nhallo Du\nAch Blah"
def cmd(self):
self.type("test\thallo Du\nHi\b\bAch Blah")
def runTest(self): self.check_output()
class SimpleMirrorDelete_ExceptCorrectResult(_VimTest):
snippets = ("test", "$1\n$1")
wanted = "hal\nhal"
def cmd(self):
self.type("test\thallo")
self.type("\b\b")
def runTest(self): self.check_output()
class SimpleMirrorSameLine_ExceptCorrectResult(_VimTest):
snippets = ("test", "$1 $1")
wanted = "hallo hallo"
def cmd(self):
self.type("test\thallo")
def runTest(self): self.check_output()
class SimpleMirrorSameLineMany_ExceptCorrectResult(_VimTest):
snippets = ("test", "$1 $1 $1")
wanted = "hallo du hallo du hallo du hallo du"
def cmd(self):
self.type("test\thallo du")
class SimpleMirrorSameLineManyMultiline_ExceptCorrectResult(_VimTest):
snippets = ("test", "$1 $1 $1 $1")
wanted = "hallo du\nwie gehts? hallo du\nwie gehts? hallo du\nwie gehts?" \
" hallo du\nwie gehts?"
def cmd(self):
self.type("test\thallo du\nwie gehts?")
def runTest(self): self.check_output()
class SimpleMirrorDeleteSomeEnterSome_ExceptCorrectResult(_VimTest):
snippets = ("test", "$1\n$1")
wanted = "halhups\nhalhups"
def cmd(self):
self.type("test\thallo\b\bhups")
def runTest(self): self.check_output()
class SimpleTabstopWithDefaultSimpelType_ExceptCorrectResult(_VimTest):
snippets = ("test", "ha ${1:defa}\n$1")
wanted = "ha world\nworld"
def cmd(self):
self.type("test\tworld")
def runTest(self): self.check_output()
class SimpleTabstopWithDefaultComplexType_ExceptCorrectResult(_VimTest):
snippets = ("test", "ha ${1:default value} $1\nanother: $1 mirror")
wanted = "ha world world\nanother: world mirror"
def cmd(self):
self.type("test\tworld")
def runTest(self): self.check_output()
class SimpleTabstopWithDefaultSimpelKeep_ExceptCorrectResult(_VimTest):
snippets = ("test", "ha ${1:defa}\n$1")
wanted = "ha defa\ndefa"
def cmd(self):
self.type("test\t")
def runTest(self): self.check_output()
class SimpleTabstopWithDefaultComplexKeep_ExceptCorrectResult(_VimTest):
snippets = ("test", "ha ${1:default value} $1\nanother: $1 mirror")
wanted = "ha default value default value\nanother: default value mirror"
def cmd(self):
self.type("test\t")
def runTest(self): self.check_output()
# TODO: Mehrer tabs und mehrere mirrors
class TabstopWithMirrorInDefaultNoType_ExceptCorrectResult(_VimTest):
snippets = ("test", "ha ${1:blub} ${2:$1.h}")
wanted = "ha blub blub.h"
def cmd(self):
self.type("test\t")
def runTest(self): self.check_output()
class TabstopWithMirrorInDefaultTwiceAndExtra_ExceptCorrectResult(_VimTest):
snippets = ("test", "ha $1 ${2:$1.h $1.c}\ntest $1")
wanted = "ha stdin stdin.h stdin.c\ntest stdin"
def cmd(self):
self.type("test\tstdin")
def runTest(self): self.check_output()
class TabstopWithMirrorInDefaultMultipleLeave_ExceptCorrectResult(_VimTest):
snippets = ("test", "ha $1 ${2:snip} ${3:$1.h $2}")
wanted = "ha stdin snip stdin.h snip"
def cmd(self):
self.type("test\tstdin")
def runTest(self): self.check_output()
class TabstopWithMirrorInDefaultMultipleOverwrite_ExceptCorrectResult(_VimTest):
snippets = ("test", "ha $1 ${2:snip} ${3:$1.h $2}")
wanted = "ha stdin do snap stdin.h do snap"
def cmd(self):
self.type("test\tstdin\tdo snap")
def runTest(self): self.check_output()
class TabstopWithMirrorInDefaultOverwrite_ExceptCorrectResult(_VimTest):
snippets = ("test", "ha $1 ${2:$1.h}")
wanted = "ha stdin overwritten"
def cmd(self):
self.type("test\tstdin\toverwritten")
def runTest(self): self.check_output()
class MirrorRealLifeExample_ExceptCorrectResult(_VimTest):
snippets = (
("for", "for(size_t ${2:i} = 0; $2 < ${1:count}; ${3:++$2})" \
"\n{\n\t${0:/* code */}\n}"),
)
wanted = """for(size_t a_variable = 0; a_variable < 100; a_variable *= 2)
{
\t// do nothing
}"""
def cmd(self):
self.type("for\t100\tavar\b\b\b\ba_variable\ta_variable *= 2"
"\t// do nothing")
def runTest(self): self.check_output()
# TODO: recursive expansion
###################
# TRANSFORMATIONS #
###################
class Transformation_SimpleCase_ExceptCorrectResult(_VimTest):
snippets = ("test", "$1 ${1/foo/batzl/}")
wanted = "hallo foo boy hallo batzl boy"
def cmd(self):
self.type("test\thallo foo boy")
def runTest(self): self.check_output()
class Transformation_SimpleCaseNoTransform_ExceptCorrectResult(_VimTest):
snippets = ("test", "$1 ${1/foo/batzl/}")
wanted = "hallo hallo"
def cmd(self):
self.type("test\thallo")
def runTest(self): self.check_output()
class Transformation_Backreference_ExceptCorrectResult(_VimTest):
snippets = ("test", "$1 ${1/([ab])oo/$1ull/}")
wanted = "foo boo aoo foo bull aoo"
def cmd(self):
self.type("test\tfoo boo aoo")
def runTest(self): self.check_output()
class Transformation_BackreferenceTwice_ExceptCorrectResult(_VimTest):
snippets = ("test", r"$1 ${1/(dead) (par[^ ]*)/this $2 is a bit $1/}")
wanted = "dead parrot this parrot is a bit dead"
def cmd(self):
self.type("test\tdead parrot")
def runTest(self): self.check_output()
class Transformation_CleverTransformUpercaseChar_ExceptCorrectResult(_VimTest):
snippets = ("test", "$1 ${1/(.)/\u$1/}")
wanted = "hallo Hallo"
def cmd(self):
self.type("test\thallo")
def runTest(self): self.check_output()
class Transformation_CleverTransformLowercaseChar_ExceptCorrectResult(_VimTest):
snippets = ("test", "$1 ${1/(.*)/\l$1/}")
wanted = "Hallo hallo"
def cmd(self):
self.type("test\tHallo")
def runTest(self): self.check_output()
class Transformation_CleverTransformLongUpper_ExceptCorrectResult(_VimTest):
snippets = ("test", "$1 ${1/(.*)/\U$1\E/}")
wanted = "hallo HALLO"
def cmd(self):
self.type("test\thallo")
def runTest(self): self.check_output()
class Transformation_CleverTransformLongLower_ExceptCorrectResult(_VimTest):
snippets = ("test", "$1 ${1/(.*)/\L$1\E/}")
wanted = "HALLO hallo"
def cmd(self):
self.type("test\tHALLO")
def runTest(self): self.check_output()
# TODO: python init replacement
# class Transformation_RealLifeExample_ExceptCorrectResult(_VimTest):
# snippets = ("func", "def func$1 ${1/foo/batzl/}")
# wanted = "hallo hallo"
# def cmd(self):
# self.type("test\thallo")
# def runTest(self): self.check_output()
2009-07-01 09:20:40 -04:00
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
test_loader = unittest.TestLoader()
all_test_suites = test_loader.loadTestsFromModule(__import__("test"))
2009-07-01 09:20:40 -04:00
# 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