package org.atlasapi.system.bootstrap.workers;

import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import org.atlasapi.entity.Id;
import org.atlasapi.entity.ResourceRef;
import org.atlasapi.entity.util.Resolved;
import org.atlasapi.messaging.ResourceUpdatedMessage;
import org.atlasapi.organisation.Organisation;
import org.atlasapi.organisation.OrganisationResolver;
import org.atlasapi.organisation.OrganisationWriter;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.runners.MockitoJUnitRunner;

import com.codahale.metrics.MetricRegistry;
import com.codahale.metrics.Timer;
import com.google.common.collect.ImmutableList;
import com.google.common.util.concurrent.Futures;

@RunWith(MockitoJUnitRunner.class)
public class OrganisationBootstrapWorkerTest {
    private @Mock OrganisationResolver resolver;
    private @Mock OrganisationWriter writer;
    private @Mock MetricRegistry metric;
    private @Mock ResourceRef updatedResource;
    private @Mock ResourceUpdatedMessage message;
    private @Mock Organisation organisation;

    private OrganisationBootstrapWorker worker;

    @Before
    public void setUp() {
        when(metric.timer("OrganisationBootstrapWorker")).thenReturn(new Timer());
        worker = new OrganisationBootstrapWorker(resolver,writer,metric.timer("OrganisationBootstrapWorker"));

        Id id = Id.valueOf(0L);
        when(message.getUpdatedResource()).thenReturn(updatedResource);
        when(updatedResource.getId()).thenReturn(id);
        when(resolver.resolveIds(ImmutableList.of(id)))
                .thenReturn(Futures.immediateFuture(Resolved.valueOf(ImmutableList.of(organisation))));
    }

    @Test
    public void testProcess() throws Exception {
        worker.process(message);
        verify(writer).write(organisation);
    }

}
package org.atlasapi.system.bootstrap.workers;

import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import org.atlasapi.entity.Id;
import org.atlasapi.entity.ResourceRef;
import org.atlasapi.entity.util.Resolved;
import org.atlasapi.messaging.ResourceUpdatedMessage;
import org.atlasapi.organisation.Organisation;
import org.atlasapi.organisation.OrganisationResolver;
import org.atlasapi.organisation.OrganisationWriter;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.runners.MockitoJUnitRunner;

import com.codahale.metrics.MetricRegistry;
import com.codahale.metrics.Timer;
import com.google.common.collect.ImmutableList;
import com.google.common.util.concurrent.Futures;

@RunWith(MockitoJUnitRunner.class)
public class OrganisationBootstrapWorkerTest {
    private @Mock OrganisationResolver resolver;
    private @Mock OrganisationWriter writer;
    private @Mock MetricRegistry metric;
    private @Mock ResourceRef updatedResource;
    private @Mock ResourceUpdatedMessage message;
    private @Mock Organisation organisation;

    private OrganisationBootstrapWorker worker;

    @Before
    public void setUp() {
        when(metric.timer("OrganisationBootstrapWorker")).thenReturn(new Timer());
        worker = new OrganisationBootstrapWorker(resolver,writer,metric);

        Id id = Id.valueOf(0L);
        when(message.getUpdatedResource()).thenReturn(updatedResource);
        when(updatedResource.getId()).thenReturn(id);
        when(resolver.resolveIds(ImmutableList.of(id)))
                .thenReturn(Futures.immediateFuture(Resolved.valueOf(ImmutableList.of(organisation))));
    }

    @Test
    public void testProcess() throws Exception {
        worker.process(message);
        verify(writer).write(organisation);
    }

}
Revert "fix test"