Lagom - 使用标题测试服务调用

时间:2017-01-19 19:26:26

标签: java lagom

我试图按照文档here关于"处理标题。"我无法弄清楚如何使用标头测试服务调用调用。通常,我会做

service.sayHello().invoke(...)

我注意到ServerServiceCallHeaderServiceCall接受invokeWithHeaders(...)而我的ServiceCall被实现为HeaderServiceCall,但每当我尝试将我的服务API更改为ServerServiceCallHeaderServiceCall,我得到:

 Error in custom provider, java.lang.IllegalArgumentException: Service calls must return ServiceCall, subtypes are not allowed

如何使用自定义请求标头编写调用服务调用的测试?我已经尝试将调用保留为API中的ServiceCall,使用HeaderServiceCall实现,并在测试中转换调用HeaderServiceCall,但在尝试时遇到了强制转换异常去做。任何帮助深表感谢。感谢。

1 个答案:

答案 0 :(得分:1)

我刚刚采用了我的测试代码来检查它:

 public HeaderServiceCall<NewUser, RegUserStatus> addUser(String id) {

    return (reqHeaders, postedUser) -> {
        System.out.println(reqHeaders.getHeader("Referer"));
        PersistentEntityRef<UserCommand> ref = persistentEntityRegistry.refFor(UserEntity.class, id);
        return ref.ask(new UserCommand.RegisterUser(id, postedUser)).thenApply( stat -> Pair.create(ResponseHeader.OK, stat));
    };
}

我的api:

ServiceCall<NewUser, RegUserStatus> addUser(String id);

并测试:

@Test
public void testIt() {
    withServer(defaultSetup(), server -> {
        UsersService service = server.client(UsersService.class);
        RegUserStatus created = service.addUser("aaa").handleRequestHeader(
                rh -> rh.withHeader("Referer" ,"winter")
        ).invoke(new NewUser("aaa")).toCompletableFuture().get(5, SECONDS);
        assertEquals(true, created.ok); // default greeting

    });
}