Fix dests of flags added by FlagAdder

raviqqe · Jan 11, 2017 · 1ce1dc3 · 1ce1dc3
1 parent 1e89026
commit 1ce1dc3
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/qnd/flag.py b/qnd/flag.py
@@ -45,7 +45,7 @@ def add_flag(self, name, *args, **kwargs):
         See `add_flag()`.
         """
         add_flag(name, *args, **kwargs)
-        self._flags.append(name)
+        self._flags.append(kwargs.get("dest") or name)
 
     def add_required_flag(self, name, *args, **kwargs):
         """Add a required flag.

diff --git a/qnd/flag_test.py b/qnd/flag_test.py
@@ -0,0 +1,11 @@
+import sys
+
+from .flag import FlagAdder
+
+
+def test_flag_adder():
+    sys.argv = ["command", "--foo", "baz"]
+
+    adder = FlagAdder()
+    adder.add_flag("foo", dest="bar")
+    assert adder.flags["bar"] == "baz"