Revision 3a5b1ceb

b/kamaki/cli/argument/test.py
191 191
            grp, key = (grp, key) if key else ('global', grp)
192 192
            self.assertEqual(override.mock_calls[-1], call(grp, key, val))
193 193

  
194

  
195
class FlagArgument(TestCase):
196

  
197
    @patch('kamaki.cli.argument.Argument.__init__')
198
    def test___init__(self, arg):
199
        help, pname, default = 'help', 'pname', 'default'
200
        fa = argument.FlagArgument(help, pname, default)
201
        self.assertTrue(isinstance(fa, argument.FlagArgument))
202
        arg.assert_called_once(0, help, pname, default)
203

  
204

  
205
class ValueArgument(TestCase):
206

  
207
    @patch('kamaki.cli.argument.Argument.__init__')
208
    def test___init__(self, arg):
209
        help, pname, default = 'help', 'pname', 'default'
210
        fa = argument.ValueArgument(help, pname, default)
211
        self.assertTrue(isinstance(fa, argument.ValueArgument))
212
        arg.assert_called_once(1, help, pname, default)
213

  
214

  
194 215
if __name__ == '__main__':
195 216
    from sys import argv
196 217
    from kamaki.cli.test import runTestCase
197 218
    runTestCase(Argument, 'Argument', argv[1:])
198 219
    runTestCase(ConfigArgument, 'ConfigArgument', argv[1:])
199 220
    runTestCase(RuntimeConfigArgument, 'RuntimeConfigArgument', argv[1:])
221
    runTestCase(FlagArgument, 'FlagArgument', argv[1:])
222
    runTestCase(FlagArgument, 'ValueArgument', argv[1:])
b/kamaki/cli/test.py
35 35
from inspect import getmembers, isclass
36 36

  
37 37
from kamaki.cli.command_tree.test import Command, CommandTree
38
from kamaki.cli.argument.test import Argument, ConfigArgument
39
from kamaki.cli.argument.test import RuntimeConfigArgument
38
from kamaki.cli.argument.test import (
39
    Argument, ConfigArgument, RuntimeConfigArgument, FlagArgument,
40
    ValueArgument)
40 41

  
41 42

  
42 43
#  TestCase auxiliary methods

Also available in: Unified diff