/*
 * Copyright (c) 2013 Cisco 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.netconf.persist.impl.osgi;

import com.google.common.collect.Sets;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.opendaylight.controller.config.api.ConflictingVersionException;
import org.opendaylight.controller.netconf.api.NetconfDocumentedException;
import org.opendaylight.controller.netconf.mapping.api.Capability;
import org.opendaylight.controller.netconf.mapping.api.HandlingPriority;
import org.opendaylight.controller.netconf.mapping.api.NetconfOperation;
import org.opendaylight.controller.netconf.mapping.api.NetconfOperationChainedExecution;
import org.opendaylight.controller.netconf.mapping.api.NetconfOperationService;
import org.opendaylight.controller.netconf.persist.impl.osgi.MockedBundleContext.DummyAdapterWithInitialSnapshot;
import org.opendaylight.controller.netconf.util.xml.XmlUtil;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.w3c.dom.Document;
import org.xml.sax.SAXException;

import javax.management.MBeanServer;
import java.io.IOException;
import java.lang.management.ManagementFactory;

import static org.mockito.Matchers.any;
import static org.mockito.Matchers.anyString;
import static org.mockito.Mockito.doNothing;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.doThrow;
import static org.mockito.Mockito.mock;

public class ConfigPersisterTest {
    private static final Logger logger = LoggerFactory.getLogger(ConfigPersisterTest.class);

    private MockedBundleContext ctx;
    private ConfigPersisterActivator configPersisterActivator;
    private static final MBeanServer mBeanServer = ManagementFactory.getPlatformMBeanServer();
    private TestingExceptionHandler handler;


    private void setUpContextAndStartPersister(String requiredCapability) throws Exception {
        DummyAdapterWithInitialSnapshot.expectedCapability = requiredCapability;
        ctx = new MockedBundleContext(1000, 1000);
        configPersisterActivator = new ConfigPersisterActivator();
        configPersisterActivator.start(ctx.getBundleContext());
    }

    @Before
    public void setUp() {
        handler = new TestingExceptionHandler();
        Thread.setDefaultUncaughtExceptionHandler(handler);
    }

    @After
    public void tearDown() throws Exception {
        Thread.setDefaultUncaughtExceptionHandler(null);
        configPersisterActivator.stop(ctx.getBundleContext());
    }

    @Test
    public void testPersisterNotAllCapabilitiesProvided() throws Exception {
        setUpContextAndStartPersister("required-cap");
        Thread.sleep(2000);
        handler.assertException(IllegalStateException.class, "Max wait for capabilities reached.Not enough capabilities " +
                "for <data><config-snapshot/></data>. Expected but not found: [required-cap]");

    }

    @Test
    public void testPersisterSuccessfulPush() throws Exception {
        setUpContextAndStartPersister("cap1");
        NetconfOperationService service = getWorkingService(getOKDocument());
        doReturn(service).when(ctx.serviceFactory).createService(anyString());
        Thread.sleep(2000);
        assertCannotRegisterAsJMXListener_pushWasSuccessful();
    }

    // this means pushing of config was successful
    public void assertCannotRegisterAsJMXListener_pushWasSuccessful() {
        handler.assertException(RuntimeException.class, "Cannot register as JMX listener to netconf");
    }

    public NetconfOperationService getWorkingService(Document document) throws SAXException, IOException, NetconfDocumentedException {
        NetconfOperationService service = mock(NetconfOperationService.class);
        Capability capability = mock(Capability.class);
        doReturn(Sets.newHashSet(capability)).when(service).getCapabilities();
        doReturn("cap1").when(capability).getCapabilityUri();


        NetconfOperation mockedOperation = mock(NetconfOperation.class);
        doReturn(Sets.newHashSet(mockedOperation)).when(service).getNetconfOperations();
        doReturn(HandlingPriority.getHandlingPriority(1)).when(mockedOperation).canHandle(any(Document.class));
        doReturn(document).when(mockedOperation).handle(any(Document.class), any(NetconfOperationChainedExecution.class));
        doNothing().when(service).close();
        return service;
    }

    private Document getOKDocument() throws SAXException, IOException {
        return XmlUtil.readXmlToDocument(
                "<rpc-reply message-id=\"1\" xmlns=\"urn:ietf:params:xml:ns:netconf:base:1.0\">\n" +
                        "<ok/>\n" +
                        "</rpc-reply>"
        );
    }


    @Test
    public void testPersisterConflictingVersionException() throws Exception {
        setUpContextAndStartPersister("cap1");

        doReturn(getConflictingService()).when(ctx.serviceFactory).createService(anyString());
        Thread.sleep(2000);
        handler.assertException(IllegalStateException.class, "Max wait for conflicting version stabilization timeout");
    }

    private NetconfOperationService getConflictingService() throws Exception {
        NetconfOperationService service =  getWorkingService(getOKDocument());
        ConflictingVersionException cve = new ConflictingVersionException("");
        try {
            NetconfDocumentedException.wrap(cve);
            throw new AssertionError("Should throw an exception");
        }catch(NetconfDocumentedException e) {
            NetconfOperation mockedOperation = service.getNetconfOperations().iterator().next();
            doThrow(e).when(mockedOperation).handle(any(Document.class), any(NetconfOperationChainedExecution.class));
            return service;
        }
    }

    @Test
    public void testSuccessConflictingVersionException() throws Exception {
        setUpContextAndStartPersister("cap1");
        doReturn(getConflictingService()).when(ctx.serviceFactory).createService(anyString());
        Thread.sleep(500);
        // working service:
        logger.info("Switching to working service **");
        doReturn(getWorkingService(getOKDocument())).when(ctx.serviceFactory).createService(anyString());
        Thread.sleep(1000);
        assertCannotRegisterAsJMXListener_pushWasSuccessful();
    }

}