如何以编程方式在 OpenEJB 中指定我的用户/角色？

Question

在我正在从事的项目中，我们使用 OpenEJB 作为框架来测试我们的 EJB。我们以编程方式构造一个InitialContext，并使用它来获取事务管理器和各种EJB。

但是，我现在必须测试的 EJB 具有 @RolesAllowed 注释，因此 OpenEJB 拒绝获取该 EJB，认为我没有所需的权限。

我如何向 OpenEJB 指定此测试要模拟的用户以及与他关联的角色？

Answer 1

bkail 提到的 @RunAs 建议绝对是一个好方法。第二种不涉及内部类的方法是登录测试用例。

登录方法

当您引导 OpenEJB 时，请在 InitialContext 属性中指定用户/密码，如下所示：

public void testAsManager() throws Exception {
    Properties p = new Properties();
    p.put(Context.INITIAL_CONTEXT_FACTORY, "org.apache.openejb.client.LocalInitialContextFactory");
    p.put(Context.SECURITY_PRINCIPAL, "jane");
    p.put(Context.SECURITY_CREDENTIALS, "waterfall");

    InitialContext context = new InitialContext(p);

    try {
        movies.addMovie(new Movie("Quentin Tarantino", "Reservoir Dogs", 1992));
        movies.addMovie(new Movie("Joel Coen", "Fargo", 1996));
        movies.addMovie(new Movie("Joel Coen", "The Big Lebowski", 1998));

        List<Movie> list = movies.getMovies();
        assertEquals("List.size()", 3, list.size());

        for (Movie movie : list) {
            movies.deleteMovie(movie);
        }

        assertEquals("Movies.getMovies()", 0, movies.getMovies().size());
    } finally {
        context.close();
    }
}

然后也许以不同的用户身份再次测试：

public void testAsEmployee() throws Exception {
    Properties p = new Properties();
    p.put(Context.INITIAL_CONTEXT_FACTORY, "org.apache.openejb.client.LocalInitialContextFactory");
    p.put(Context.SECURITY_PRINCIPAL, "joe");
    p.put(Context.SECURITY_CREDENTIALS, "cool");

    InitialContext context = new InitialContext(p);

    try {
        movies.addMovie(new Movie("Quentin Tarantino", "Reservoir Dogs", 1992));
        movies.addMovie(new Movie("Joel Coen", "Fargo", 1996));
        movies.addMovie(new Movie("Joel Coen", "The Big Lebowski", 1998));

        List<Movie> list = movies.getMovies();
        assertEquals("List.size()", 3, list.size());

        for (Movie movie : list) {
            try {
                movies.deleteMovie(movie);
                fail("Employees should not be allowed to delete");
            } catch (EJBAccessException e) {
                // Good, Employees cannot delete things
            }
        }

        // The list should still be three movies long
        assertEquals("Movies.getMovies()", 3, movies.getMovies().size());
    } finally {
        context.close();
    }
}

测试用户和组

您可以通过添加 <测试用例的类路径中的 code>users.properties 和 groups.properties 文件。在 Maven 中，它位于以下位置：

• src/test/resources/users.properties
• src/test/resources/groups.properties

users.properties 文件可能如下所示

joe=cool
jane=waterfall

和 groups .properties 像这样

Manager=jane
Employee=jane,joe

Answer 2

请参阅 OpenEJB 测试安全示例。基本上，您间接通过一个没有@RolesAllowed 的测试bean，但在调用第二个bean 之前使用@RunAs 来切换角色。