Code Duplication    Length = 15-15 lines in 3 locations

tests/helpers.py 3 locations

@@ 95-109 (lines=15) @@
92
93
94
# pylint: disable=unused-argument
95
def e2e(size='small'):
96
    """Handle tokens from requests."""
97
    env_test_size = os.environ.get("KYTOS_TESTS_SIZE", 'small')
98
    env_test_type = os.environ.get("KYTOS_TESTS_TYPE", 'unit')
99
100
    def inner_func(func):
101
102
        @wraps(func)
103
        def wrapper(*args, **kwargs):
104
            if env_test_type == 'e2e' and size == env_test_size:
105
                return func(*args, **kwargs)
106
            return None
107
        return wrapper
108
109
    return inner_func
110
@@ 77-91 (lines=15) @@
74
75
76
# pylint: disable=unused-argument
77
def integration(size='small'):
78
    """Handle tokens from requests."""
79
    env_test_size = os.environ.get("KYTOS_TESTS_SIZE", 'small')
80
    env_test_type = os.environ.get("KYTOS_TESTS_TYPE", 'unit')
81
82
    def inner_func(func):
83
84
        @wraps(func)
85
        def wrapper(*args, **kwargs):
86
            if env_test_type == 'integration' and size == env_test_size:
87
                return func(*args, **kwargs)
88
            return None
89
        return wrapper
90
91
    return inner_func
92
93
94
# pylint: disable=unused-argument
@@ 59-73 (lines=15) @@
56
57
58
# pylint: disable=unused-argument
59
def unit(size='small'):
60
    """Handle tokens from requests."""
61
    env_test_size = os.environ.get("KYTOS_TESTS_SIZE", 'small')
62
    env_test_type = os.environ.get("KYTOS_TESTS_TYPE", 'unit')
63
64
    def inner_func(func):
65
66
        @wraps(func)
67
        def wrapper(*args, **kwargs):
68
            if env_test_type == 'unit' and size == env_test_size:
69
                return func(*args, **kwargs)
70
            return None
71
        return wrapper
72
73
    return inner_func
74
75
76
# pylint: disable=unused-argument