blob: 5d667678a4f630f416fefda13c94b5348edc5c7f (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
# coding=utf-8
# flake8: noqa E302
"""
Unit testing for get_defining_class in cmd2/utils.py module.
"""
import functools
import cmd2.utils as cu
class ParentClass(object):
def func_with_overrides(self):
pass
def parent_only_func(self, param1, param2):
pass
class ChildClass(ParentClass):
def func_with_overrides(self):
super(ChildClass, self).func_with_overrides()
def child_function(self):
pass
lambda1 = lambda: 1
lambda2 = (lambda: lambda: 2)()
@classmethod
def class_method(cls):
pass
@staticmethod
def static_meth():
pass
def func_not_in_class():
pass
def test_get_defining_class():
parent_instance = ParentClass()
child_instance = ChildClass()
# validate unbound class functions
assert cu.get_defining_class(ParentClass.func_with_overrides) is ParentClass
assert cu.get_defining_class(ParentClass.parent_only_func) is ParentClass
assert cu.get_defining_class(ChildClass.func_with_overrides) is ChildClass
assert cu.get_defining_class(ChildClass.parent_only_func) is ParentClass
assert cu.get_defining_class(ChildClass.child_function) is ChildClass
assert cu.get_defining_class(ChildClass.class_method) is ChildClass
assert cu.get_defining_class(ChildClass.static_meth) is ChildClass
# validate bound class methods
assert cu.get_defining_class(parent_instance.func_with_overrides) is ParentClass
assert cu.get_defining_class(parent_instance.parent_only_func) is ParentClass
assert cu.get_defining_class(child_instance.func_with_overrides) is ChildClass
assert cu.get_defining_class(child_instance.parent_only_func) is ParentClass
assert cu.get_defining_class(child_instance.child_function) is ChildClass
assert cu.get_defining_class(child_instance.class_method) is ChildClass
assert cu.get_defining_class(child_instance.static_meth) is ChildClass
# bare functions resolve to nothing
assert cu.get_defining_class(func_not_in_class) is None
# lambdas and nested lambdas
assert cu.get_defining_class(ChildClass.lambda1) is ChildClass
assert cu.get_defining_class(ChildClass.lambda2) is ChildClass
assert cu.get_defining_class(ChildClass().lambda1) is ChildClass
assert cu.get_defining_class(ChildClass().lambda2) is ChildClass
# partials
partial_unbound = functools.partial(ParentClass.parent_only_func, 1)
nested_partial_unbound = functools.partial(partial_unbound, 2)
assert cu.get_defining_class(partial_unbound) is ParentClass
assert cu.get_defining_class(nested_partial_unbound) is ParentClass
partial_bound = functools.partial(parent_instance.parent_only_func, 1)
nested_partial_bound = functools.partial(partial_bound, 2)
assert cu.get_defining_class(partial_bound) is ParentClass
assert cu.get_defining_class(nested_partial_bound) is ParentClass
|
