From 49fb981e7fca171b14bfa4287e43cd9d8fb043fe Mon Sep 17 00:00:00 2001
From: Florian Bruhin <git@the-compiler.org>
Date: Tue, 10 May 2016 07:12:50 +0200
Subject: [PATCH] Add test for completion count in cmdutils.register

---
 tests/unit/commands/test_cmdutils.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/tests/unit/commands/test_cmdutils.py b/tests/unit/commands/test_cmdutils.py
index 4c86333d4..5fa0c8c8f 100644
--- a/tests/unit/commands/test_cmdutils.py
+++ b/tests/unit/commands/test_cmdutils.py
@@ -173,6 +173,13 @@ class TestRegister:
             pass
         assert cmdutils.cmd_dict['fun'].hide
 
+    def test_wrong_completion_count(self):
+        with pytest.raises(ValueError):
+            @cmdutils.register(completion=['one', 'two'])
+            def fun(arg):
+                """Blah."""
+                pass
+
     def test_star_args(self):
         """Check handling of *args"""
         @cmdutils.register()