Commit 3a5b1ceb authored by Stavros Sachtouris's avatar Stavros Sachtouris
Browse files

Test FlagArgument and ValueArgument

Refs: #4058
parent ea7f6e36
......@@ -191,9 +191,32 @@ class RuntimeConfigArgument(TestCase):
grp, key = (grp, key) if key else ('global', grp)
self.assertEqual(override.mock_calls[-1], call(grp, key, val))
class FlagArgument(TestCase):
def test___init__(self, arg):
help, pname, default = 'help', 'pname', 'default'
fa = argument.FlagArgument(help, pname, default)
self.assertTrue(isinstance(fa, argument.FlagArgument))
arg.assert_called_once(0, help, pname, default)
class ValueArgument(TestCase):
def test___init__(self, arg):
help, pname, default = 'help', 'pname', 'default'
fa = argument.ValueArgument(help, pname, default)
self.assertTrue(isinstance(fa, argument.ValueArgument))
arg.assert_called_once(1, help, pname, default)
if __name__ == '__main__':
from sys import argv
from kamaki.cli.test import runTestCase
runTestCase(Argument, 'Argument', argv[1:])
runTestCase(ConfigArgument, 'ConfigArgument', argv[1:])
runTestCase(RuntimeConfigArgument, 'RuntimeConfigArgument', argv[1:])
runTestCase(FlagArgument, 'FlagArgument', argv[1:])
runTestCase(FlagArgument, 'ValueArgument', argv[1:])
......@@ -35,8 +35,9 @@ from unittest import makeSuite, TestSuite, TextTestRunner, TestCase
from inspect import getmembers, isclass
from kamaki.cli.command_tree.test import Command, CommandTree
from kamaki.cli.argument.test import Argument, ConfigArgument
from kamaki.cli.argument.test import RuntimeConfigArgument
from kamaki.cli.argument.test import (
Argument, ConfigArgument, RuntimeConfigArgument, FlagArgument,
# TestCase auxiliary methods
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment