DMDirc-Client/test/com/dmdirc/TestMain.java

package com.dmdirc;

import com.dmdirc.actions.ActionGroup;
import com.dmdirc.actions.ActionManager;
import com.dmdirc.commandline.CommandLineParser;
import com.dmdirc.commandparser.CommandManager;
import com.dmdirc.config.IdentityManager;
import com.dmdirc.config.InvalidIdentityFileException;
import com.dmdirc.plugins.PluginManager;
import com.dmdirc.updater.manager.UpdateManager;

import java.io.File;
import java.io.IOException;
import java.util.Collections;

import javax.inject.Provider;
import static org.mockito.Mockito.*;

/**
 * Main subclass to init things needed for testing.
 *
 * @deprecated Tests shouldn't need the entire universe created.
 */
@Deprecated
public class TestMain extends Main {

    private static Main instance;

    private final IdentityManager identityManager;
    private final PluginManager pluginManager;
    private final CommandManager commandManager;

    public TestMain(final IdentityManager identityManager,
            final ServerManager serverManager,
            final ActionManager actionManager,
            final CommandLineParser commandLineParser,
            final PluginManager pluginManager,
            final CommandManager commandManager,
            final CorePluginExtractor corePluginExtractor) {
        super(identityManager, serverManager, actionManager, commandLineParser,
                pluginManager, commandManager, null, null, corePluginExtractor);
        this.identityManager = identityManager;
        this.pluginManager = pluginManager;
        this.commandManager = commandManager;
    }

    /** {@inheritDoc} */
    @Override
    public void init() {
        try {
            identityManager.initialise();
        } catch (InvalidIdentityFileException ex) {
            // If a bad config dir exists, try to continue anyway, maybe the
            // test doesn't need it.
            // DON'T do anything to the user's configuration... (so no calls
            // to handleInvalidConfigFile(); here)
        }

        pluginManager.refreshPlugins();
        commandManager.initialise(identityManager.getGlobalConfiguration());
        CommandManager.setCommandManager(commandManager);

        ActionManager.getActionManager().initialise();
    }

    /**
     * Singleton method for convenience so that we don't need to init a billion
     * TestMain instances to run tests.
     *
     * Separate instances of TestMain are available, but probably pointless as
     * long as IdentityManager, PluginManager and ConfigManater are still
     * singletons.
     *
     * @return A Singleton instance of TestMain
     */
    public static Main getTestMain() {
        if (instance == null) {
            try {
                File tempFile = File.createTempFile("dmdirc", "test");
                tempFile.delete();
                tempFile.mkdir();
                tempFile.deleteOnExit();

                final String configDirectory = tempFile.getAbsolutePath() + File.separator;
                final String pluginDirectory = configDirectory + "plugins" + File.separator;

                // TODO: Tests probably shouldn't rely on a config dir... Who knows
                //       what the user has done with their config.
                final IdentityManager identityManager = new IdentityManager(configDirectory);
                IdentityManager.setIdentityManager(identityManager);
                IdentityManager.getIdentityManager().loadVersionIdentity();

                final ServerManager serverManager = mock(ServerManager.class);
                final CommandManager commandManager = new CommandManager(serverManager);

                final ActionManager actionManager = new ActionManager(
                        serverManager, identityManager,
                        new DummyProvider<>(Collections.<ActionGroup>emptySet()));
                ActionManager.setActionManager(actionManager);

                final PluginManager pluginManager = new PluginManager(
                        identityManager, actionManager,
                        mock(UpdateManager.class), pluginDirectory);
                final CorePluginExtractor corePluginExtractor =
                        new CorePluginExtractor(pluginManager, pluginDirectory);

                instance = new TestMain(identityManager, serverManager,
                        actionManager, null, pluginManager, commandManager,
                        corePluginExtractor);
                instance.init();
            } catch (IOException ex) {
                // Blargh.
            }
        }
        return instance;
    }

    private static class DummyProvider<T> implements Provider<T> {

        private final T value;

        public DummyProvider(T value) {
            this.value = value;
        }

        @Override
        public T get() {
            return value;
        }
    }
}