Token Refresh - Java - Example

GeMi · May 10, 2023, 10:03pm

I see that lot of people asking about solution to refresh Token when expired.
I have prepared Case0022SetOrRefreshTokenSimulation where Token is refreshed every some period of time.

What you think about this solution @slandelle @sbrevet ?

github.com

gemiusz/gatling-examples-maven-java/blob/master/src/test/java/pl/gemiusz/Case0022SetOrRefreshTokenSimulation.java

package pl.gemiusz;

import io.gatling.javaapi.core.OpenInjectionStep;
import io.gatling.javaapi.core.ScenarioBuilder;
import io.gatling.javaapi.core.Simulation;
import io.gatling.javaapi.http.HttpProtocolBuilder;

import java.time.Duration;
import java.util.Collections;
import java.util.Iterator;
import java.util.Map;
import java.util.UUID;
import java.util.function.Supplier;
import java.util.stream.Stream;

import static io.gatling.javaapi.core.CoreDsl.*;
import static io.gatling.javaapi.http.HttpDsl.http;

public class Case0022SetOrRefreshTokenSimulation extends Simulation {

This file has been truncated. show original

sbrevet · May 11, 2023, 8:00am

Hi @GeMi,

Thanks for this code suggestion!

Here my comments:

Create a pull request: it is easier to comment and discuss about code there.
I think it needs a disclaimer that is for a single load generator (in Gatling Enterprise, you can launch a single run with multiple load generators, but global variables are trickier to manage)
I’m on two minds about the scnForSetOrRefreshToken:
Perhaps a scenario with a single user that loops with a pause?
- In one hand, your code doesn’t need to have a max duration
- In the other, it’s complicated to pre-compute the amount of requests this way
A small comment next to nothingFor(2) : “waiting for the first credentials”

Note:

Avoid string injection in the URL by using queryParam (personal taste)

Cheers!

GeMi · May 11, 2023, 8:15pm

Hi @sbrevet,
Thank you for your valuable suggestions
I have make huge changes in code including your suggerstions
If it is possible please check my PR → Case0022SetOrRefreshTokenSimulation in new version by gemiusz · Pull Request #17 · gemiusz/gatling-examples-maven-java · GitHub

In few words what it’s going in Simulation now:

I changed injection profile to have dependencies on necessary steps: first get token → token refresh loop AND Core Scenario → after finish my simulation “shut down” token refresh loop.
Simulation time = time of getting first token + Core Scenario time + 1 second (shut down token refresh loop)
Don’t need set any tricky counters etc.
Set queryParam
Silent request for token - that was from beginning but I want to mention about that here

Cheers!

sbrevet · May 12, 2023, 7:12am

Hi @GeMi,

Nice improvements!

Cheers!

GeMi · May 12, 2023, 10:05am

@sbrevet thank you for review and comments in PR.
I will look closely at them at Sunday afternoon because I’m at trip

Cheers!