Search in sources :

Example 1 with JWTBuilder

use of io.gravitee.am.jwt.JWTBuilder in project gravitee-access-management by gravitee-io.

the class JWTServiceTest method setUp.

@Before
public void setUp() {
    JWTBuilder rs256JWTBuilder = mock(JWTBuilder.class);
    JWTBuilder rs512JWTBuilder = mock(JWTBuilder.class);
    JWTBuilder defaultJWTBuilder = mock(JWTBuilder.class);
    JWTBuilder noneAlgBuilder = mock(JWTBuilder.class);
    when(rs256JWTBuilder.sign(any())).thenReturn("token_rs_256");
    when(rs512JWTBuilder.sign(any())).thenReturn("token_rs_512");
    when(defaultJWTBuilder.sign(any())).thenReturn("token_default");
    when(noneAlgBuilder.sign(any())).thenReturn("not_signed_jwt");
    io.gravitee.am.gateway.certificate.CertificateProvider rs256CertProvider = mock(io.gravitee.am.gateway.certificate.CertificateProvider.class);
    io.gravitee.am.gateway.certificate.CertificateProvider rs512CertProvider = mock(io.gravitee.am.gateway.certificate.CertificateProvider.class);
    io.gravitee.am.gateway.certificate.CertificateProvider defaultCertProvider = mock(io.gravitee.am.gateway.certificate.CertificateProvider.class);
    io.gravitee.am.gateway.certificate.CertificateProvider noneAlgCertProvider = mock(io.gravitee.am.gateway.certificate.CertificateProvider.class);
    when(rs256CertProvider.getJwtBuilder()).thenReturn(rs256JWTBuilder);
    when(rs512CertProvider.getJwtBuilder()).thenReturn(rs512JWTBuilder);
    when(defaultCertProvider.getJwtBuilder()).thenReturn(defaultJWTBuilder);
    when(noneAlgCertProvider.getJwtBuilder()).thenReturn(noneAlgBuilder);
    when(certificateManager.findByAlgorithm("unknown")).thenReturn(Maybe.empty());
    when(certificateManager.findByAlgorithm("RS512")).thenReturn(Maybe.just(rs512CertProvider));
    when(certificateManager.get(null)).thenReturn(Maybe.empty());
    when(certificateManager.get("notExistingId")).thenReturn(Maybe.empty());
    when(certificateManager.get("existingId")).thenReturn(Maybe.just(rs256CertProvider));
    when(certificateManager.defaultCertificateProvider()).thenReturn(defaultCertProvider);
    when(certificateManager.noneAlgorithmCertificateProvider()).thenReturn(noneAlgCertProvider);
}
Also used : JWTBuilder(io.gravitee.am.jwt.JWTBuilder) Before(org.junit.Before)

Aggregations

JWTBuilder (io.gravitee.am.jwt.JWTBuilder)1 Before (org.junit.Before)1