/*

 * All Sigmah code is released under the GNU General Public License v3

 * See COPYRIGHT.txt and LICENSE.txt.

 */



package org.sigmah.server.endpoint.gwtrpc;





import com.google.gwt.user.client.rpc.AsyncCallback;

import com.google.inject.Inject;

import com.google.inject.Injector;



import org.sigmah.client.dispatch.AsyncMonitor;

import org.sigmah.client.dispatch.Dispatcher;

import org.sigmah.shared.domain.User;

import org.sigmah.server.util.BeanMappingModule;

import org.sigmah.server.util.TemplateModule;

import org.sigmah.shared.command.Command;

import org.sigmah.shared.command.result.CommandResult;

import org.sigmah.shared.exception.CommandException;

import org.sigmah.test.MockHibernateModule;

import org.sigmah.test.Modules;



import javax.persistence.EntityManager;

import java.util.Collections;

import java.util.List;



@Modules({

        MockHibernateModule.class,

        TemplateModule.class,

        BeanMappingModule.class,

        GwtRpcModule.class

})

public abstract class CommandTestCase {



    @Inject

    protected CommandServlet servlet;

    @Inject

    protected EntityManager em;



    @Inject

    protected Injector injector;



    protected int userId = 1;



    

    protected void setUser(int userId) {

        this.userId = userId;

    }



    protected <T extends CommandResult> T execute(Command<T> command) throws CommandException {

        User user = em.find(User.class, userId);

        assert user != null;



        List<CommandResult> results = servlet.handleCommands(user, Collections.<Command>singletonList(command));



        // normally each request and so each handleCommand() gets its own

        // EntityManager, but here successive requests in the same test

        // will share an EntityManager, which can be bad if there are collections

        // still living in the first-level cache

        //

        // I think these command tests should ultimately become real end-to-end

        // tests and so would go through the actual servlet process, but for the moment,

        // we'll just add this work aroudn that clears the cache after each command.

        em.clear();





        CommandResult result = results.get(0);

        if (result instanceof CommandException) {

            throw (CommandException) result;

        }



        return (T) result;

    }

    

    protected Dispatcher dispatcher = new Dispatcher() {

		

		@Override

		public <T extends CommandResult> void execute(Command<T> command,

				AsyncMonitor monitor, AsyncCallback<T> callback) {

			try {

				callback.onSuccess(CommandTestCase.this.execute(command));

			} catch(Exception e) {

				System.out.println("Exception thrown while handling command " + command.toString() + ": ");

				e.printStackTrace();

				callback.onFailure(e);

			}

		}

	};

    

}