/*
 * Copyright (c) 2009-2010 Lockheed Martin Corporation
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.eurekastreams.commons.search.bootstrap;

import javax.persistence.EntityManager;

import org.hibernate.search.FullTextSession;
import org.jmock.Expectations;
import org.jmock.integration.junit4.JUnit4Mockery;
import org.jmock.lib.legacy.ClassImposteriser;
import org.junit.Before;
import org.junit.Test;

/**
 * Test fixture for EntityReindexer. 
 */
public class EntityReindexerTest
{
    /**
     * Helper class to mock out singleton-wrapping method.
     */
    private class EntityReindexerTester extends EntityReindexer
    {
        /**
         * Mocked out fullTextSession.
         */
        private FullTextSession fullTextSession;

        /**
         * Set the FullTextSession mock.
         * 
         * @param theFullTextSession
         *            a mocked FullTextSession to inject into the parent class with the overriden getFullTextSession()
         *            method.
         */
        public void setFullTextSession(final FullTextSession theFullTextSession)
        {
            fullTextSession = theFullTextSession;
        }

        /**
         * Overridden method to avoid necessary Hibernate singleton call.
         * 
         * @return the mocked fullTextSession
         */
        @Override
        protected FullTextSession getFullTextSession()
        {
            return fullTextSession;
        }
    }

    /**
     * Context for mocking.
     */
    private final JUnit4Mockery context = new JUnit4Mockery()
    {
        {
            setImposteriser(ClassImposteriser.INSTANCE);
        }
    };

    /**
     * Test class for reindexing.
     */
    private class TestClass1
    {
    }

    /**
     * Test class for reindexing.
     */
    private class TestClass2
    {
    }

    /**
     * FullTextSession mock.
     */
    FullTextSession fullTextSessionMock;

    /**
     * Setup method for tests.
     */
    @Before
    public void setup()
    {
        fullTextSessionMock = context.mock(FullTextSession.class);
    }

    /**
     * Test the reindex method, making sure it uses the input SearchIndexManager to reindex.
     */
    @Test
    public void testReindex()
    {
        final SearchIndexManager searchIndexManagerMock = context.mock(SearchIndexManager.class);
        final EntityManager emMock = context.mock(EntityManager.class);
        
        context.checking(new Expectations()
        {
            {
                one(searchIndexManagerMock).reindexEntities(TestClass1.class, fullTextSessionMock);
                one(searchIndexManagerMock).reindexEntities(TestClass2.class, fullTextSessionMock);
            }
        });

        EntityReindexerTester reindexer = new EntityReindexerTester();
        reindexer.setFullTextSession(fullTextSessionMock);
        reindexer.setSearchIndexManager(searchIndexManagerMock);
        reindexer.setEntityManager(emMock);
        reindexer.setEntitiesToReindex(new Class[] { TestClass1.class, TestClass2.class });

        // invoke SUT
        reindexer.reindex();

        // all expectations met?
        context.assertIsSatisfied();
    }

    /**
     * Test that setting the entity manager passes through to the underlying SearchIndexManager.
     */
    @Test
    public void testSetEntityManager()
    {
        EntityReindexer reindexer = new EntityReindexer();

        // invoke SUT
        reindexer.setEntityManager(context.mock(EntityManager.class));
    }
}