Skip to content
This repository was archived by the owner on Apr 15, 2026. It is now read-only.

Optimized JWT unit tests #16

Merged
corbadoman merged 2 commits into from
May 23, 2025
Merged

Optimized JWT unit tests #16

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
aad2705
Optimized JWT unit tests
corbadoman May 23, 2025
cbb3ee2
Bugfix
corbadoman May 23, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
31 changes: 23 additions & 8 deletions src/test/java/com/corbado/unit/SessionServiceTest.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,10 +19,10 @@
import java.security.spec.PKCS8EncodedKeySpec;
import java.util.ArrayList;
import java.util.Base64;
import java.util.Collections;
import java.util.Date;
import java.util.List;
import java.util.Map;
import java.util.Collections;
import java.util.stream.Stream;

import org.junit.jupiter.api.BeforeAll;
Expand All @@ -42,6 +42,7 @@
import com.auth0.jwk.SigningKeyNotFoundException;
import com.auth0.jwt.JWT;
import com.auth0.jwt.algorithms.Algorithm;
import com.auth0.jwt.exceptions.AlgorithmMismatchException;
import com.auth0.jwt.exceptions.IncorrectClaimException;
import com.auth0.jwt.exceptions.JWTDecodeException;
import com.auth0.jwt.exceptions.JWTVerificationException;
Expand Down Expand Up @@ -143,7 +144,7 @@ void test_testDataIsPresent() throws InvalidKeySpecException, NoSuchAlgorithmExc
*/
@Test
void test_testGenerateJwt() throws InvalidKeySpecException, NoSuchAlgorithmException {
assertNotNull(generateJwt("1", 3, 4));
assertNotNull(generateJwt("1", 3, 4, Algorithm.RSA256(privateKey)));
}

/**
Expand Down Expand Up @@ -243,7 +244,8 @@ static List<Object[]> provideJwts() throws InvalidKeySpecException, NoSuchAlgori
generateJwt(
"https://auth.acme.com",
System.currentTimeMillis() / 1000 + 100,
System.currentTimeMillis() / 1000 + 100),
System.currentTimeMillis() / 1000 + 100,
Algorithm.RSA256(privateKey)),
IncorrectClaimException.class
});

Expand All @@ -253,7 +255,8 @@ static List<Object[]> provideJwts() throws InvalidKeySpecException, NoSuchAlgori
generateJwt(
"https://auth.acme.com",
System.currentTimeMillis() / 1000 - 100,
System.currentTimeMillis() / 1000 - 100),
System.currentTimeMillis() / 1000 - 100,
Algorithm.RSA256(privateKey)),
TokenExpiredException.class
});

Expand All @@ -263,17 +266,30 @@ static List<Object[]> provideJwts() throws InvalidKeySpecException, NoSuchAlgori
generateJwt(
"https://invalid.com",
System.currentTimeMillis() / 1000 + 100,
System.currentTimeMillis() / 1000 - 100),
System.currentTimeMillis() / 1000 - 100,
Algorithm.RSA256(privateKey)),
IncorrectClaimException.class
});

// Invalid alg "none"
testData.add(
new Object[] {
generateJwt(
"https://auth.acme.com",
System.currentTimeMillis() / 1000 + 100,
System.currentTimeMillis() / 1000 - 100,
Algorithm.none()),
AlgorithmMismatchException.class
});

// Success
testData.add(
new Object[] {
generateJwt(
"https://auth.acme.com",
System.currentTimeMillis() / 1000 + 100,
System.currentTimeMillis() / 1000 - 100),
System.currentTimeMillis() / 1000 - 100,
Algorithm.RSA256(privateKey)),
null
});

Expand Down Expand Up @@ -317,10 +333,9 @@ private static RSAPrivateKey readPrivateKey(final String privateKeyPath)
* @throws InvalidKeySpecException the invalid key spec exception
* @throws NoSuchAlgorithmException the no such algorithm exception
*/
private static String generateJwt(final String iss, final long exp, final long nbf)
private static String generateJwt(final String iss, final long exp, final long nbf, final Algorithm algorithm)
throws InvalidKeySpecException, NoSuchAlgorithmException {

final Algorithm algorithm = Algorithm.RSA256(privateKey);
return JWT.create()
.withHeader(Collections.singletonMap("kid", "kid123"))
.withIssuer(iss)
Expand Down