Merge pull request #650 from csmith/master

Separate Server and FrameContainer.

src/com/dmdirc/FrameContainer.java

@@ -28,6 +28,7 @@ import com.dmdirc.events.FrameComponentRemovedEvent;
 import com.dmdirc.events.FrameIconChangedEvent;
 import com.dmdirc.events.FrameNameChangedEvent;
 import com.dmdirc.events.FrameTitleChangedEvent;
+import com.dmdirc.interfaces.Connection;
 import com.dmdirc.interfaces.InputModel;
 import com.dmdirc.interfaces.WindowModel;
 import com.dmdirc.interfaces.config.AggregateConfigProvider;
@@ -51,7 +52,7 @@ import javax.annotation.Nullable;
  * The frame container implements basic methods that should be present in all objects that handle a
  * frame.
  */
-public abstract class FrameContainer implements WindowModel {
+public class FrameContainer implements WindowModel {
 
     /** Listeners not yet using ListenerSupport. */
     protected final ListenerList listeners = new ListenerList();
@@ -79,11 +80,13 @@ public abstract class FrameContainer implements WindowModel {
     private BackBuffer backBuffer;
     /** The input model for this container. */
     private Optional<InputModel> inputModel = Optional.empty();
+    /** The connection associated with this model. */
+    private Optional<Connection> connection = Optional.empty();
 
     /**
      * Instantiate new frame container.
      */
-    protected FrameContainer(
+    public FrameContainer(
             final String icon,
             final String name,
             final String title,
@@ -137,12 +140,8 @@ public abstract class FrameContainer implements WindowModel {
         return eventBus;
     }
 
-    /**
-     * Changes the name of this container, and fires a {@link FrameNameChangedEvent}.
-     *
-     * @param name The new name for this frame.
-     */
-    protected void setName(final String name) {
+    @Override
+    public void setName(final String name) {
         this.name = name;
 
         eventBus.publishAsync(new FrameNameChangedEvent(this, name));
@@ -234,4 +233,13 @@ public abstract class FrameContainer implements WindowModel {
         return unreadStatusManager;
     }
 
+    public void setConnection(@Nullable final Connection connection) {
+        this.connection = Optional.ofNullable(connection);
+    }
+
+    @Override
+    public Optional<Connection> getConnection() {
+        return connection;
+    }
+
 }

src/com/dmdirc/Server.java

@@ -23,6 +23,7 @@
 package com.dmdirc;
 
 import com.dmdirc.config.profiles.Profile;
+import com.dmdirc.events.FrameClosingEvent;
 import com.dmdirc.events.ServerConnectErrorEvent;
 import com.dmdirc.events.ServerConnectedEvent;
 import com.dmdirc.events.ServerConnectingEvent;
@@ -49,9 +50,7 @@ import com.dmdirc.parser.interfaces.ProtocolDescription;
 import com.dmdirc.parser.interfaces.SecureParser;
 import com.dmdirc.parser.interfaces.StringConverter;
 import com.dmdirc.tls.CertificateManager;
-import com.dmdirc.ui.core.components.WindowComponent;
 import com.dmdirc.ui.input.TabCompletionType;
-import com.dmdirc.ui.messages.BackBufferFactory;
 import com.dmdirc.ui.messages.ColourManager;
 import com.dmdirc.ui.messages.Formatter;
 import com.dmdirc.ui.messages.HighlightManager;
@@ -63,7 +62,6 @@ import java.net.URI;
 import java.net.URISyntaxException;
 import java.net.UnknownHostException;
 import java.util.ArrayList;
-import java.util.Arrays;
 import java.util.Collection;
 import java.util.Collections;
 import java.util.Map;
@@ -83,6 +81,8 @@ import javax.net.ssl.SSLException;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
+import net.engio.mbassy.listener.Handler;
+
 import static com.dmdirc.util.LogUtils.APP_ERROR;
 import static com.google.common.base.Preconditions.checkNotNull;
 
@@ -90,7 +90,7 @@ import static com.google.common.base.Preconditions.checkNotNull;
  * The Server class represents the client's view of a server. It maintains a list of all channels,
  * queries, etc, and handles parser callbacks pertaining to the server.
  */
-public class Server extends FrameContainer implements Connection {
+public class Server implements Connection {
 
     private static final Logger LOG = LoggerFactory.getLogger(Server.class);
     /** The name of the general domain. */
@@ -163,6 +163,7 @@ public class Server extends FrameContainer implements Connection {
     private final HighlightManager highlightManager;
     /** Listener to use for config changes. */
     private final ConfigChangeListener configListener = (domain, key) -> updateTitle();
+    private final WindowModel windowModel;
     /** The future used when a reconnect timer is scheduled. */
     private ScheduledFuture<?> reconnectTimerFuture;
 
@@ -170,30 +171,19 @@ public class Server extends FrameContainer implements Connection {
      * Creates a new server which will connect to the specified URL with the specified profile.
      */
     public Server(
+            final WindowModel windowModel,
             final ConfigProviderMigrator configMigrator,
             final ParserFactory parserFactory,
             final IdentityFactory identityFactory,
             final QueryFactory queryFactory,
-            final DMDircMBassador eventBus,
             final MessageEncoderFactory messageEncoderFactory,
             final ConfigProvider userSettings,
             final GroupChatManagerImplFactory groupChatManagerFactory,
             final ScheduledExecutorService executorService,
             @Nonnull final URI uri,
             @Nonnull final Profile profile,
-            final BackBufferFactory backBufferFactory,
             final UserManager userManager) {
-        super("server-disconnected",
-                getHost(uri),
-                getHost(uri),
-                configMigrator.getConfigProvider(),
-                backBufferFactory,
-                eventBus,
-                Arrays.asList(
-                        WindowComponent.TEXTAREA.getIdentifier(),
-                        WindowComponent.INPUTFIELD.getIdentifier(),
-                        WindowComponent.CERTIFICATE_VIEWER.getIdentifier()));
-
+        this.windowModel = windowModel;
         this.parserFactory = parserFactory;
         this.identityFactory = identityFactory;
         this.configMigrator = configMigrator;
@@ -206,7 +196,7 @@ public class Server extends FrameContainer implements Connection {
         this.inviteManager = new InviteManagerImpl(this);
 
         awayMessage = Optional.empty();
-        eventHandler = new ServerEventHandler(this, groupChatManager, getEventBus());
+        eventHandler = new ServerEventHandler(this, groupChatManager, windowModel.getEventBus());
 
         this.address = uri;
         this.profile = profile;
@@ -214,14 +204,14 @@ public class Server extends FrameContainer implements Connection {
 
         updateIcon();
 
-        getConfigManager().addChangeListener("formatter", "serverName", configListener);
-        getConfigManager().addChangeListener("formatter", "serverTitle", configListener);
+        windowModel.getConfigManager().addChangeListener("formatter", "serverName", configListener);
+        windowModel.getConfigManager().addChangeListener("formatter", "serverTitle", configListener);
 
-        initBackBuffer();
-        this.highlightManager = new HighlightManager(getConfigManager(),
-                new ColourManager(getConfigManager()));
+        this.highlightManager = new HighlightManager(windowModel.getConfigManager(),
+                new ColourManager(windowModel.getConfigManager()));
         highlightManager.init();
-        getEventBus().subscribe(highlightManager);
+        windowModel.getEventBus().subscribe(highlightManager);
+        windowModel.getEventBus().subscribe(this);
     }
 
     /**
@@ -278,7 +268,8 @@ public class Server extends FrameContainer implements Connection {
                     return;
                 case CONNECTED:
                 case CONNECTING:
-                    disconnect(getConfigManager().getOption(DOMAIN_GENERAL, "quitmessage"));
+                    disconnect(windowModel.getConfigManager()
+                            .getOption(DOMAIN_GENERAL, "quitmessage"));
                 case DISCONNECTING:
                     while (!myState.getState().isDisconnected()) {
                         try {
@@ -313,7 +304,7 @@ public class Server extends FrameContainer implements Connection {
                 parser = Optional.ofNullable(buildParser());
 
                 if (!parser.isPresent()) {
-                    getEventBus().publishAsync(
+                    windowModel.getEventBus().publishAsync(
                             new ServerUnknownProtocolEvent(this, address.getScheme()));
                     return;
                 }
@@ -338,7 +329,7 @@ public class Server extends FrameContainer implements Connection {
             }
         }
 
-        getEventBus().publish(new ServerConnectingEvent(this, address));
+        windowModel.getEventBus().publish(new ServerConnectingEvent(this, address));
     }
 
     @Override
@@ -356,12 +347,12 @@ public class Server extends FrameContainer implements Connection {
 
     @Override
     public void reconnect() {
-        reconnect(getConfigManager().getOption(DOMAIN_GENERAL, "reconnectmessage"));
+        reconnect(windowModel.getConfigManager().getOption(DOMAIN_GENERAL, "reconnectmessage"));
     }
 
     @Override
     public void disconnect() {
-        disconnect(getConfigManager().getOption(DOMAIN_GENERAL, "quitmessage"));
+        disconnect(windowModel.getConfigManager().getOption(DOMAIN_GENERAL, "quitmessage"));
     }
 
     @Override
@@ -403,7 +394,8 @@ public class Server extends FrameContainer implements Connection {
                 parserLock.readLock().unlock();
             }
 
-            if (getConfigManager().getOptionBool(DOMAIN_GENERAL, "closequeriesonquit")) {
+            if (windowModel.getConfigManager()
+                    .getOptionBool(DOMAIN_GENERAL, "closequeriesonquit")) {
                 closeQueries();
             }
         }
@@ -423,9 +415,10 @@ public class Server extends FrameContainer implements Connection {
             }
 
             final int delay = Math.max(1000,
-                    getConfigManager().getOptionInt(DOMAIN_GENERAL, "reconnectdelay"));
+                    windowModel.getConfigManager().getOptionInt(DOMAIN_GENERAL, "reconnectdelay"));
 
-            getEventBus().publishAsync(new ServerReconnectScheduledEvent(this, delay / 1000));
+            windowModel.getEventBus().publishAsync(
+                    new ServerReconnectScheduledEvent(this, delay / 1000));
 
             reconnectTimerFuture = executorService.schedule(() -> {
                 synchronized (myStateLock) {
@@ -483,7 +476,8 @@ public class Server extends FrameContainer implements Connection {
             if (!getState().isDisconnected()) {
                 newQuery.reregister();
             }
-            getInputModel().get().getTabCompleter().addEntry(TabCompletionType.QUERY_NICK, nick);
+            windowModel.getInputModel().get().getTabCompleter()
+                    .addEntry(TabCompletionType.QUERY_NICK, nick);
             queries.put(lnick, newQuery);
         }
 
@@ -492,8 +486,10 @@ public class Server extends FrameContainer implements Connection {
 
     @Override
     public void updateQuery(final Query query, final String oldNick, final String newNick) {
-        getInputModel().get().getTabCompleter().removeEntry(TabCompletionType.QUERY_NICK, oldNick);
-        getInputModel().get().getTabCompleter().addEntry(TabCompletionType.QUERY_NICK, newNick);
+        windowModel.getInputModel().get().getTabCompleter()
+                .removeEntry(TabCompletionType.QUERY_NICK, oldNick);
+        windowModel.getInputModel().get().getTabCompleter()
+                .addEntry(TabCompletionType.QUERY_NICK, newNick);
 
         queries.put(converter.toLowerCase(newNick), query);
         queries.remove(converter.toLowerCase(oldNick));
@@ -506,7 +502,7 @@ public class Server extends FrameContainer implements Connection {
 
     @Override
     public void delQuery(final Query query) {
-        getInputModel().get().getTabCompleter().removeEntry(
+        windowModel.getInputModel().get().getTabCompleter().removeEntry(
                 TabCompletionType.QUERY_NICK, query.getNickname());
         queries.remove(converter.toLowerCase(query.getNickname()));
     }
@@ -518,26 +514,6 @@ public class Server extends FrameContainer implements Connection {
         new ArrayList<>(queries.values()).forEach(Query::close);
     }
 
-    /**
-     * Retrieves the host component of the specified URI, or throws a relevant exception if this is
-     * not possible.
-     *
-     * @param uri The URI to be processed
-     *
-     * @return The URI's host component, as returned by {@link URI#getHost()}.
-     *
-     * @throws NullPointerException     If <code>uri</code> is null
-     * @throws IllegalArgumentException If the specified URI has no host
-     * @since 0.6.4
-     */
-    private static String getHost(final URI uri) {
-        if (uri.getHost() == null) {
-            throw new IllegalArgumentException("URIs must have hosts");
-        }
-
-        return uri.getHost();
-    }
-
     /**
      * Builds an appropriately configured {@link Parser} for this server.
      *
@@ -545,7 +521,8 @@ public class Server extends FrameContainer implements Connection {
      */
     @Nullable
     private Parser buildParser() {
-        final Parser myParser = parserFactory.getParser(profile, address, getConfigManager())
+        final Parser myParser = parserFactory
+                .getParser(profile, address, windowModel.getConfigManager())
                 .orElse(null);
 
         if (myParser != null) {
@@ -554,8 +531,8 @@ public class Server extends FrameContainer implements Connection {
 
         if (myParser instanceof SecureParser) {
             final CertificateManager certificateManager =
-                    new CertificateManager(this, address.getHost(), getConfigManager(),
-                            userSettings, getEventBus());
+                    new CertificateManager(this, address.getHost(), windowModel.getConfigManager(),
+                            userSettings, windowModel.getEventBus());
             final SecureParser secureParser = (SecureParser) myParser;
             secureParser.setTrustManagers(certificateManager);
             secureParser.setKeyManagers(certificateManager.getKeyManager());
@@ -582,7 +559,7 @@ public class Server extends FrameContainer implements Connection {
         final String icon = myState.getState() == ServerState.CONNECTED
                 ? protocolDescription.get().isSecure(address)
                 ? "secure-server" : "server" : "server-disconnected";
-        setIcon(icon);
+        windowModel.setIcon(icon);
     }
 
     /**
@@ -726,37 +703,38 @@ public class Server extends FrameContainer implements Connection {
         return myState;
     }
 
-    @Override
-    public void close() {
-        synchronized (myStateLock) {
-            eventHandler.unregisterCallbacks();
-            getConfigManager().removeListener(configListener);
-            highlightManager.stop();
-            getEventBus().unsubscribe(highlightManager);
-            executorService.shutdown();
+    @Handler
+    private void handleClose(final FrameClosingEvent event) {
+        if (event.getContainer() == windowModel) {
+            synchronized (myStateLock) {
+                eventHandler.unregisterCallbacks();
+                windowModel.getConfigManager().removeListener(configListener);
+                highlightManager.stop();
+                windowModel.getEventBus().unsubscribe(highlightManager);
+                executorService.shutdown();
 
-            disconnect();
+                disconnect();
 
-            myState.transition(ServerState.CLOSING);
-        }
-
-        groupChatManager.closeAll();
-        closeQueries();
-        inviteManager.removeInvites();
+                myState.transition(ServerState.CLOSING);
+            }
 
-        super.close();
+            groupChatManager.closeAll();
+            closeQueries();
+            inviteManager.removeInvites();
+            windowModel.getEventBus().unsubscribe(this);
+        }
     }
 
     @Override
     public WindowModel getWindowModel() {
-        return this;
+        return windowModel;
     }
 
     @Override
     public void sendCTCPReply(final String source, final String type, final String args) {
         if ("VERSION".equalsIgnoreCase(type)) {
             parser.get().sendCTCPReply(source, "VERSION",
-                    "DMDirc " + getConfigManager().getOption("version", "version") +
+                    "DMDirc " + windowModel.getConfigManager().getOption("version", "version") +
                             " - https://www.dmdirc.com/");
         } else if ("PING".equalsIgnoreCase(type)) {
             parser.get().sendCTCPReply(source, "PING", args);
@@ -765,11 +743,6 @@ public class Server extends FrameContainer implements Connection {
         }
     }
 
-    @Override
-    public Optional<Connection> getConnection() {
-        return Optional.of(this);
-    }
-
     @Override
     public void updateTitle() {
         synchronized (myStateLock) {
@@ -785,9 +758,9 @@ public class Server extends FrameContainer implements Connection {
                     getLocalUser().map(User::getNickname).orElse("Unknown")
                 };
 
-                setName(Formatter.formatMessage(getConfigManager(),
+                windowModel.setName(Formatter.formatMessage(windowModel.getConfigManager(),
                         "serverName", arguments));
-                setTitle(Formatter.formatMessage(getConfigManager(),
+                windowModel.setTitle(Formatter.formatMessage(windowModel.getConfigManager(),
                         "serverTitle", arguments));
             } finally {
                 parserLock.readLock().unlock();
@@ -808,7 +781,7 @@ public class Server extends FrameContainer implements Connection {
             return;
         }
 
-        getEventBus().publish(new ServerDisconnectedEvent(this));
+        windowModel.getEventBus().publish(new ServerDisconnectedEvent(this));
 
         eventHandler.unregisterCallbacks();
 
@@ -837,14 +810,16 @@ public class Server extends FrameContainer implements Connection {
 
             updateIcon();
 
-            if (getConfigManager().getOptionBool(DOMAIN_GENERAL, "closequeriesondisconnect")) {
+            if (windowModel.getConfigManager()
+                    .getOptionBool(DOMAIN_GENERAL, "closequeriesondisconnect")) {
                 closeQueries();
             }
 
             inviteManager.removeInvites();
             updateAwayState(Optional.empty());
 
-            if (getConfigManager().getOptionBool(DOMAIN_GENERAL, "reconnectondisconnect")
+            if (windowModel.getConfigManager()
+                    .getOptionBool(DOMAIN_GENERAL, "reconnectondisconnect")
                     && myState.getState() == ServerState.TRANSIENTLY_DISCONNECTED) {
                 doDelayedReconnect();
             }
@@ -884,10 +859,11 @@ public class Server extends FrameContainer implements Connection {
 
             updateIcon();
 
-            getEventBus().publish(new ServerConnectErrorEvent(this, getErrorDescription
-                    (errorInfo)));
+            windowModel.getEventBus().publish(new ServerConnectErrorEvent(this,
+                    getErrorDescription(errorInfo)));
 
-            if (getConfigManager().getOptionBool(DOMAIN_GENERAL, "reconnectonconnectfailure")) {
+            if (windowModel.getConfigManager()
+                    .getOptionBool(DOMAIN_GENERAL, "reconnectonconnectfailure")) {
                 doDelayedReconnect();
             }
         }
@@ -949,7 +925,7 @@ public class Server extends FrameContainer implements Connection {
             groupChatManager.handleConnected();
         }
 
-        getEventBus().publish(new ServerConnectedEvent(this));
+        windowModel.getEventBus().publish(new ServerConnectedEvent(this));
     }
 
     @Override
@@ -960,7 +936,7 @@ public class Server extends FrameContainer implements Connection {
     @Override
     public void updateIgnoreList() {
         ignoreList.clear();
-        ignoreList.addAll(getConfigManager().getOptionList("network", "ignorelist"));
+        ignoreList.addAll(windowModel.getConfigManager().getOptionList("network", "ignorelist"));
     }
 
     @Override

src/com/dmdirc/ServerEventHandler.java

@@ -248,14 +248,14 @@ public class ServerEventHandler extends EventHandler {
     @Handler
     public void onPingFailed(final PingFailureEvent event) {
         eventBus.publishAsync(new StatusBarMessageEvent(new StatusMessage(
-                "No ping reply from " + owner.getName() + " for over " +
+                "No ping reply from " + owner.getWindowModel().getName() + " for over " +
                         (int) Math.floor(event.getParser().getPingTime() / 1000.0) + " seconds.",
-                owner.getConfigManager())));
+                owner.getWindowModel().getConfigManager())));
 
         eventBus.publishAsync(new ServerNoPingEvent(owner, event.getParser().getPingTime()));
 
         if (event.getParser().getPingTime()
-                >= owner.getConfigManager().getOptionInt("server", "pingtimeout")) {
+                >= owner.getWindowModel().getConfigManager().getOptionInt("server", "pingtimeout")) {
             LOG.warn("Server appears to be stoned, reconnecting");
             eventBus.publishAsync(new ServerStonedEvent(owner));
             owner.reconnect();

src/com/dmdirc/ServerFactoryImpl.java

@@ -29,10 +29,12 @@ import com.dmdirc.interfaces.CommandController;
 import com.dmdirc.interfaces.config.ConfigProvider;
 import com.dmdirc.interfaces.config.ConfigProviderMigrator;
 import com.dmdirc.interfaces.config.IdentityFactory;
+import com.dmdirc.ui.core.components.WindowComponent;
 import com.dmdirc.ui.input.TabCompleterFactory;
 import com.dmdirc.ui.messages.BackBufferFactory;
 
 import java.net.URI;
+import java.util.Arrays;
 import java.util.concurrent.ScheduledExecutorService;
 
 import javax.inject.Inject;
@@ -40,7 +42,7 @@ import javax.inject.Provider;
 import javax.inject.Singleton;
 
 /**
- * Factory for {@link Server}s
+ * Factory for {@link Server}s.
  */
 @Singleton
 public class ServerFactoryImpl {
@@ -88,14 +90,22 @@ public class ServerFactoryImpl {
             final ScheduledExecutorService executorService,
             final URI uri,
             final Profile profile) {
-        final Server server = new Server(configMigrator, parserFactory,
-                identityFactory, queryFactory.get(), eventBus,
+        final FrameContainer windowModel =
+                new FrameContainer("server-disconnected", getHost(uri), getHost(uri),
+                        configMigrator.getConfigProvider(), backBufferFactory, eventBus,
+                        Arrays.asList(WindowComponent.TEXTAREA.getIdentifier(),
+                                WindowComponent.INPUTFIELD.getIdentifier(),
+                                WindowComponent.CERTIFICATE_VIEWER.getIdentifier()));
+        final Server server = new Server(windowModel, configMigrator, parserFactory,
+                identityFactory, queryFactory.get(),
                 messageEncoderFactory, userSettings, groupChatManagerFactory, executorService,
-                uri, profile, backBufferFactory, userManager);
-        server.setInputModel(new DefaultInputModel(
+                uri, profile, userManager);
+        windowModel.setConnection(server);
+        windowModel.initBackBuffer();
+        windowModel.setInputModel(new DefaultInputModel(
                 server::sendLine,
                 new ServerCommandParser(
-                        server.getConfigManager(),
+                        server.getWindowModel().getConfigManager(),
                         commandController.get(),
                         eventBus,
                         server),
@@ -106,4 +116,25 @@ public class ServerFactoryImpl {
                 server::getMaxLineLength));
         return server;
     }
+
+    /**
+     * Retrieves the host component of the specified URI, or throws a relevant exception if this is
+     * not possible.
+     *
+     * @param uri The URI to be processed
+     *
+     * @return The URI's host component, as returned by {@link URI#getHost()}.
+     *
+     * @throws NullPointerException     If <code>uri</code> is null
+     * @throws IllegalArgumentException If the specified URI has no host
+     * @since 0.6.4
+     */
+    private static String getHost(final URI uri) {
+        if (uri.getHost() == null) {
+            throw new IllegalArgumentException("URIs must have hosts");
+        }
+
+        return uri.getHost();
+    }
+
 }

src/com/dmdirc/ServerManager.java

@@ -113,7 +113,7 @@ public class ServerManager implements ConnectionManager {
                 uri,
                 profile);
         registerServer(server);
-        windowManager.addWindow(server);
+        windowManager.addWindow(server.getWindowModel());
         return server;
     }
 
@@ -152,7 +152,7 @@ public class ServerManager implements ConnectionManager {
     public void closeAll(final String message) {
         for (Server server : servers) {
             server.disconnect(message);
-            server.close();
+            server.getWindowModel().close();
         }
     }
 

src/com/dmdirc/interfaces/WindowModel.java

@@ -24,6 +24,7 @@ package com.dmdirc.interfaces;
 
 import com.dmdirc.DMDircMBassador;
 import com.dmdirc.events.FrameIconChangedEvent;
+import com.dmdirc.events.FrameNameChangedEvent;
 import com.dmdirc.events.FrameTitleChangedEvent;
 import com.dmdirc.interfaces.config.AggregateConfigProvider;
 import com.dmdirc.ui.messages.BackBuffer;
@@ -47,6 +48,13 @@ public interface WindowModel {
 
     DMDircMBassador getEventBus();
 
+    /**
+     * Changes the name of this container, and fires a {@link FrameNameChangedEvent}.
+     *
+     * @param name The new name for this frame.
+     */
+    void setName(String name);
+
     /**
      * Changes the title of this container, and fires a {@link FrameTitleChangedEvent}.
      *

test/com/dmdirc/ServerManagerTest.java

@@ -25,6 +25,7 @@ package com.dmdirc;
 import com.dmdirc.config.profiles.Profile;
 import com.dmdirc.config.profiles.ProfileManager;
 import com.dmdirc.interfaces.GroupChatManager;
+import com.dmdirc.interfaces.WindowModel;
 import com.dmdirc.interfaces.config.AggregateConfigProvider;
 import com.dmdirc.interfaces.config.ConfigProviderMigrator;
 import com.dmdirc.interfaces.config.IdentityFactory;
@@ -63,6 +64,7 @@ public class ServerManagerTest {
     @Mock private WindowManager windowManager;
     @Mock private ServerFactoryImpl serverFactoryImpl;
     @Mock private Server server;
+    @Mock private WindowModel windowModel;
     @Mock private GroupChatManager groupChatManager;
     @Mock private DMDircMBassador eventBus;
     @Mock private Channel channel;
@@ -77,6 +79,7 @@ public class ServerManagerTest {
 
         when(server.getState()).thenReturn(ServerState.DISCONNECTED);
         when(server.getGroupChatManager()).thenReturn(groupChatManager);
+        when(server.getWindowModel()).thenReturn(windowModel);
 
         when(profileManager.getProfiles()).thenReturn(Collections.singletonList(profile));
         when(profileManager.getDefault()).thenReturn(profile);
@@ -138,10 +141,12 @@ public class ServerManagerTest {
     @Test
     public void testCloseAllWithMessage() {
         final Server serverA = mock(Server.class);
+        final WindowModel model = mock(WindowModel.class);
+        when(serverA.getWindowModel()).thenReturn(model);
         serverManager.registerServer(serverA);
         serverManager.closeAll("message here");
         verify(serverA).disconnect("message here");
-        verify(serverA).close();
+        verify(model).close();
     }
 
     @Test
@@ -206,7 +211,7 @@ public class ServerManagerTest {
     @Test
     public void testAddsNewServersToWindowManager() {
         serverManager.connectToAddress(URI.create("irc://fobar"));
-        verify(windowManager).addWindow(server);
+        verify(windowManager).addWindow(windowModel);
     }
 
 }