Test FlagArgument and ValueArgument

Refs: #4058

diff --git a/kamaki/cli/argument/test.py b/kamaki/cli/argument/test.py
index f56fed7..2d5905b 100644 (file)
--- a/kamaki/cli/argument/test.py
+++ b/kamaki/cli/argument/test.py
@@ -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):
+
+    @patch('kamaki.cli.argument.Argument.__init__')
+    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):
+
+    @patch('kamaki.cli.argument.Argument.__init__')
+    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:])

diff --git a/kamaki/cli/test.py b/kamaki/cli/test.py
index de51c79..b66c5a3 100644 (file)
--- a/kamaki/cli/test.py
+++ b/kamaki/cli/test.py
@@ -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,
+    ValueArgument)
 
 
 #  TestCase auxiliary methods