/*
 * Copyright (c) 2015 Brocade Communications Systems, Inc. and others.  All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
 * and is available at http://www.eclipse.org/legal/epl-v10.html
 */
package org.opendaylight.controller.cluster.raft;

import static org.junit.Assert.assertEquals;
import static org.mockito.Matchers.eq;
import static org.mockito.Matchers.same;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.reset;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.verifyNoMoreInteractions;
import akka.japi.Procedure;
import com.google.common.base.Supplier;
import java.util.Collections;
import org.hamcrest.BaseMatcher;
import org.hamcrest.Description;
import org.hamcrest.Matcher;
import org.junit.Before;
import org.junit.Test;
import org.mockito.ArgumentCaptor;
import org.mockito.Matchers;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.mockito.MockitoAnnotations;
import org.mockito.internal.matchers.Same;
import org.opendaylight.controller.cluster.DataPersistenceProvider;
import org.opendaylight.controller.cluster.raft.MockRaftActorContext.MockPayload;
import org.opendaylight.controller.cluster.raft.MockRaftActorContext.MockReplicatedLogEntry;
import org.opendaylight.controller.cluster.raft.RaftActor.DeleteEntries;
import org.opendaylight.controller.cluster.raft.behaviors.RaftActorBehavior;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Unit tests for ReplicatedLogImpl.
 *
 * @author Thomas Pantelis
 */
public class ReplicatedLogImplTest {
    private static final Logger LOG = LoggerFactory.getLogger(RaftActorRecoverySupportTest.class);

    @Mock
    private DataPersistenceProvider mockPersistence;

    @Mock
    private RaftActorBehavior mockBehavior;

    @Mock
    private SnapshotManager mockSnapshotManager;

    private RaftActorContext context;
    private final DefaultConfigParamsImpl configParams = new DefaultConfigParamsImpl();

    @Before
    public void setup() {
        MockitoAnnotations.initMocks(this);

        context = new RaftActorContextImpl(null, null, "test",
                new ElectionTermImpl(mockPersistence, "test", LOG),
                -1, -1, Collections.<String,String>emptyMap(), configParams, LOG)  {
            @Override
            public SnapshotManager getSnapshotManager() {
                return mockSnapshotManager;
            }
        };
    }

    private void verifyPersist(Object message) throws Exception {
        verifyPersist(message, new Same(message));
    }

    @SuppressWarnings({ "unchecked", "rawtypes" })
    private void verifyPersist(Object message, Matcher<?> matcher) throws Exception {
        ArgumentCaptor<Procedure> procedure = ArgumentCaptor.forClass(Procedure.class);
        verify(mockPersistence).persist(Matchers.argThat(matcher), procedure.capture());

        procedure.getValue().apply(message);
    }

    @SuppressWarnings("unchecked")
    @Test
    public void testAppendAndPersistExpectingNoCapture() throws Exception {
        ReplicatedLog log = ReplicatedLogImpl.newInstance(context, mockPersistence, mockBehavior);

        MockReplicatedLogEntry logEntry = new MockReplicatedLogEntry(1, 1, new MockPayload("1"));

        log.appendAndPersist(logEntry);

        verifyPersist(logEntry);

        assertEquals("size", 1, log.size());

        reset(mockPersistence);

        Procedure<ReplicatedLogEntry> mockCallback = Mockito.mock(Procedure.class);
        log.appendAndPersist(logEntry, mockCallback);

        verifyPersist(logEntry);

        verify(mockCallback).apply(same(logEntry));
        verifyNoMoreInteractions(mockSnapshotManager);

        assertEquals("size", 2, log.size());
    }

    @Test
    public void testAppendAndPersistExpectingCaptureDueToJournalCount() throws Exception {
        configParams.setSnapshotBatchCount(2);

        doReturn(1L).when(mockBehavior).getReplicatedToAllIndex();

        ReplicatedLog log = ReplicatedLogImpl.newInstance(context, mockPersistence, mockBehavior);

        MockReplicatedLogEntry logEntry1 = new MockReplicatedLogEntry(1, 2, new MockPayload("2"));
        MockReplicatedLogEntry logEntry2 = new MockReplicatedLogEntry(1, 3, new MockPayload("3"));

        log.appendAndPersist(logEntry1);
        verifyPersist(logEntry1);

        verifyNoMoreInteractions(mockSnapshotManager);
        reset(mockPersistence);

        log.appendAndPersist(logEntry2);
        verifyPersist(logEntry2);

        verify(mockSnapshotManager).capture(same(logEntry2), eq(1L));

        assertEquals("size", 2, log.size());
    }

    @Test
    public void testAppendAndPersistExpectingCaptureDueToDataSize() throws Exception {
        doReturn(1L).when(mockBehavior).getReplicatedToAllIndex();

        context.setTotalMemoryRetriever(new Supplier<Long>() {
            @Override
            public Long get() {
                return 100L;
            }
        });

        ReplicatedLog log = ReplicatedLogImpl.newInstance(context, mockPersistence, mockBehavior);

        int dataSize = 600;
        MockReplicatedLogEntry logEntry = new MockReplicatedLogEntry(1, 2, new MockPayload("2", dataSize));

        doReturn(true).when(mockSnapshotManager).capture(same(logEntry), eq(1L));

        log.appendAndPersist(logEntry);
        verifyPersist(logEntry);

        verify(mockSnapshotManager).capture(same(logEntry), eq(1L));

        reset(mockPersistence, mockSnapshotManager);

        logEntry = new MockReplicatedLogEntry(1, 3, new MockPayload("3", 5));

        log.appendAndPersist(logEntry);
        verifyPersist(logEntry);

        verifyNoMoreInteractions(mockSnapshotManager);

        assertEquals("size", 2, log.size());
    }

    @Test
    public void testRemoveFromAndPersist() throws Exception {

        ReplicatedLog log = ReplicatedLogImpl.newInstance(context, mockPersistence, mockBehavior);

        log.append(new MockReplicatedLogEntry(1, 0, new MockPayload("0")));
        log.append(new MockReplicatedLogEntry(1, 1, new MockPayload("1")));
        log.append(new MockReplicatedLogEntry(1, 2, new MockPayload("2")));

        log.removeFromAndPersist(1);

        DeleteEntries deleteEntries = new DeleteEntries(1);
        verifyPersist(deleteEntries, match(deleteEntries));

        assertEquals("size", 1, log.size());

        reset(mockPersistence);

        log.removeFromAndPersist(1);

        verifyNoMoreInteractions(mockPersistence);
    }

    public Matcher<DeleteEntries> match(final DeleteEntries actual){
        return new BaseMatcher<DeleteEntries>() {
            @Override
            public boolean matches(Object o) {
                DeleteEntries other = (DeleteEntries) o;
                return actual.getFromIndex() == other.getFromIndex();
            }

            @Override
            public void describeTo(Description description) {
                description.appendText("DeleteEntries: fromIndex: " + actual.getFromIndex());
            }
        };
    }
}