/*
 * JBoss, Home of Professional Open Source.
 * Copyright 2011, Red Hat, Inc., and individual contributors
 * as indicated by the @author tags. See the copyright.txt file in the
 * distribution for a full listing of individual contributors.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * This software is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */

package org.jboss.as.naming;

import java.util.Hashtable;
import java.util.List;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
import javax.naming.Binding;
import javax.naming.CompositeName;
import javax.naming.Context;
import javax.naming.Name;
import javax.naming.NameClassPair;
import javax.naming.NameParser;
import javax.naming.NamingEnumeration;
import javax.naming.NamingException;

import org.jboss.msc.service.Service;
import org.jboss.msc.service.ServiceContainer;
import org.jboss.msc.service.ServiceName;
import org.jboss.msc.service.StartContext;
import org.jboss.msc.service.StartException;
import org.jboss.msc.service.StopContext;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;

/**
 * @author John Bailey
 */
public class ServiceBasedNamingStoreTestCase {

    private ServiceContainer container;
    private ServiceBasedNamingStore store;

    @Before
    public void setupServiceContainer() {
        container = ServiceContainer.Factory.create();
        store = new ServiceBasedNamingStore(container, ServiceName.JBOSS);
    }

    @After
    public void shutdownServiceContainer() {
        if (container != null) {
            container.shutdown();
            try {
                container.awaitTermination(5, TimeUnit.SECONDS);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            finally {
                container = null;
            }
        }
        store = null;
    }

    @Test
    public void testLookupBase() throws Exception {
        final Object obj = store.lookup(new CompositeName());
        assertNotNull(obj);
    }

    @Test
    public void testLookupBinding() throws Exception {
        final ServiceName bindingName = ServiceName.JBOSS.append("foo", "bar");
        final Object value = new Object();
        bindObject(bindingName, value);

        final Object obj = store.lookup(new CompositeName("foo/bar"));
        assertNotNull(obj);
        assertEquals(value, obj);
    }

    @Test
    public void testLookupParentContext() throws Exception {
        final ServiceName bindingName = ServiceName.JBOSS.append("foo", "bar");
        store.add(bindingName);
        final Object obj = store.lookup(new CompositeName("foo"));
        assertNotNull(obj);
        assertTrue(obj instanceof Context);
    }

    @Test
    public void testStoredContext() throws Exception {
        final ServiceName bindingName = ServiceName.JBOSS.append("foo-stored").append("again");
        bindObject(bindingName, new Context() {
            @Override
            public Object lookup(Name name) throws NamingException {
                if ("blah/blah2".equals(name.toString())) {
                    return 5;
                }

                return null;
            }

            @Override
            public Object lookup(String name) throws NamingException {
                return lookup(new CompositeName(name));
            }

            @Override
            public void bind(Name name, Object obj) throws NamingException {
            }

            @Override
            public void bind(String name, Object obj) throws NamingException {
            }

            @Override
            public void rebind(Name name, Object obj) throws NamingException {
            }

            @Override
            public void rebind(String name, Object obj) throws NamingException {
            }

            @Override
            public void unbind(Name name) throws NamingException {
            }

            @Override
            public void unbind(String name) throws NamingException {
            }

            @Override
            public void rename(Name oldName, Name newName) throws NamingException {
            }

            @Override
            public void rename(String oldName, String newName) throws NamingException {
            }

            @Override
            public NamingEnumeration<NameClassPair> list(Name name) throws NamingException {
                return null;
            }

            @Override
            public NamingEnumeration<NameClassPair> list(String name) throws NamingException {
                return null;
            }

            @Override
            public NamingEnumeration<Binding> listBindings(Name name) throws NamingException {
                if (!"hi/there".equals(name.toString()))
                    throw new IllegalArgumentException("Expected hi/there");
                return null;
            }

            @Override
            public NamingEnumeration<Binding> listBindings(String name) throws NamingException {
                return null;
            }

            @Override
            public void destroySubcontext(Name name) throws NamingException {
            }

            @Override
            public void destroySubcontext(String name) throws NamingException {
            }

            @Override
            public Context createSubcontext(Name name) throws NamingException {
                return null;
            }

            @Override
            public Context createSubcontext(String name) throws NamingException {
                return null;
            }

            @Override
            public Object lookupLink(Name name) throws NamingException {
                return null;
            }

            @Override
            public Object lookupLink(String name) throws NamingException {
                return null;
            }

            @Override
            public NameParser getNameParser(Name name) throws NamingException {
                return null;
            }

            @Override
            public NameParser getNameParser(String name) throws NamingException {
                return null;
            }

            @Override
            public Name composeName(Name name, Name prefix) throws NamingException {
                return null;
            }

            @Override
            public String composeName(String name, String prefix) throws NamingException {
                return null;
            }

            @Override
            public Object addToEnvironment(String propName, Object propVal) throws NamingException {
                return null;
            }

            @Override
            public Object removeFromEnvironment(String propName) throws NamingException {
                return null;
            }

            @Override
            public Hashtable<?, ?> getEnvironment() throws NamingException {
                return null;
            }

            @Override
            public void close() throws NamingException {
            }

            @Override
            public String getNameInNamespace() throws NamingException {
                return null;
            }
        });
        final NamingContext ctx = new NamingContext(new CompositeName(), store, null);
        final Object obj = ctx.lookup(new CompositeName("foo-stored/again/blah/blah2"));
        ctx.listBindings("foo-stored/again/hi/there");
        assertNotNull(obj);
        assertEquals(5, obj);
    }

    @Test
    public void testLookupNestedContext() throws Exception {
        final ServiceName bindingName = ServiceName.JBOSS.append("foo", "bar", "baz", "TestBean");
        store.add(bindingName);
        store.add(ServiceName.JBOSS.append("foos", "bar"));
        store.add(ServiceName.JBOSS.append("fo", "bar"));
        store.add(ServiceName.JBOSS.append("foo", "ba"));
        store.add(ServiceName.JBOSS.append("foo", "bart"));
        store.add(ServiceName.JBOSS.append("foo", "bar", "ba"));
        store.add(ServiceName.JBOSS.append("foo", "bar", "bazt"));
        store.add(ServiceName.JBOSS.append("foo", "bar", "art"));

        Object obj = store.lookup(new CompositeName("foo"));
        assertNotNull(obj);
        assertTrue(obj instanceof Context);

        obj = Context.class.cast(obj).lookup(new CompositeName("bar"));
        assertNotNull(obj);
        assertTrue(obj instanceof Context);

        obj = Context.class.cast(obj).lookup(new CompositeName("baz"));
        assertNotNull(obj);
        assertTrue(obj instanceof Context);
    }

    @Test
    public void testLookupBindingUsingNestedContext() throws Exception {
        final ServiceName bindingName = ServiceName.JBOSS.append("foo", "bar", "baz", "TestBean");
        final Object value = new Object();
        bindObject(bindingName, value);

        Object context = store.lookup(new CompositeName("foo"));
        assertNotNull(context);
        assertTrue(context instanceof Context);

        Object obj = Context.class.cast(context).lookup(new CompositeName("bar/baz/TestBean"));
        assertNotNull(obj);
        assertEquals(value, obj);

        context = Context.class.cast(context).lookup(new CompositeName("bar"));
        obj = Context.class.cast(context).lookup(new CompositeName("baz/TestBean"));
        assertNotNull(obj);
        assertEquals(value, obj);


        context = Context.class.cast(context).lookup(new CompositeName("baz"));
        obj = Context.class.cast(context).lookup(new CompositeName("TestBean"));
        assertNotNull(obj);
        assertEquals(value, obj);
    }


    @Test
    public void testList() throws Exception {
        final Object value = new Object();
        bindObject(ServiceName.JBOSS.append("TestBean"), value);
        bindObject(ServiceName.JBOSS.append("foo", "TestBean"), value);
        bindObject(ServiceName.JBOSS.append("foo", "bar", "TestBean"), value);
        bindObject(ServiceName.JBOSS.append("foo", "bar", "baz", "TestBean"), value);

        store.add(ServiceName.JBOSS.append("foos", "bar"));
        store.add(ServiceName.JBOSS.append("fo", "bar"));
        store.add(ServiceName.JBOSS.append("foo", "ba", "baz"));
        store.add(ServiceName.JBOSS.append("foo", "bart", "baz"));
        store.add(ServiceName.JBOSS.append("foo", "bar", "ba"));
        store.add(ServiceName.JBOSS.append("foo", "bar", "bazt"));
        store.add(ServiceName.JBOSS.append("foo", "bar", "art"));
        store.add(ServiceName.JBOSS.append("other", "one"));

        List<NameClassPair> list = store.list(new CompositeName(""));
        assertEquals(5, list.size());
        assertContains(list, "TestBean", Object.class);
        assertContains(list, "foo", Context.class);
        assertContains(list, "fo", Context.class);
        assertContains(list, "foos", Context.class);
        assertContains(list, "other", Context.class);


        list = store.list(new CompositeName("foo"));
        assertEquals(4, list.size());
        assertContains(list, "TestBean", Object.class);
        assertContains(list, "ba", Context.class);
        assertContains(list, "bart", Context.class);
        assertContains(list, "bar", Context.class);
    }

    @Test
    public void testListBindings() throws Exception {
        final Object value = new Object();
        bindObject(ServiceName.JBOSS.append("TestBean"), value);
        bindObject(ServiceName.JBOSS.append("foo", "TestBean"), value);
        bindObject(ServiceName.JBOSS.append("foo", "bar", "TestBean"), value);
        bindObject(ServiceName.JBOSS.append("foo", "bar", "baz", "TestBean"), value);

        store.add(ServiceName.JBOSS.append("foos", "bar"));
        store.add(ServiceName.JBOSS.append("fo", "bar"));
        store.add(ServiceName.JBOSS.append("foo", "ba", "baz"));
        store.add(ServiceName.JBOSS.append("foo", "bart", "baz"));
        store.add(ServiceName.JBOSS.append("foo", "bar", "ba"));
        store.add(ServiceName.JBOSS.append("foo", "bar", "bazt"));
        store.add(ServiceName.JBOSS.append("foo", "bar", "art"));
        store.add(ServiceName.JBOSS.append("other", "one"));

        List<Binding> list = store.listBindings(new CompositeName(""));
        assertEquals(5, list.size());
        assertContains(list, "TestBean", Object.class);
        assertContains(list, "foo", NamingContext.class);
        assertContains(list, "fo", NamingContext.class);
        assertContains(list, "foos", NamingContext.class);
        assertContains(list, "other", NamingContext.class);


        list = store.listBindings(new CompositeName("foo"));
        assertEquals(4, list.size());
        assertContains(list, "TestBean", Object.class);
        assertContains(list, "ba", NamingContext.class);
        assertContains(list, "bart", NamingContext.class);
        assertContains(list, "bar", NamingContext.class);

        for (Binding binding : list) {
            if (binding.getName().equals("bar")) {
                final Object bean = Context.class.cast(binding.getObject()).lookup("TestBean");
                assertNotNull(bean);
                assertEquals(value, bean);
            }
        }
    }

    private void assertContains(final List<? extends NameClassPair> list, String name, Class<?> type) {
        for (NameClassPair value : list) {
            if (value instanceof Binding) {
                assertNotNull(Binding.class.cast(value).getObject());
            }
            if (value.getName().equals(name) && value.getClassName().equals(type.getName())) {
                return;
            }
        }
        fail("Child [" + name + "] not found in [" + list + "]");
    }

    private void bindObject(final ServiceName serviceName, final Object value) throws Exception {
        final CountDownLatch latch = new CountDownLatch(1);
        container.addService(serviceName, new Service<ManagedReferenceFactory>() {
            public void start(StartContext context) throws StartException {
                store.add(serviceName);
                latch.countDown();
            }

            public void stop(StopContext context) {
            }

            public ManagedReferenceFactory getValue() throws IllegalStateException, IllegalArgumentException {
                return new ValueManagedReferenceFactory(value);
            }
        }).install();
        latch.await();
    }
}
