Examples with TesterContext org.apache.tomcat.unittest.TesterContext used on opensource projects

Example 16 with TesterContext

use of org.apache.tomcat.unittest.TesterContext in project tomcat by apache.

the class TestJNDIRealm method buildRealm.

private JNDIRealm buildRealm(String password) throws javax.naming.NamingException, NoSuchFieldException, IllegalAccessException, LifecycleException {
    Context context = new TesterContext();
    JNDIRealm realm = new JNDIRealm();
    realm.setContainer(context);
    realm.setUserSearch("");
    // Usually everything is created in create() but that's not the case here
    Field field = JNDIRealm.class.getDeclaredField("singleConnection");
    field.setAccessible(true);
    Field field2 = JNDIRealm.JNDIConnection.class.getDeclaredField("context");
    field2.setAccessible(true);
    field2.set(field.get(realm), mockDirContext(mockSearchResults(password)));
    realm.start();
    return realm;
}

Example 17 with TesterContext

use of org.apache.tomcat.unittest.TesterContext in project tomcat by apache.

the class TestJNDIRealm method testErrorRealm.

@Test
public void testErrorRealm() throws Exception {
    Context context = new TesterContext();
    JNDIRealm realm = new JNDIRealm();
    realm.setContainer(context);
    realm.setUserSearch("");
    // Connect to something that will fail
    realm.setConnectionURL("ldap://127.0.0.1:12345");
    realm.start();
    final CountDownLatch latch = new CountDownLatch(3);
    (new Thread(() -> {
        realm.authenticate("foo", "bar");
        latch.countDown();
    })).start();
    (new Thread(() -> {
        realm.authenticate("foo", "bar");
        latch.countDown();
    })).start();
    (new Thread(() -> {
        realm.authenticate("foo", "bar");
        latch.countDown();
    })).start();
    Assert.assertTrue(latch.await(30, TimeUnit.SECONDS));
}

Example 18 with TesterContext

use of org.apache.tomcat.unittest.TesterContext in project tomcat by apache.

the class TestRealmBase method doRoleTest.

private void doRoleTest(List<String> userRoles, List<String> constraintOneRoles, List<String> constraintTwoRoles, List<String> applicationRoles, boolean expected) throws IOException {
    TesterMapRealm mapRealm = new TesterMapRealm();
    // Configure the security constraints for the resource
    SecurityConstraint constraintOne = new SecurityConstraint();
    if (constraintOneRoles != null) {
        constraintOne.setAuthConstraint(true);
        for (String constraintRole : constraintOneRoles) {
            constraintOne.addAuthRole(constraintRole);
            if (applicationRoles.contains(SecurityConstraint.ROLE_ALL_AUTHENTICATED_USERS)) {
                constraintOne.treatAllAuthenticatedUsersAsApplicationRole();
            }
        }
    }
    SecurityConstraint constraintTwo = new SecurityConstraint();
    if (constraintTwoRoles != null) {
        constraintTwo.setAuthConstraint(true);
        for (String constraintRole : constraintTwoRoles) {
            constraintTwo.addAuthRole(constraintRole);
            if (applicationRoles.contains(SecurityConstraint.ROLE_ALL_AUTHENTICATED_USERS)) {
                constraintTwo.treatAllAuthenticatedUsersAsApplicationRole();
            }
        }
    }
    SecurityConstraint[] constraints = new SecurityConstraint[] { constraintOne, constraintTwo };
    // Set up the mock request and response
    Request request = new Request(null);
    Response response = new TesterResponse();
    Context context = new TesterContext();
    for (String applicationRole : applicationRoles) {
        context.addSecurityRole(applicationRole);
    }
    request.getMappingData().context = context;
    // Configure the users in the Realm
    if (userRoles != null) {
        GenericPrincipal gp = new GenericPrincipal(USER1, userRoles);
        request.setUserPrincipal(gp);
    }
    // Check if user meets constraints
    boolean result = mapRealm.hasResourcePermission(request, response, constraints, null);
    Assert.assertEquals(Boolean.valueOf(expected), Boolean.valueOf(result));
}