Passing self in a decorator with params

Question

Am trying to set a decorator on a unit test, the purpose of the decorator is to ensure that a user logins in before accessing other endpoints.

Below is the decorator for logging in :

def login_decorator(*args, **kwargs):
    def login_user():
        kwargs['self'].client.login(
            username=kwargs['email'],
            password=kwargs['password'],
        )

Then am trying to utilize it on the unit test function as below :

@login_decorator(self=self, email=self.email, password=self.password)
    def test_customer_profile_add(self):
        response = super().create_customer_profile(self.customer_data)
        self.assertEqual(response.status_code, status.HTTP_201_CREATED)

however self is underlines with an error Unresolved reference , how can I achieve passing of the params through the decorator with self.

There are many problems here. I think you would do well to start by reviewing a [general guide to writing decorators with arguments](https://stackoverflow.com/a/1594484) (link is to one of the "classic" highly-rated Stack Overflow answers). — Karl Knechtel, Mar 18 '21 at 21:58

Phoenix · Accepted Answer · 2021-03-18T22:05:10.227

You should pass your params to login_user function:


def login_decorator(func):
    def login_user(self, *args, **kwargs):
        # function args
        print(self.username, self.password)
        self.client.login(
            username=self.username,
            password=self.password,
        )
        func(self, *args, **kwargs)

and your test function:

@login_decorator
def test_customer_profile_add(self):
    response = super().create_customer_profile(self.customer_data)
    self.assertEqual(response.status_code, status.HTTP_201_CREATED)

Passing self in a decorator with params

1 Answers1