Test also the result of the test() function.

This commit is contained in:
Bastian Kleineidam 2010-03-11 11:13:51 +01:00
parent e2cb9208d1
commit 5c5a5c36fe
1 changed files with 12 additions and 8 deletions

View File

@ -23,29 +23,33 @@ class TestBaker (unittest.TestCase):
@baker.command @baker.command
def func(a, b, c): def func(a, b, c):
return (a, b, c) return (a, b, c)
res = baker.run(argv=[__file__, 'func', '1', '2', '3']) argv = [__file__, 'func', '1', '2', '3']
self.assertEqual(res, ('1', '2', '3')) self.assertEqual(baker.test(argv=argv), "func('1','2','3')")
self.assertEqual(baker.run(argv=argv), ('1','2','3'))
def test_func_noargs (self): def test_func_noargs (self):
@baker.command @baker.command
def func(): def func():
return 42 return 42
res = baker.run(argv=[__file__, 'func']) argv = [__file__, 'func']
self.assertEqual(res, 42) self.assertEqual(baker.test(argv=argv), "func()")
self.assertEqual(baker.run(argv=argv), 42)
def test_func_kwargs (self): def test_func_kwargs (self):
@baker.command @baker.command
def func(arg1, arg2, *args, **kwargs): def func(arg1, arg2, *args, **kwargs):
return arg1, arg2, kwargs['verbose'] return arg1, arg2, kwargs['verbose']
res = baker.run(argv=[__file__, 'func', 'argvalue1', 'argvalue2', '--verbose']) argv = [__file__, 'func', 'argvalue1', 'argvalue2', '--verbose']
self.assertEqual(res, ('argvalue1', 'argvalue2', True)) self.assertEqual(baker.test(argv=argv), "func('argvalue1','argvalue2',verbose=True)")
self.assertEqual(baker.run(argv=argv), ('argvalue1', 'argvalue2', True))
def test_func_kwargs_revorder (self): def test_func_kwargs_revorder (self):
@baker.command @baker.command
def func(arg1, arg2, *args, **kwargs): def func(arg1, arg2, *args, **kwargs):
return arg1, arg2, kwargs['verbose'] return arg1, arg2, kwargs['verbose']
res = baker.run(argv=[__file__, 'func', 'argvalue1', '--verbose', 'argvalue2']) argv = [__file__, 'func', 'argvalue1', '--verbose', 'argvalue2']
self.assertEqual(res, ('argvalue1', 'argvalue2', True)) self.assertEqual(baker.test(argv=argv), "func('argvalue1','argvalue2',verbose=True)")
self.assertEqual(baker.run(argv=argv), ('argvalue1', 'argvalue2', True))
def test_func_kwargs_params (self): def test_func_kwargs_params (self):
@baker.command(shortopts={"verbose": "v"}, params={"verbose": "Be verbose"}) @baker.command(shortopts={"verbose": "v"}, params={"verbose": "Be verbose"})