use of org.cloudfoundry.identity.uaa.mfa.UserGoogleMfaCredentialsProvisioning in project uaa by cloudfoundry.
the class TotpMfaEndpointTest method setup.
@Before
public void setup() {
userId = new RandomValueStringGenerator(5).generate();
userGoogleMfaCredentialsProvisioning = mock(UserGoogleMfaCredentialsProvisioning.class);
mfaProviderProvisioning = mock(MfaProviderProvisioning.class);
uaaAuthentication = mock(UaaAuthentication.class);
mfaProvider = new MfaProvider();
mfaProvider.setName("provider-name");
mfaProvider.setId("provider_id1");
mfaProvider.setConfig(new GoogleMfaProviderConfig());
mfaProvider.setType(MfaProvider.MfaProviderType.GOOGLE_AUTHENTICATOR);
otherMfaProvider = new MfaProvider();
otherMfaProvider.setName("other-provider-name");
otherMfaProvider.setId("provider_id2");
otherMfaProvider.setConfig(new GoogleMfaProviderConfig());
otherMfaProvider.setType(MfaProvider.MfaProviderType.GOOGLE_AUTHENTICATOR);
mockSuccessHandler = mock(SavedRequestAwareAuthenticationSuccessHandler.class);
SecurityContextHolder.getContext().setAuthentication(uaaAuthentication);
publisher = mock(ApplicationEventPublisher.class);
eventCaptor = ArgumentCaptor.forClass(ApplicationEvent.class);
doNothing().when(publisher).publishEvent(eventCaptor.capture());
userDb = mock(UaaUserDatabase.class);
mockMfaPolicy = mock(CommonLoginPolicy.class);
when(mockMfaPolicy.isAllowed(anyString())).thenReturn(new LoginPolicy.Result(true, 0));
endpoint = new TotpMfaEndpoint(userGoogleMfaCredentialsProvisioning, mfaProviderProvisioning, "/login/mfa/completed", userDb, mockMfaPolicy);
endpoint.setApplicationEventPublisher(publisher);
}
Aggregations