Extend test_mock_decorator for class method.

This commit is contained in:
Christian Walch 2022-04-01 07:37:07 +02:00
parent 9fd40d3ab3
commit 57d6ec7697

View File

@ -114,6 +114,11 @@ def test_mock_decorator():
def meth(self): def meth(self):
pass pass
@classmethod
@mock.method_deco
def class_meth(cls):
pass
@mock.class_deco @mock.class_deco
class Bar: class Bar:
pass pass
@ -124,6 +129,7 @@ def test_mock_decorator():
assert undecorate(func).__name__ == "func" assert undecorate(func).__name__ == "func"
assert undecorate(Foo.meth).__name__ == "meth" assert undecorate(Foo.meth).__name__ == "meth"
assert undecorate(Foo.class_meth).__name__ == "class_meth"
assert undecorate(Bar).__name__ == "Bar" assert undecorate(Bar).__name__ == "Bar"
assert undecorate(Baz).__name__ == "Baz" assert undecorate(Baz).__name__ == "Baz"