/*
 * Copyright © 2018 Orange 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.transportpce.renderer.provisiondevice;

import java.util.ArrayList;
import java.util.List;
import org.junit.Assert;
import org.junit.Test;
import org.mockito.Mockito;
import org.opendaylight.controller.md.sal.binding.api.MountPoint;
import org.opendaylight.controller.md.sal.binding.api.MountPointService;
import org.opendaylight.transportpce.common.crossconnect.CrossConnect;
import org.opendaylight.transportpce.common.crossconnect.CrossConnectImpl;
import org.opendaylight.transportpce.common.device.DeviceTransactionManager;
import org.opendaylight.transportpce.common.device.DeviceTransactionManagerImpl;
import org.opendaylight.transportpce.common.mapping.PortMapping;
import org.opendaylight.transportpce.common.mapping.PortMappingImpl;
import org.opendaylight.transportpce.common.openroadminterfaces.OpenRoadmInterfaceException;
import org.opendaylight.transportpce.common.openroadminterfaces.OpenRoadmInterfaces;
import org.opendaylight.transportpce.common.openroadminterfaces.OpenRoadmInterfacesImpl;
import org.opendaylight.transportpce.renderer.openroadminterface.OpenRoadmInterfaceFactory;
import org.opendaylight.transportpce.renderer.stub.MountPointServiceStub;
import org.opendaylight.transportpce.renderer.stub.MountPointStub;
import org.opendaylight.transportpce.renderer.utils.ServiceImplementationDataUtils;
import org.opendaylight.transportpce.test.AbstractTest;
import org.opendaylight.yang.gen.v1.http.org.transportpce.common.types.rev170907.node.interfaces.NodeInterface;
import org.opendaylight.yang.gen.v1.http.org.transportpce.common.types.rev170907.node.interfaces.NodeInterfaceBuilder;
import org.opendaylight.yang.gen.v1.http.org.transportpce.common.types.rev170907.node.interfaces.NodeInterfaceKey;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.renderer.rev170228.RendererRollbackInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.renderer.rev170228.RendererRollbackInputBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.renderer.rev170228.RendererRollbackOutput;

public class DeviceRendererServiceImplRollbackTest extends AbstractTest {

    private DeviceRendererService deviceRendererService;
    private CrossConnect crossConnect;
    private OpenRoadmInterfaces openRoadmInterfaces;

    private void setMountPoint(MountPoint mountPoint) {
        MountPointService mountPointService = new MountPointServiceStub(mountPoint);
        DeviceTransactionManager deviceTransactionManager = new DeviceTransactionManagerImpl(mountPointService, 3000);
        this.openRoadmInterfaces = new OpenRoadmInterfacesImpl(deviceTransactionManager);
        this.openRoadmInterfaces = Mockito.spy(this.openRoadmInterfaces);
        PortMapping portMapping = new PortMappingImpl(this.getDataBroker(), deviceTransactionManager,
                this.openRoadmInterfaces);
        OpenRoadmInterfaceFactory openRoadmInterfaceFactory = new OpenRoadmInterfaceFactory(portMapping,
                this.openRoadmInterfaces);
        this.crossConnect = new CrossConnectImpl(deviceTransactionManager);
        this.crossConnect = Mockito.spy(this.crossConnect);
        this.deviceRendererService = new DeviceRendererServiceImpl(this.getDataBroker(),
                deviceTransactionManager, openRoadmInterfaceFactory, this.openRoadmInterfaces, this.crossConnect,
            portMapping);
    }



    @Test
    public void testRollbackEmptyInterface() {
        setMountPoint(new MountPointStub(getDataBroker()));
        RendererRollbackInput rendererRollbackInput = ServiceImplementationDataUtils.buildRendererRollbackInput();
        RendererRollbackOutput rendererRollbackOutput =
                this.deviceRendererService.rendererRollback(rendererRollbackInput);
        Assert.assertTrue(rendererRollbackOutput.isSuccess());
        Assert.assertTrue(rendererRollbackOutput.getFailedToRollback().isEmpty());
    }

    @Test
    public void testRollbackConnectionIdNotExist() {
        setMountPoint(new MountPointStub(getDataBroker()));

        NodeInterfaceBuilder nodeInterfaceBuilder = new NodeInterfaceBuilder();
        nodeInterfaceBuilder.setNodeId("node1");
        nodeInterfaceBuilder.withKey(new NodeInterfaceKey("node1"));
        List<String> connectionID = new ArrayList<>();
        connectionID.add("node1-PP");
        nodeInterfaceBuilder.setConnectionId(connectionID);
        List<NodeInterface> nodeInterfaces = new ArrayList<>();
        nodeInterfaces.add(nodeInterfaceBuilder.build());
        RendererRollbackInputBuilder rendererRollbackInputBuilder = new RendererRollbackInputBuilder();
        rendererRollbackInputBuilder.setNodeInterface(nodeInterfaces);

        RendererRollbackOutput rendererRollbackOutput =
                this.deviceRendererService.rendererRollback(rendererRollbackInputBuilder.build());
        Assert.assertFalse(rendererRollbackOutput.isSuccess());
        Assert.assertEquals(1, rendererRollbackOutput.getFailedToRollback().size());
        Assert.assertEquals("node1", rendererRollbackOutput.getFailedToRollback().get(0).getNodeId());
    }

    @Test
    public void testRollbackConnectionId() {
        setMountPoint(new MountPointStub(getDataBroker()));

        NodeInterfaceBuilder nodeInterfaceBuilder = new NodeInterfaceBuilder();
        nodeInterfaceBuilder.setNodeId("node1");
        nodeInterfaceBuilder.withKey(new NodeInterfaceKey("node1"));
        List<String> connectionID = new ArrayList<>();
        connectionID.add("src-PP-dest-PP-20");
        nodeInterfaceBuilder.setConnectionId(connectionID);
        List<NodeInterface> nodeInterfaces = new ArrayList<>();
        nodeInterfaces.add(nodeInterfaceBuilder.build());
        RendererRollbackInputBuilder rendererRollbackInputBuilder = new RendererRollbackInputBuilder();
        rendererRollbackInputBuilder.setNodeInterface(nodeInterfaces);

        Mockito.doReturn(true).when(this.crossConnect).deleteCrossConnect("node1", connectionID.get(0));
        RendererRollbackOutput rendererRollbackOutput =
                this.deviceRendererService.rendererRollback(rendererRollbackInputBuilder.build());
        Assert.assertTrue("Rollback must success when cross connect returns true", rendererRollbackOutput.isSuccess());
        Assert.assertEquals(1, rendererRollbackOutput.getFailedToRollback().size());
        Assert.assertTrue("There must not be any failed interfaces when cross connect returns true",
            rendererRollbackOutput.getFailedToRollback().get(0).getInterface().isEmpty());

        Mockito.doReturn(false).when(this.crossConnect).deleteCrossConnect("node1", connectionID.get(0));
        rendererRollbackOutput =
                this.deviceRendererService.rendererRollback(rendererRollbackInputBuilder.build());
        Assert.assertFalse("Rollback must fail when cross connect returns false",rendererRollbackOutput.isSuccess());
        Assert.assertEquals(1, rendererRollbackOutput.getFailedToRollback().size());
        Assert.assertFalse(rendererRollbackOutput.getFailedToRollback().get(0).getInterface().isEmpty());
        Assert.assertEquals("node1", rendererRollbackOutput.getFailedToRollback().get(0).getNodeId());

        Mockito.verify(this.crossConnect, Mockito.times(2)).deleteCrossConnect("node1", connectionID.get(0));
    }

    @Test
    public void testRollbackInterfaces() throws OpenRoadmInterfaceException {
        setMountPoint(new MountPointStub(getDataBroker()));

        NodeInterfaceBuilder nodeInterfaceBuilder = new NodeInterfaceBuilder();
        nodeInterfaceBuilder.setNodeId("node1");
        nodeInterfaceBuilder.withKey(new NodeInterfaceKey("node1"));
        nodeInterfaceBuilder.setConnectionId(new ArrayList<>());
        List<String> oduInterfacesId = new ArrayList<>();
        oduInterfacesId.add("node1-" + OpenRoadmInterfacesImpl.NETWORK_TOKEN + "-ODU");
        nodeInterfaceBuilder.setOduInterfaceId(oduInterfacesId);
        List<String> otuInterfacesId = new ArrayList<>();
        otuInterfacesId.add("node1-" + OpenRoadmInterfacesImpl.NETWORK_TOKEN + "-OTU");
        nodeInterfaceBuilder.setOtuInterfaceId(otuInterfacesId);
        List<String> ochInterfacesId = new ArrayList<>();
        ochInterfacesId.add("node1-" + OpenRoadmInterfacesImpl.NETWORK_TOKEN + "-20");
        nodeInterfaceBuilder.setOchInterfaceId(ochInterfacesId);
        List<NodeInterface> nodeInterfaces = new ArrayList<>();
        nodeInterfaces.add(nodeInterfaceBuilder.build());
        RendererRollbackInputBuilder rendererRollbackInputBuilder = new RendererRollbackInputBuilder();
        rendererRollbackInputBuilder.setNodeInterface(nodeInterfaces);

        Mockito.doNothing().when(this.openRoadmInterfaces).deleteInterface(Mockito.eq("node1"), Mockito.anyString());
        RendererRollbackOutput rendererRollbackOutput =
                this.deviceRendererService.rendererRollback(rendererRollbackInputBuilder.build());
        Assert.assertTrue(rendererRollbackOutput.isSuccess());
        Assert.assertFalse(rendererRollbackOutput.getFailedToRollback().isEmpty());
        Assert.assertTrue(rendererRollbackOutput.getFailedToRollback().get(0).getInterface().isEmpty());
        Mockito.verify(this.crossConnect, Mockito.times(0)).deleteCrossConnect(Mockito.anyString(),
            Mockito.anyString());
        Mockito.verify(this.openRoadmInterfaces, Mockito.times(3)).deleteInterface(Mockito.eq("node1"),
            Mockito.anyString());
    }

    @Test
    public void testRollbackInterfacesException() throws OpenRoadmInterfaceException {
        setMountPoint(new MountPointStub(getDataBroker()));

        NodeInterfaceBuilder nodeInterfaceBuilder = new NodeInterfaceBuilder();
        nodeInterfaceBuilder.setNodeId("node1");
        nodeInterfaceBuilder.withKey(new NodeInterfaceKey("node1"));
        nodeInterfaceBuilder.setConnectionId(new ArrayList<>());
        List<String> ethInterfacesId = new ArrayList<>();
        ethInterfacesId.add("node1-" + OpenRoadmInterfacesImpl.CLIENT_TOKEN + "-ETHERNET");
        nodeInterfaceBuilder.setEthInterfaceId(ethInterfacesId);
        List<NodeInterface> nodeInterfaces = new ArrayList<>();
        nodeInterfaces.add(nodeInterfaceBuilder.build());
        RendererRollbackInputBuilder rendererRollbackInputBuilder = new RendererRollbackInputBuilder();
        rendererRollbackInputBuilder.setNodeInterface(nodeInterfaces);

        Mockito.doThrow(OpenRoadmInterfaceException.class).when(this.openRoadmInterfaces)
            .deleteInterface(Mockito.eq("node1"), Mockito.anyString());
        RendererRollbackOutput rendererRollbackOutput =
                this.deviceRendererService.rendererRollback(rendererRollbackInputBuilder.build());
        Assert.assertFalse(rendererRollbackOutput.isSuccess());
        Assert.assertEquals(1,rendererRollbackOutput.getFailedToRollback().size());
        Assert.assertEquals(1, rendererRollbackOutput.getFailedToRollback().get(0).getInterface().size());
        Assert.assertEquals(ethInterfacesId.get(0), rendererRollbackOutput.getFailedToRollback().get(0)
            .getInterface().get(0));
        Mockito.verify(this.crossConnect, Mockito.times(0))
            .deleteCrossConnect(Mockito.anyString(), Mockito.anyString());
        Mockito.verify(this.openRoadmInterfaces, Mockito.times(1))
            .deleteInterface("node1",ethInterfacesId.get(0));
    }
}