/*
 * Copyright (c) 2014 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.sal.rest.doc.impl;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import java.util.Arrays;
import java.util.List;
import java.util.Optional;
import java.util.Set;
import java.util.TreeSet;
import javax.ws.rs.core.UriInfo;
import org.junit.Before;
import org.junit.Test;
import org.opendaylight.mdsal.dom.api.DOMMountPoint;
import org.opendaylight.mdsal.dom.api.DOMMountPointService;
import org.opendaylight.mdsal.dom.api.DOMSchemaService;
import org.opendaylight.netconf.sal.rest.doc.impl.MountPointSwaggerGeneratorDraft02;
import org.opendaylight.netconf.sal.rest.doc.mountpoints.MountPointSwagger;
import org.opendaylight.netconf.sal.rest.doc.swagger.Api;
import org.opendaylight.netconf.sal.rest.doc.swagger.ApiDeclaration;
import org.opendaylight.netconf.sal.rest.doc.swagger.Operation;
import org.opendaylight.netconf.sal.rest.doc.swagger.Resource;
import org.opendaylight.netconf.sal.rest.doc.swagger.ResourceList;
import org.opendaylight.yangtools.yang.common.QName;
import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
import org.opendaylight.yangtools.yang.model.api.SchemaContext;

public class MountPointSwaggerTest {

    private static final String HTTP_URL = "http://localhost/path";
    private static final YangInstanceIdentifier INSTANCE_ID = YangInstanceIdentifier.builder()
            .node(QName.create("", "nodes"))
            .node(QName.create("", "node"))
            .nodeWithKey(QName.create("", "node"), QName.create("", "id"), "123").build();
    private static final String INSTANCE_URL = "/nodes/node/123/";
    private MountPointSwagger swagger;
    private DocGenTestHelper helper;
    private SchemaContext schemaContext;

    @SuppressWarnings("resource")
    @Before
    public void setUp() throws Exception {
        this.helper = new DocGenTestHelper();
        this.helper.setUp();
        this.schemaContext = this.helper.getSchemaContext();

        // We are sharing the global schema service and the mount schema service
        // in our test.
        // OK for testing - real thing would have seperate instances.
        final SchemaContext context = this.helper.createMockSchemaContext();
        final DOMSchemaService schemaService = this.helper.createMockSchemaService(context);

        final DOMMountPoint mountPoint = mock(DOMMountPoint.class);
        when(mountPoint.getSchemaContext()).thenReturn(context);

        final DOMMountPointService service = mock(DOMMountPointService.class);
        when(service.getMountPoint(INSTANCE_ID)).thenReturn(Optional.of(mountPoint));

        MountPointSwaggerGeneratorDraft02 generator = new MountPointSwaggerGeneratorDraft02(schemaService, service);
        this.swagger = generator.getMountPointSwagger();
    }

    @Test()
    public void testGetResourceListBadIid() throws Exception {
        final UriInfo mockInfo = this.helper.createMockUriInfo(HTTP_URL);

        assertEquals(null, this.swagger.getResourceList(mockInfo, 1L));
    }

    @Test()
    public void getInstanceIdentifiers() throws Exception {
        assertEquals(0, this.swagger.getInstanceIdentifiers().size());
        this.swagger.onMountPointCreated(INSTANCE_ID); // add this ID into the list of
                                                 // mount points
        assertEquals(1, this.swagger.getInstanceIdentifiers().size());
        assertEquals((Long) 1L, this.swagger.getInstanceIdentifiers().entrySet().iterator().next()
                .getValue());
        assertEquals(INSTANCE_URL, this.swagger.getInstanceIdentifiers().entrySet().iterator().next()
                .getKey());
        this.swagger.onMountPointRemoved(INSTANCE_ID); // remove ID from list of mount
                                                 // points
        assertEquals(0, this.swagger.getInstanceIdentifiers().size());
    }

    @Test
    public void testGetResourceListGoodId() throws Exception {
        final UriInfo mockInfo = this.helper.createMockUriInfo(HTTP_URL);
        this.swagger.onMountPointCreated(INSTANCE_ID); // add this ID into the list of
                                                 // mount points
        final ResourceList resourceList = this.swagger.getResourceList(mockInfo, 1L);

        Resource dataStoreResource = null;
        for (final Resource r : resourceList.getApis()) {
            if (r.getPath().endsWith("/Datastores(-)")) {
                dataStoreResource = r;
            }
        }
        assertNotNull("Failed to find data store resource", dataStoreResource);
    }

    @Test
    public void testGetDataStoreApi() throws Exception {
        final UriInfo mockInfo = this.helper.createMockUriInfo(HTTP_URL);
        this.swagger.onMountPointCreated(INSTANCE_ID); // add this ID into the list of
                                                 // mount points

        final ApiDeclaration mountPointApi = this.swagger.getMountPointApi(mockInfo, 1L, "Datastores", "-");
        assertNotNull("failed to find Datastore API", mountPointApi);
        final List<Api> apis = mountPointApi.getApis();
        assertEquals("Unexpected api list size", 3, apis.size());

        final Set<String> actualApis = new TreeSet<>();
        for (final Api api : apis) {
            actualApis.add(api.getPath());
            final List<Operation> operations = api.getOperations();
            assertEquals("unexpected operation size on " + api.getPath(), 1, operations.size());
            assertEquals("unexpected operation method " + api.getPath(), "GET", operations.get(0)
                    .getMethod());
            assertNotNull("expected non-null desc on " + api.getPath(), operations.get(0)
                    .getNotes());
        }
        final Set<String> expectedApis = new TreeSet<>(Arrays.asList(new String[] {
            "/restconf/config" + INSTANCE_URL + "yang-ext:mount",
            "/restconf/operational" + INSTANCE_URL + "yang-ext:mount",
            "/restconf/operations" + INSTANCE_URL + "yang-ext:mount",}));
        assertEquals(expectedApis, actualApis);
    }
}