Locust passing headers

Question

Is it possible to have Locust pass a header command with a secure token to load test an API?

I am trying to test our api for an encoder with a header flag for a token as the server being tested has to receive a token with the request, ie.

curl -H “Authorization: Token token string” http://someserver

How are you doing this currently? `self.client.get(uri, headers={“Authorization": "Token token string”})` should work — enderland, Aug 24 '18 at 16:39
@uberrebu you'd instead use the `auth` field, e.g. `auth=("username", "password")` and leave the `headers` field as `None`. — Matt Doyle, Sep 29 '20 at 23:40
actually got it eventually after searching around..yup correct...thanks! — uberrebu, Sep 30 '20 at 01:14

Matt Doyle · Answer 1 · 2019-03-04T19:55:55.880

45

Yes, you can use:

token_string = "token string"

resp = self.client.post(
            url="http://someserver",
            data=json.dumps(data),
            auth=None,
            headers={"authorization": "Token " + token_string},
            name="http://someserver",
        )

edited Mar 04 '19 at 19:55

answered Mar 04 '19 at 19:34

Matt Doyle

867
8
12

matusko · Answer 2 · 2022-11-02T21:15:02.777

20

Additionally, if you want to use same headers for every request you can also set them to the client in the on_start method. They will be automatically used with every client request.

class User(HttpUser):

    def on_start(self):
        self.client.headers = {'Authorization': 'my-auth-token'}

    @task
    def my_authenticated_task(self):
        self.client.post('enspoint')  # this will use headers we set earlier

edited Nov 02 '22 at 21:15

answered Aug 09 '21 at 09:33

matusko

3,487
3
20
31

1

How do you do this for FastHttpUser? – Shinto C V Apr 26 '23 at 05:25

score 1 · Answer 3 · answered Sep 24 '21 at 06:00

class User(HttpUser):

    host = 'http://127.0.0.1:8000'      
    @task
    def predict_post(self):
        self.client.post("/route/", data=json.dumps({
        "text": "finish all the track to hear and great good tune u do a great job alan walker",
        "max_op_words": 10,
        "max_time": 1,
        "max_predictions": 1
        }),auth=None,
        headers={"authtoken":"your-auth-token", 'content-type': 'application/json'})

I was missing out on content-type header and was getting a 401 error authorization error.

Locust passing headers

3 Answers3