Add test for decorating a method (re #51)

neithere · Jan 2, 2014 · c2248ca · c2248ca
1 parent ab3856f
commit c2248ca
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/test/test_decorators.py b/test/test_decorators.py
@@ -37,6 +37,15 @@ def func():
     assert attr == 'new-name'
 
 
+def test_named_method():
+    class A:
+        @argh.named('new-name')
+        def meth(self):
+            pass
+    assert getattr(A.meth, argh.constants.ATTR_NAME) == 'new-name'
+    assert getattr(A().meth, argh.constants.ATTR_NAME) == 'new-name'
+
+
 def test_command():
     @argh.command
     def func():