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);
}
Aggregations