optimum-habana

1
class Secret:
2
    """
3
    Taken from: https://stackoverflow.com/a/67393351
4
    """
5

6
    def __init__(self, value):
7
        self.value = value
8

9
    def __repr__(self):
10
        return "Secret(********)"
11

12
    def __str___(self):
13
        return "*******"
14

15

16
def pytest_addoption(parser):
17
    parser.addoption("--token", action="store", default=None)
18

19

20
def pytest_generate_tests(metafunc):
21
    # This is called for every test. Only get/set command line arguments
22
    # if the argument is specified in the list of test "fixturenames".
23
    option_value = Secret(metafunc.config.option.token)
24
    if "token" in metafunc.fixturenames:
25
        metafunc.parametrize("token", [option_value])
26