/*
 * Copyright (c) 2014 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.mdsal.dom.broker;

import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertSame;
import static org.junit.Assert.assertTrue;
import static org.mockito.Mockito.doNothing;
import static org.mockito.Mockito.mock;

import java.util.Map;
import org.junit.Before;
import org.junit.Test;
import org.opendaylight.mdsal.dom.api.DOMMountPoint;
import org.opendaylight.mdsal.dom.api.DOMMountPointListener;
import org.opendaylight.mdsal.dom.api.DOMMountPointService;
import org.opendaylight.mdsal.dom.api.DOMMountPointService.DOMMountPointBuilder;
import org.opendaylight.mdsal.dom.api.DOMService;
import org.opendaylight.mdsal.dom.broker.DOMMountPointServiceImpl.DOMMountPointBuilderImpl;
import org.opendaylight.yangtools.concepts.ObjectRegistration;
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 MountPointServiceTest {

    private static final YangInstanceIdentifier PATH = YangInstanceIdentifier.of(QName.create("namespace", "12-12-2012",
        "top"));

    private DOMMountPointService mountService;

    @Before
    public void setup() {
        mountService = new DOMMountPointServiceImpl();
    }

    @Test
    public void createSimpleMountPoint() throws Exception {
        final DOMMountPointListener listener = mock(DOMMountPointListener.class);
        doNothing().when(listener).onMountPointCreated(PATH);
        mountService.registerProvisionListener(listener);

        assertFalse(mountService.getMountPoint(PATH).isPresent());

        mountService.createMountPoint(PATH).register();

        assertTrue(mountService.getMountPoint(PATH).isPresent());
    }

    @Test
    public void unregisterTest() throws Exception {
        final DOMMountPointListener listener = mock(DOMMountPointListener.class);
        doNothing().when(listener).onMountPointCreated(PATH);
        doNothing().when(listener).onMountPointRemoved(PATH);
        final DOMMountPointServiceImpl mountService = new DOMMountPointServiceImpl();
        mountService.registerProvisionListener(listener);
        mountService.createMountPoint(PATH).register();

        assertTrue(mountService.getMountPoint(PATH).isPresent());

        mountService.unregisterMountPoint(PATH);

        assertFalse(mountService.getMountPoint(PATH).isPresent());
    }

    @Test
    public void mountRegistrationTest() throws Exception {
        final DOMMountPointBuilder mountBuilder = mountService.createMountPoint(PATH);
        final ObjectRegistration<DOMMountPoint> objectRegistration = mountBuilder.register();

        assertTrue(mountService.getMountPoint(PATH).isPresent());
        assertSame(objectRegistration.getInstance(), mountService.getMountPoint(PATH).get());

        objectRegistration.close();

        assertFalse(mountService.getMountPoint(PATH).isPresent());
    }

    @Test
    public void mountBuilderTest() throws Exception {
        final DOMMountPointBuilderImpl mountBuilder = (DOMMountPointBuilderImpl) mountService.createMountPoint(PATH);
        mountBuilder.register();

        final SchemaContext mockSchemaContext = mock(SchemaContext.class);
        mountBuilder.addInitialSchemaContext(mockSchemaContext);

        assertSame(mockSchemaContext, mountBuilder.getSchemaContext());

        final Map<Class<? extends DOMService>, DOMService> services = mountBuilder.getServices();
        assertTrue(services.isEmpty());
        mountBuilder.addService(DOMService.class, mock(DOMService.class));
        assertTrue(services.containsKey(DOMService.class));
    }
}