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

package com.dmdirc;

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 java.io.File;
import java.io.IOException;
import static org.mockito.Mockito.*;

/**
 * Main subclass to init things needed for testing.
 */
public class TestMain extends Main {

    private static Main instance;

    private final IdentityManager identityManager;
    private final ServerManager serverManager;
    private final ActionManager actionManager;
    private final PluginManager pluginManager;
    private final String configdir;

    public TestMain(final IdentityManager identityManager,
            final ServerManager serverManager,
            final ActionManager actionManager,
            final CommandLineParser commandLineParser,
            final PluginManager pluginManager,
            final String configDir) {
        super(identityManager, serverManager, actionManager, commandLineParser,
                pluginManager, null, configDir);
        this.identityManager = identityManager;
        this.serverManager = serverManager;
        this.actionManager = actionManager;
        this.pluginManager = pluginManager;
        this.configdir = configDir;
    }

    /** {@inheritDoc} */
    @Override
    public void init() {
        try {
            IdentityManager.getIdentityManager().initialise(configdir);
        } 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.initCommandManager(IdentityManager.getIdentityManager(), serverManager);

        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) {
            // 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();
            IdentityManager.setIdentityManager(identityManager);
            IdentityManager.getIdentityManager().loadVersionIdentity();

            final ServerManager serverManager = mock(ServerManager.class);

            final ActionManager actionManager = new ActionManager(serverManager, identityManager);
            ActionManager.setActionManager(actionManager);

            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;
                final PluginManager pluginManager = new PluginManager(identityManager, actionManager, pluginDirectory);

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