Examples with SimpleAccountRealm org.apache.shiro.realm.SimpleAccountRealm used on opensource projects

Example 1 with SimpleAccountRealm

use of org.apache.shiro.realm.SimpleAccountRealm in project graylog2-server by Graylog2.

the class SessionCreatorTest method serviceUnavailableStateIsCleared.

/**
 * Test that the service unavailable exception is cleared when the service becomes available again
 */
@Test
public void serviceUnavailableStateIsCleared() {
    setUpUserMock();
    assertFalse(SecurityUtils.getSubject().isAuthenticated());
    final AtomicBoolean doThrow = new AtomicBoolean(true);
    final SimpleAccountRealm switchableRealm = new SimpleAccountRealm() {

        @Override
        protected AuthenticationInfo doGetAuthenticationInfo(AuthenticationToken token) throws AuthenticationException {
            if (doThrow.get()) {
                throw new AuthenticationServiceUnavailableException("not available");
            } else {
                return super.doGetAuthenticationInfo(token);
            }
        }
    };
    securityManager.setRealms(ImmutableList.of(switchableRealm, new SimpleAccountRealm()));
    // realm will throw an exception on auth attempt
    assertThatThrownBy(() -> sessionCreator.create(null, "host", validToken)).isInstanceOf(AuthenticationServiceUnavailableException.class);
    assertThat(SecurityUtils.getSubject().isAuthenticated()).isFalse();
    // switch realm to not throw an exception but simply reject the credentials
    doThrow.set(false);
    sessionCreator.create(null, "host", validToken);
    assertThat(SecurityUtils.getSubject().isAuthenticated()).isFalse();
}

Example 2 with SimpleAccountRealm

use of org.apache.shiro.realm.SimpleAccountRealm in project graylog2-server by Graylog2.

the class SessionCreatorTest method setUp.

@Before
public void setUp() throws Exception {
    MockitoAnnotations.initMocks(this);
    SimpleAccountRealm realm = new SimpleAccountRealm();
    realm.addAccount(validToken.getUsername(), String.valueOf(validToken.getPassword()));
    // Set up a security manager like in DefaultSecurityManagerProvider
    securityManager = new DefaultSecurityManager(realm);
    FirstSuccessfulStrategy strategy = new ThrowingFirstSuccessfulStrategy();
    strategy.setStopAfterFirstSuccess(true);
    ((ModularRealmAuthenticator) securityManager.getAuthenticator()).setAuthenticationStrategy(strategy);
    SecurityUtils.setSecurityManager(securityManager);
}