Question

我正在创建一个小型java服务，它根据所选位置返回餐馆列表。使用com.basho.riak从Riak检索数据：riak-client：2.0.0并将读取操作包装在TenacityCommand中。

下面介绍了重要的课程，如果您能帮助我创建一个简单的单元测试，我将非常高兴。

使用工厂创建命令：

package service.command.factory;

import com.basho.riak.client.api.RiakClient;
import com.basho.riak.client.api.commands.kv.FetchValue;
import com.basho.riak.client.core.query.Location;
import com.basho.riak.client.core.query.Namespace;

import domain.Place;
import service.command.FetchRestaurantsCommand;

public class FetchRestaurantsCommandFactory {

    private final RiakClient riakClient;
    private final Namespace namespace;

    public FetchRestaurantsCommandFactory(final RiakClient riakClient, final Namespace namespace) {
        this.riakClient = riakClient;
        this.namespace = namespace;
    }

    public FetchRestaurantsCommand create(final Place place) {
        Location location = new Location(namespace, place.getName());
        FetchValue riakCommand = new FetchValue.Builder(location).build();
        return new FetchRestaurantsCommand(riakClient, riakCommand);
    }
}

命令看起来像这样：

package service.command;

import java.util.Optional;

import service.command.keys.WhereToEatDependencyKeys;

import com.basho.riak.client.api.RiakClient;
import com.basho.riak.client.api.commands.kv.FetchValue;
import com.basho.riak.client.api.commands.kv.FetchValue.Response;
import com.yammer.tenacity.core.TenacityCommand;

import domain.Restaurant;
import domain.RestaurantList;

public class FetchRestaurantsCommand extends TenacityCommand<Optional<RestaurantList>>{

    private final RiakClient riakClient;
    private final FetchValue fetchValue; 

    public FetchRestaurantsCommand(RiakClient riakClient, FetchValue fetchValue) {
        super(WhereToEatDependencyKeys.RIAK_GET_RESTAURANTS);
        this.fetchValue = fetchValue;
        this.riakClient = riakClient;
    }

    @Override
    protected Optional<RestaurantList> run() throws Exception {
        Response response = riakClient.execute(fetchValue);
        return Optional.ofNullable(response.getValue(RestaurantList.class));
    }

    @Override
    protected Optional<RestaurantList> getFallback() {
        return Optional.of(RestaurantList.createFallback(new Restaurant("My failure suggestion")));
    }

}

以上类使用如下：

Place place = // Created from url parameter
RiakClient riakClient = // created on start using the app's conf
Namespace namespace = // created on start using the app's conf

FetchRestaurantsCommandFactory factory = new FetchRestaurantsCommandFactory(riakClient, namespace);
FetchRestaurantsCommand command = factory.create(place);
return command.execute();

除了TenacityCommand提供的功能外，我应该如何断言我的系统是否已按照预期获取数据？

我最初的想法是模拟RiakClient以返回预定义的FetchValue.Response，然后对结果RestaurantList进行断言。不幸的是，由于其设计，它无法实例化Mockito.mock FetchValue.Response。

How to mock riak java client?中接受的答案描述了为什么Mockito无法工作。

Answer 1

据我所知，你想要编写单元测试。因此，您希望测试假设某些Response是否正确构造Optional<RestaurantList>实例。我能想到的是将riakClient.execute(fetchValue);包装在受保护（或包私有）帮助函数中，如：

Response fetch() {
  return riakClient.execute(fetchValue);
}

然后在您的测试中，您可以继承FetchRestaurantsCommand并覆盖fetch函数，方法是返回任何Response

现在，您可以编写任何测试，以查看给定Response到Optional<RestaurantList>的转换是否符合预期。

如果您需要完整的代码，而且我的解释不够明确，请告知我们。

Answer 2

我最终按照@gontard的建议使用PowerMock。在GitHub上查看我的单元测试：FetchRestaurantsCommandTest.java

我考虑在com.basho.riak.client包中创建一个假/模拟RiakClient。这样的类可以希望以与真实客户端相同的方式实例化Response对象。它可能适用于fetchValue但是当涉及更高级的Riak概念时，它会变得太大。兄弟姐妹。

如何测试使用Basho的riak-java-client的java代码？

2 个答案: