AbstractLocalAsyncWorkerTestCase.java

package org.skywalking.apm.collector.actor;

import org.junit.Test;
import org.mockito.ArgumentMatcher;
import org.skywalking.apm.collector.queue.EndOfBatchCommand;
import org.skywalking.apm.collector.queue.MessageHolder;

import static org.mockito.Mockito.*;

/**
 * @author pengys5
 */
public class AbstractLocalAsyncWorkerTestCase {

    @Test
    public void testAllocateJob() throws Exception {
        AbstractLocalAsyncWorker worker = mock(AbstractLocalAsyncWorker.class);

        String message = "Test";
        worker.allocateJob(message);
        verify(worker).onWork(message);
    }

    @Test
    public void testOnEventWhenNotEnd() throws Exception {
        AbstractLocalAsyncWorker worker = mock(AbstractLocalAsyncWorker.class);

        AbstractLocalAsyncWorker.WorkerWithDisruptor disruptor = new AbstractLocalAsyncWorker.WorkerWithDisruptor(null, worker);

        MessageHolder holder = new MessageHolder();
        String message = "Test";
        holder.setMessage(message);
        disruptor.onEvent(holder, 0, false);

        verify(worker).onWork(message);
    }

    @Test
    public void testOnEventWhenEnd() throws Exception {
        AbstractLocalAsyncWorker worker = mock(AbstractLocalAsyncWorker.class);

        AbstractLocalAsyncWorker.WorkerWithDisruptor disruptor = new AbstractLocalAsyncWorker.WorkerWithDisruptor(null, worker);

        MessageHolder holder = new MessageHolder();
        String message = "Test";
        holder.setMessage(message);
        disruptor.onEvent(holder, 0, true);

        verify(worker, times(1)).onWork(message);
        verify(worker, times(1)).onWork(argThat(new IsEndOfBatchCommandClass()));
    }

    class IsEndOfBatchCommandClass extends ArgumentMatcher<EndOfBatchCommand> {
        public boolean matches(Object para) {
            return para.getClass() == EndOfBatchCommand.class;
        }
    }

}