MockServiceInventory.java

package io.airlift.airship.coordinator;

import com.google.common.base.Preconditions;
import com.google.common.collect.ImmutableList;
import io.airlift.airship.shared.SlotStatus;
import io.airlift.discovery.client.ServiceDescriptor;

import java.util.List;

public class MockServiceInventory
        implements ServiceInventory
{
    private ImmutableList<ServiceDescriptor> serviceInventory = ImmutableList.of();

    public MockServiceInventory()
    {
        serviceInventory = ImmutableList.of();
    }

    @Override
    public List<ServiceDescriptor> getServiceInventory(Iterable<SlotStatus> allSlotStatus)
    {
        return serviceInventory;
    }

    public void setServiceInventory(Iterable<ServiceDescriptor> serviceInventory)
    {
        Preconditions.checkNotNull(serviceInventory, "serviceInventory is null");
        this.serviceInventory = ImmutableList.copyOf(serviceInventory);
    }
}