Move business log out of handler

* Remove  unnecessary test base class

5 files changed, 226 insertions, 238 deletions

diff --git a/configserver/src/main/java/com/yahoo/vespa/config/server/ApplicationRepository.java b/configserver/src/main/java/com/yahoo/vespa/config/server/ApplicationRepository.java
index 698e7baaaf0..951e73e6e8a 100644
--- a/configserver/src/main/java/com/yahoo/vespa/config/server/ApplicationRepository.java
+++ b/configserver/src/main/java/com/yahoo/vespa/config/server/ApplicationRepository.java
@@ -14,16 +14,20 @@ import com.yahoo.log.LogLevel;
 import com.yahoo.transaction.NestedTransaction;
 import com.yahoo.vespa.config.server.application.Application;
 import com.yahoo.vespa.config.server.application.ApplicationConvergenceChecker;
+import com.yahoo.vespa.config.server.application.ApplicationSet;
 import com.yahoo.vespa.config.server.application.LogServerLogGrabber;
 import com.yahoo.vespa.config.server.application.TenantApplications;
+import com.yahoo.vespa.config.server.configchange.ConfigChangeActions;
 import com.yahoo.vespa.config.server.deploy.Deployment;
 import com.yahoo.vespa.config.server.http.ContentHandler;
 import com.yahoo.vespa.config.server.http.v2.ApplicationContentRequest;
 import com.yahoo.vespa.config.server.provision.HostProvisionerProvider;
 import com.yahoo.vespa.config.server.session.LocalSession;
 import com.yahoo.vespa.config.server.session.LocalSessionRepo;
+import com.yahoo.vespa.config.server.session.PrepareParams;
 import com.yahoo.vespa.config.server.session.RemoteSession;
 import com.yahoo.vespa.config.server.session.SessionFactory;
+import com.yahoo.vespa.config.server.session.Session;
 import com.yahoo.vespa.config.server.session.SilentDeployLogger;
 import com.yahoo.vespa.config.server.tenant.ActivateLock;
 import com.yahoo.vespa.config.server.tenant.Rotations;
@@ -235,6 +239,48 @@ public class ApplicationRepository implements com.yahoo.config.provision.Deploye
         return getLocalSession(tenant, sessionId).getMetaData();
     }
 
+    public void validateThatSessionIsNotActive(Tenant tenant, long sessionId) {
+        LocalSession session = getLocalSession(tenant, sessionId);
+        if (Session.Status.ACTIVATE.equals(session.getStatus())) {
+            throw new IllegalStateException("Session is active: " + sessionId);
+        }
+    }
+
+    public void validateThatSessionIsPrepared(Tenant tenant, long sessionId) {
+        LocalSession session = getLocalSession(tenant, sessionId);
+        if (!Session.Status.PREPARE.equals(session.getStatus()))
+            throw new IllegalStateException("Session not prepared: " + sessionId);
+    }
+
+    private Optional<ApplicationSet> getCurrentActiveApplicationSet(Tenant tenant, ApplicationId appId) {
+        Optional<ApplicationSet> currentActiveApplicationSet = Optional.empty();
+        TenantApplications applicationRepo = tenant.getApplicationRepo();
+        try {
+            long currentActiveSessionId = applicationRepo.getSessionIdForApplication(appId);
+            RemoteSession currentActiveSession = getRemoteSession(tenant, currentActiveSessionId);
+            if (currentActiveSession != null) {
+                currentActiveApplicationSet = Optional.ofNullable(currentActiveSession.ensureApplicationLoaded());
+            }
+        } catch (IllegalArgumentException e) {
+            // Do nothing if we have no currently active session
+        }
+        return currentActiveApplicationSet;
+    }
+
+    public ConfigChangeActions prepare(Tenant tenant,
+                                       long sessionId,
+                                       DeployLogger logger,
+                                       PrepareParams params) {
+        LocalSession session = getLocalSession(tenant, sessionId);
+        ApplicationId appId = params.getApplicationId();
+        Optional<ApplicationSet> currentActiveApplicationSet = getCurrentActiveApplicationSet(tenant, appId);
+        return session.prepare(logger,
+                               params,
+                               currentActiveApplicationSet,
+                               tenant.getPath());
+    }
+
+
     private List<ApplicationId> listApplicationIds(Tenant tenant) {
         TenantApplications applicationRepo = tenant.getApplicationRepo();
         return applicationRepo.listApplications();
diff --git a/configserver/src/main/java/com/yahoo/vespa/config/server/http/v2/SessionPrepareHandler.java b/configserver/src/main/java/com/yahoo/vespa/config/server/http/v2/SessionPrepareHandler.java
index e1c20ca2c43..731f7db262d 100644
--- a/configserver/src/main/java/com/yahoo/vespa/config/server/http/v2/SessionPrepareHandler.java
+++ b/configserver/src/main/java/com/yahoo/vespa/config/server/http/v2/SessionPrepareHandler.java
@@ -12,13 +12,8 @@ import com.yahoo.container.logging.AccessLog;
 import com.yahoo.log.LogLevel;
 import com.yahoo.slime.Slime;
 import com.yahoo.vespa.config.server.ApplicationRepository;
-import com.yahoo.vespa.config.server.application.ApplicationSet;
-import com.yahoo.vespa.config.server.application.TenantApplications;
 import com.yahoo.vespa.config.server.configchange.RestartActions;
-import com.yahoo.vespa.config.server.session.LocalSession;
 import com.yahoo.vespa.config.server.session.PrepareParams;
-import com.yahoo.vespa.config.server.session.RemoteSession;
-import com.yahoo.vespa.config.server.session.Session;
 import com.yahoo.vespa.config.server.tenant.Tenant;
 import com.yahoo.vespa.config.server.tenant.Tenants;
 import com.yahoo.vespa.config.server.configchange.ConfigChangeActions;
@@ -27,7 +22,6 @@ import com.yahoo.vespa.config.server.http.SessionHandler;
 import com.yahoo.vespa.config.server.http.Utils;
 
 import java.time.Duration;
-import java.util.Optional;
 import java.util.concurrent.Executor;
 import java.util.logging.Level;
 import java.util.logging.Logger;
@@ -62,19 +56,16 @@ public class SessionPrepareHandler extends SessionHandler {
         Tenant tenant = getExistingTenant(request);
         TenantName tenantName = tenant.getName();
         long sessionId = getSessionIdV2(request);
-        LocalSession session = applicationRepository.getLocalSession(tenant, getSessionIdV2(request));
-        validateThatSessionIsNotActive(session);
-        log.log(LogLevel.DEBUG, "session=" + session);
-        boolean verbose = request.getBooleanProperty("verbose");
-        Slime rawDeployLog = createDeployLog();
+        applicationRepository.validateThatSessionIsNotActive(tenant, sessionId);
         PrepareParams prepareParams = PrepareParams.fromHttpRequest(request, tenantName, zookeeperBarrierTimeout);
         // An app id currently using only the name
         ApplicationId appId = prepareParams.getApplicationId();
-        DeployLogger logger = createLogger(rawDeployLog, verbose, appId);
-        ConfigChangeActions actions = session.prepare(logger,
-                                                      prepareParams,
-                                                      getCurrentActiveApplicationSet(tenant, appId),
-                                                      tenant.getPath());
+        Slime rawDeployLog = createDeployLog();
+        DeployLogger logger = createLogger(rawDeployLog, request.getBooleanProperty("verbose"), appId);
+        ConfigChangeActions actions = applicationRepository.prepare(tenant,
+                                                                    sessionId,
+                                                                    logger,
+                                                                    prepareParams);
         logConfigChangeActions(actions, logger);
         log.log(LogLevel.INFO, Tenants.logPre(appId) + "Session " + sessionId + " prepared successfully. ");
         return new SessionPrepareResponse(rawDeployLog, tenantName, request, sessionId, actions);
@@ -98,36 +89,10 @@ public class SessionPrepareHandler extends SessionHandler {
     @Override
     protected HttpResponse handleGET(HttpRequest request) {
         Tenant tenant = getExistingTenant(request);
-        RemoteSession session = applicationRepository.getRemoteSession(tenant, getSessionIdV2(request));
-        validateThatSessionIsNotActive(session);
-        validateThatSessionIsPrepared(session);
         long sessionId = getSessionIdV2(request);
-        return new SessionPrepareResponse(createDeployLog(), tenant.getName(), request, sessionId, new ConfigChangeActions());
-    }
-
-    private void validateThatSessionIsNotActive(Session session) {
-        if (Session.Status.ACTIVATE.equals(session.getStatus()))
-            throw new IllegalStateException("Session is active: " + session.getSessionId());
-    }
-
-    private void validateThatSessionIsPrepared(Session session) {
-        if ( ! Session.Status.PREPARE.equals(session.getStatus()))
-            throw new IllegalStateException("Session not prepared: " + session.getSessionId());
-    }
-
-    private static Optional<ApplicationSet> getCurrentActiveApplicationSet(Tenant tenant, ApplicationId appId) {
-        Optional<ApplicationSet> currentActiveApplicationSet = Optional.empty();
-        TenantApplications applicationRepo = tenant.getApplicationRepo();
-        try {
-            long currentActiveSessionId = applicationRepo.getSessionIdForApplication(appId);
-            final RemoteSession currentActiveSession = tenant.getRemoteSessionRepo().getSession(currentActiveSessionId);
-            if (currentActiveSession != null) {
-                currentActiveApplicationSet = Optional.ofNullable(currentActiveSession.ensureApplicationLoaded());
-            }
-        } catch (IllegalArgumentException e) {
-            // Do nothing if we have no currently active session
-        }
-        return currentActiveApplicationSet;
+        applicationRepository.validateThatSessionIsNotActive(tenant, sessionId);
+        applicationRepository.validateThatSessionIsPrepared(tenant, sessionId);
+        return new SessionPrepareResponse(createDeployLog(), tenant.getName(), request, sessionId);
     }
 
     private Tenant getExistingTenant(HttpRequest request) {
diff --git a/configserver/src/main/java/com/yahoo/vespa/config/server/http/v2/SessionPrepareResponse.java b/configserver/src/main/java/com/yahoo/vespa/config/server/http/v2/SessionPrepareResponse.java
index 84a2a50d12e..69b70011013 100644
--- a/configserver/src/main/java/com/yahoo/vespa/config/server/http/v2/SessionPrepareResponse.java
+++ b/configserver/src/main/java/com/yahoo/vespa/config/server/http/v2/SessionPrepareResponse.java
@@ -16,6 +16,10 @@ import com.yahoo.vespa.config.server.http.SessionResponse;
  */
 class SessionPrepareResponse extends SessionResponse {
 
+    public SessionPrepareResponse(Slime deployLog, TenantName tenantName, HttpRequest request, long sessionId) {
+        this(deployLog, tenantName, request, sessionId, new ConfigChangeActions());
+    }
+
     public SessionPrepareResponse(Slime deployLog, TenantName tenantName, HttpRequest request, long sessionId, ConfigChangeActions actions) {
         super(deployLog, deployLog.get());
         String message = "Session " + sessionId + " for tenant '" + tenantName.value() + "' prepared.";
diff --git a/configserver/src/test/java/com/yahoo/vespa/config/server/http/SessionPrepareHandlerTestBase.java b/configserver/src/test/java/com/yahoo/vespa/config/server/http/SessionPrepareHandlerTestBase.java
deleted file mode 100644
index 2a3769ae978..00000000000
--- a/configserver/src/test/java/com/yahoo/vespa/config/server/http/SessionPrepareHandlerTestBase.java
+++ /dev/null
@@ -1,185 +0,0 @@
-// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
-package com.yahoo.vespa.config.server.http;
-
-import com.yahoo.config.provision.TenantName;
-import com.yahoo.container.jdisc.HttpRequest;
-import com.yahoo.container.jdisc.HttpResponse;
-import com.yahoo.path.Path;
-import com.yahoo.vespa.config.server.*;
-import com.yahoo.vespa.config.server.session.*;
-
-import com.yahoo.vespa.curator.Curator;
-import org.junit.Test;
-
-import java.io.IOException;
-
-import static com.yahoo.jdisc.http.HttpRequest.Method;
-import static com.yahoo.jdisc.http.HttpResponse.Status.*;
-import static org.hamcrest.Matchers.containsString;
-import static org.hamcrest.Matchers.endsWith;
-import static org.hamcrest.Matchers.not;
-import static org.hamcrest.core.Is.is;
-import static org.junit.Assert.assertNotNull;
-import static org.junit.Assert.assertThat;
-
-/**
- * @author hmusum
- * @since 5.1.14
- */
-public abstract class SessionPrepareHandlerTestBase extends SessionHandlerTest {
-
-    protected Curator curator;
-    private SessionZooKeeperClient zooKeeperClient;
-    protected LocalSessionRepo localRepo;
-
-    protected String preparedMessage = " prepared.\"}";
-    protected String tenantMessage = "";
-
-
-    @Test
-    public void require_error_when_session_id_does_not_exist() throws Exception {
-        // No session with this id exists
-        HttpResponse response = createHandler().handle(SessionHandlerTest.createTestRequest(pathPrefix, Method.PUT, Cmd.PREPARED, 9999L));
-        HandlerTest.assertHttpStatusCodeErrorCodeAndMessage(response, NOT_FOUND, HttpErrorResponse.errorCodes.NOT_FOUND, "Session 9999 was not found");
-    }
-
-    @Test
-    public void require_error_when_session_id_not_a_number() throws Exception {
-        final String session = "notanumber/prepared";
-        HttpResponse response = createHandler().handle(SessionHandlerTest.createTestRequest(pathPrefix + session));
-        HandlerTest.assertHttpStatusCodeErrorCodeAndMessage(response, BAD_REQUEST,
-                HttpErrorResponse.errorCodes.BAD_REQUEST,
-                "Session id in request is not a number, request was 'http://" + hostname + ":" + port + pathPrefix + session + "'");
-    }
-
-    @Test
-    public void require_that_handler_gives_error_for_unsupported_methods() throws Exception {
-        testUnsupportedMethod(SessionHandlerTest.createTestRequest(pathPrefix, Method.POST, Cmd.PREPARED, 1L));
-        testUnsupportedMethod(SessionHandlerTest.createTestRequest(pathPrefix, Method.DELETE, Cmd.PREPARED, 1L));
-    }
-
-    protected void testUnsupportedMethod(HttpRequest request) throws Exception {
-        HttpResponse response = createHandler().handle(request);
-        HandlerTest.assertHttpStatusCodeErrorCodeAndMessage(response, METHOD_NOT_ALLOWED,
-                HttpErrorResponse.errorCodes.METHOD_NOT_ALLOWED,
-                "Method '" + request.getMethod().name() + "' is not supported");
-    }
-
-    @Test
-    public void require_that_activate_url_is_returned_on_success() throws Exception {
-        MockSession session = new MockSession(1, null);
-        localRepo.addSession(session);
-        HttpResponse response = createHandler().handle(SessionHandlerTest.createTestRequest(pathPrefix, Method.PUT, Cmd.PREPARED, 1L));
-        assertThat(session.getStatus(), is(Session.Status.PREPARE));
-        assertNotNull(response);
-        assertThat(response.getStatus(), is(OK));
-        assertResponseContains(response, "\"activate\":\"http://foo:1337" + pathPrefix + "1/active\",\"message\":\"Session 1" + preparedMessage);
-    }
-
-    @Test
-    public void require_debug() throws Exception {
-        HttpResponse response = createHandler().handle(SessionHandlerTest.createTestRequest(pathPrefix, Method.PUT, Cmd.PREPARED, 9999L, "?debug=true"));
-        assertThat(response.getStatus(), is(NOT_FOUND));
-        assertThat(SessionHandlerTest.getRenderedString(response), containsString("NotFoundException"));
-    }
-
-    @Test
-    public void require_verbose() throws Exception {
-        MockSession session = new MockSession(1, null);
-        session.doVerboseLogging = true;
-        localRepo.addSession(session);
-        HttpResponse response = createHandler().handle(SessionHandlerTest.createTestRequest(pathPrefix, Method.PUT, Cmd.PREPARED, 1L, "?verbose=true"));
-        assertThat(response.getStatus(), is(OK));
-        assertThat(SessionHandlerTest.getRenderedString(response), containsString("debuglog"));
-    }
-
-    /**
-     * A mock remote session repo based on contents of local repo
-     */
-    private RemoteSessionRepo fromLocalSessionRepo(LocalSessionRepo localRepo) {
-        RemoteSessionRepo remoteRepo = new RemoteSessionRepo();
-        PathProvider pathProvider = new PathProvider(Path.createRoot());
-        for (LocalSession ls : localRepo.listSessions()) {
-
-            zooKeeperClient = new MockSessionZKClient(curator, pathProvider.getSessionDirs().append(String.valueOf(ls.getSessionId())));
-            if (ls.getStatus()!=null) zooKeeperClient.writeStatus(ls.getStatus());
-            RemoteSession remSess = new RemoteSession(TenantName.from("default"), ls.getSessionId(),
-                    new TestComponentRegistry.Builder().curator(curator).build(),
-                    zooKeeperClient);
-            remoteRepo.addSession(remSess);
-        }
-        return remoteRepo;
-    }
-
-    @Test
-    public void require_get_response_activate_url_on_ok() throws Exception {
-        MockSession session = new MockSession(1, null);
-        localRepo.addSession(session);
-        SessionHandler sessHandler = createHandler(fromLocalSessionRepo(localRepo));
-        sessHandler.handle(SessionHandlerTest.createTestRequest(pathPrefix, Method.PUT, Cmd.PREPARED, 1L));
-        session.setStatus(Session.Status.PREPARE);
-        zooKeeperClient.writeStatus(Session.Status.PREPARE);
-        HttpResponse getResponse = sessHandler.handle(SessionHandlerTest.createTestRequest(pathPrefix, Method.GET, Cmd.PREPARED, 1L));
-        assertResponseContains(getResponse, "\"activate\":\"http://foo:1337" + pathPrefix + "1/active\",\"message\":\"Session 1" + preparedMessage);
-    }
-
-    @Test
-    public void require_get_response_error_on_not_prepared() throws Exception {
-        MockSession session = new MockSession(1, null);
-        localRepo.addSession(session);
-        SessionHandler sessHandler = createHandler(fromLocalSessionRepo(localRepo));
-        session.setStatus(Session.Status.NEW);
-        zooKeeperClient.writeStatus(Session.Status.NEW);
-        HttpResponse getResponse = sessHandler.handle(SessionHandlerTest.createTestRequest(pathPrefix, Method.GET, Cmd.PREPARED, 1L));
-        HandlerTest.assertHttpStatusCodeErrorCodeAndMessage(getResponse, BAD_REQUEST,
-                HttpErrorResponse.errorCodes.BAD_REQUEST,
-                "Session not prepared: 1");
-        session.setStatus(Session.Status.ACTIVATE);
-        zooKeeperClient.writeStatus(Session.Status.ACTIVATE);
-        getResponse = sessHandler.handle(SessionHandlerTest.createTestRequest(pathPrefix, Method.GET, Cmd.PREPARED, 1L));
-        HandlerTest.assertHttpStatusCodeErrorCodeAndMessage(getResponse, BAD_REQUEST,
-                HttpErrorResponse.errorCodes.BAD_REQUEST,
-                "Session is active: 1");
-    }
-
-    @Test
-    public void require_cannot_prepare_active_session() throws Exception {
-        MockSession session = new MockSession(1, null);
-        localRepo.addSession(session);
-        session.setStatus(Session.Status.ACTIVATE);
-        SessionHandler sessionHandler = createHandler(fromLocalSessionRepo(localRepo));
-        HttpResponse putResponse = sessionHandler.handle(SessionHandlerTest.createTestRequest(pathPrefix, Method.PUT, Cmd.PREPARED, 1L));
-        HandlerTest.assertHttpStatusCodeErrorCodeAndMessage(putResponse, BAD_REQUEST,
-                HttpErrorResponse.errorCodes.BAD_REQUEST,
-                "Session is active: 1");
-    }
-
-    @Test
-    public void require_get_response_error_when_session_id_does_not_exist() throws Exception {
-        MockSession session = new MockSession(1, null);
-        localRepo.addSession(session);
-        SessionHandler sessHandler = createHandler(fromLocalSessionRepo(localRepo));
-        HttpResponse getResponse = sessHandler.handle(SessionHandlerTest.createTestRequest(pathPrefix, Method.GET, Cmd.PREPARED, 9999L));
-        HandlerTest.assertHttpStatusCodeErrorCodeAndMessage(getResponse, NOT_FOUND,
-                HttpErrorResponse.errorCodes.NOT_FOUND,
-                "Session 9999 was not found");
-    }
-
-    protected static void assertResponse(HttpResponse response, String activateString) throws IOException {
-        // TODO Test when more logging is added
-        //assertThat(baos.toString(), startsWith("{\"log\":[{\"time\":"));
-        assertThat(SessionHandlerTest.getRenderedString(response), endsWith(activateString));
-    }
-
-    protected static void assertResponseContains(HttpResponse response, String string) throws IOException {
-        assertThat(SessionHandlerTest.getRenderedString(response), containsString(string));
-    }
-
-    protected static void assertResponseNotContains(HttpResponse response, String string) throws IOException {
-        assertThat(SessionHandlerTest.getRenderedString(response), not(containsString(string)));
-    }
-
-    public abstract SessionHandler createHandler();
-
-    public abstract SessionHandler createHandler(RemoteSessionRepo remoteSessionRepo);
-}
diff --git a/configserver/src/test/java/com/yahoo/vespa/config/server/http/v2/SessionPrepareHandlerTest.java b/configserver/src/test/java/com/yahoo/vespa/config/server/http/v2/SessionPrepareHandlerTest.java
index d88de383eef..632a1c66ac0 100644
--- a/configserver/src/test/java/com/yahoo/vespa/config/server/http/v2/SessionPrepareHandlerTest.java
+++ b/configserver/src/test/java/com/yahoo/vespa/config/server/http/v2/SessionPrepareHandlerTest.java
@@ -18,6 +18,8 @@ import com.yahoo.slime.JsonDecoder;
 import com.yahoo.slime.Slime;
 import com.yahoo.transaction.Transaction;
 import com.yahoo.vespa.config.server.ApplicationRepository;
+import com.yahoo.vespa.config.server.PathProvider;
+import com.yahoo.vespa.config.server.TestComponentRegistry;
 import com.yahoo.vespa.config.server.application.ApplicationConvergenceChecker;
 import com.yahoo.vespa.config.server.application.ApplicationSet;
 import com.yahoo.vespa.config.server.application.LogServerLogGrabber;
@@ -30,6 +32,7 @@ import com.yahoo.vespa.config.server.configchange.MockRestartAction;
 import com.yahoo.vespa.config.server.http.*;
 import com.yahoo.vespa.config.server.provision.HostProvisionerProvider;
 import com.yahoo.vespa.config.server.session.*;
+import com.yahoo.vespa.curator.Curator;
 import com.yahoo.vespa.curator.mock.MockCurator;
 import org.junit.Before;
 import org.junit.Test;
@@ -42,7 +45,12 @@ import java.util.List;
 import java.util.Optional;
 import java.util.concurrent.Executor;
 
-import static com.yahoo.jdisc.Response.Status.OK;
+import static com.yahoo.jdisc.Response.Status.*;
+import static com.yahoo.jdisc.Response.Status.BAD_REQUEST;
+import static com.yahoo.jdisc.Response.Status.NOT_FOUND;
+import static org.hamcrest.Matchers.containsString;
+import static org.hamcrest.Matchers.endsWith;
+import static org.hamcrest.Matchers.not;
 import static org.hamcrest.core.Is.is;
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertNotNull;
@@ -53,10 +61,17 @@ import static org.junit.Assert.assertThat;
  *
  * @since 5.1.14
  */
-public class SessionPrepareHandlerTest extends SessionPrepareHandlerTestBase {
+public class SessionPrepareHandlerTest extends SessionHandlerTest {
     private static final TenantName tenant = TenantName.from("test");
     private TestTenantBuilder builder;
 
+    protected Curator curator;
+    private SessionZooKeeperClient zooKeeperClient;
+    private LocalSessionRepo localRepo;
+
+    private String preparedMessage = " prepared.\"}";
+    private String tenantMessage = "";
+
     @Before
     public void setupRepo() throws Exception {
         TenantApplications applicationRepo = new MemoryTenantApplications();
@@ -69,6 +84,136 @@ public class SessionPrepareHandlerTest extends SessionPrepareHandlerTestBase {
     }
 
     @Test
+    public void require_error_when_session_id_does_not_exist() throws Exception {
+        // No session with this id exists
+        HttpResponse response = createHandler().handle(SessionHandlerTest.createTestRequest(pathPrefix, HttpRequest.Method.PUT, Cmd.PREPARED, 9999L));
+        HandlerTest.assertHttpStatusCodeErrorCodeAndMessage(response, NOT_FOUND, HttpErrorResponse.errorCodes.NOT_FOUND, "Session 9999 was not found");
+    }
+
+    @Test
+    public void require_error_when_session_id_not_a_number() throws Exception {
+        final String session = "notanumber/prepared";
+        HttpResponse response = createHandler().handle(SessionHandlerTest.createTestRequest(pathPrefix + session));
+        HandlerTest.assertHttpStatusCodeErrorCodeAndMessage(response, BAD_REQUEST,
+                                                            HttpErrorResponse.errorCodes.BAD_REQUEST,
+                                                            "Session id in request is not a number, request was 'http://" + hostname + ":" + port + pathPrefix + session + "'");
+    }
+
+    @Test
+    public void require_that_handler_gives_error_for_unsupported_methods() throws Exception {
+        testUnsupportedMethod(SessionHandlerTest.createTestRequest(pathPrefix, HttpRequest.Method.POST, Cmd.PREPARED, 1L));
+        testUnsupportedMethod(SessionHandlerTest.createTestRequest(pathPrefix, HttpRequest.Method.DELETE, Cmd.PREPARED, 1L));
+    }
+
+    private void testUnsupportedMethod(com.yahoo.container.jdisc.HttpRequest request) throws Exception {
+        HttpResponse response = createHandler().handle(request);
+        HandlerTest.assertHttpStatusCodeErrorCodeAndMessage(response, METHOD_NOT_ALLOWED,
+                                                            HttpErrorResponse.errorCodes.METHOD_NOT_ALLOWED,
+                                                            "Method '" + request.getMethod().name() + "' is not supported");
+    }
+
+    @Test
+    public void require_that_activate_url_is_returned_on_success() throws Exception {
+        MockSession session = new MockSession(1, null);
+        localRepo.addSession(session);
+        HttpResponse response = createHandler().handle(SessionHandlerTest.createTestRequest(pathPrefix, HttpRequest.Method.PUT, Cmd.PREPARED, 1L));
+        assertThat(session.getStatus(), is(Session.Status.PREPARE));
+        assertNotNull(response);
+        assertThat(response.getStatus(), is(OK));
+        assertResponseContains(response, "\"activate\":\"http://foo:1337" + pathPrefix + "1/active\",\"message\":\"Session 1" + preparedMessage);
+    }
+
+    @Test
+    public void require_debug() throws Exception {
+        HttpResponse response = createHandler().handle(SessionHandlerTest.createTestRequest(pathPrefix, HttpRequest.Method.PUT, Cmd.PREPARED, 9999L, "?debug=true"));
+        assertThat(response.getStatus(), is(NOT_FOUND));
+        assertThat(SessionHandlerTest.getRenderedString(response), containsString("NotFoundException"));
+    }
+
+    @Test
+    public void require_verbose() throws Exception {
+        MockSession session = new MockSession(1, null);
+        session.doVerboseLogging = true;
+        localRepo.addSession(session);
+        HttpResponse response = createHandler().handle(SessionHandlerTest.createTestRequest(pathPrefix, HttpRequest.Method.PUT, Cmd.PREPARED, 1L, "?verbose=true"));
+        assertThat(response.getStatus(), is(OK));
+        assertThat(SessionHandlerTest.getRenderedString(response), containsString("debuglog"));
+    }
+
+    /**
+     * A mock remote session repo based on contents of local repo
+     */
+    private RemoteSessionRepo fromLocalSessionRepo(LocalSessionRepo localRepo) {
+        RemoteSessionRepo remoteRepo = new RemoteSessionRepo();
+        PathProvider pathProvider = new PathProvider(Path.createRoot());
+        for (LocalSession ls : localRepo.listSessions()) {
+
+            zooKeeperClient = new MockSessionZKClient(curator, pathProvider.getSessionDirs().append(String.valueOf(ls.getSessionId())));
+            if (ls.getStatus()!=null) zooKeeperClient.writeStatus(ls.getStatus());
+            RemoteSession remSess = new RemoteSession(TenantName.from("default"), ls.getSessionId(),
+                                                      new TestComponentRegistry.Builder().curator(curator).build(),
+                                                      zooKeeperClient);
+            remoteRepo.addSession(remSess);
+        }
+        return remoteRepo;
+    }
+
+    @Test
+    public void require_get_response_activate_url_on_ok() throws Exception {
+        MockSession session = new MockSession(1, null);
+        localRepo.addSession(session);
+        SessionHandler sessHandler = createHandler(fromLocalSessionRepo(localRepo));
+        sessHandler.handle(SessionHandlerTest.createTestRequest(pathPrefix, HttpRequest.Method.PUT, Cmd.PREPARED, 1L));
+        session.setStatus(Session.Status.PREPARE);
+        zooKeeperClient.writeStatus(Session.Status.PREPARE);
+        HttpResponse getResponse = sessHandler.handle(SessionHandlerTest.createTestRequest(pathPrefix, HttpRequest.Method.GET, Cmd.PREPARED, 1L));
+        assertResponseContains(getResponse, "\"activate\":\"http://foo:1337" + pathPrefix + "1/active\",\"message\":\"Session 1" + preparedMessage);
+    }
+
+    @Test
+    public void require_get_response_error_on_not_prepared() throws Exception {
+        MockSession session = new MockSession(1, null);
+        localRepo.addSession(session);
+        SessionHandler sessHandler = createHandler(fromLocalSessionRepo(localRepo));
+        session.setStatus(Session.Status.NEW);
+        zooKeeperClient.writeStatus(Session.Status.NEW);
+        HttpResponse getResponse = sessHandler.handle(SessionHandlerTest.createTestRequest(pathPrefix, HttpRequest.Method.GET, Cmd.PREPARED, 1L));
+        HandlerTest.assertHttpStatusCodeErrorCodeAndMessage(getResponse, BAD_REQUEST,
+                                                            HttpErrorResponse.errorCodes.BAD_REQUEST,
+                                                            "Session not prepared: 1");
+        session.setStatus(Session.Status.ACTIVATE);
+        zooKeeperClient.writeStatus(Session.Status.ACTIVATE);
+        getResponse = sessHandler.handle(SessionHandlerTest.createTestRequest(pathPrefix, HttpRequest.Method.GET, Cmd.PREPARED, 1L));
+        HandlerTest.assertHttpStatusCodeErrorCodeAndMessage(getResponse, BAD_REQUEST,
+                                                            HttpErrorResponse.errorCodes.BAD_REQUEST,
+                                                            "Session is active: 1");
+    }
+
+    @Test
+    public void require_cannot_prepare_active_session() throws Exception {
+        MockSession session = new MockSession(1, null);
+        localRepo.addSession(session);
+        session.setStatus(Session.Status.ACTIVATE);
+        SessionHandler sessionHandler = createHandler(fromLocalSessionRepo(localRepo));
+        HttpResponse putResponse = sessionHandler.handle(SessionHandlerTest.createTestRequest(pathPrefix, HttpRequest.Method.PUT, Cmd.PREPARED, 1L));
+        HandlerTest.assertHttpStatusCodeErrorCodeAndMessage(putResponse, BAD_REQUEST,
+                                                            HttpErrorResponse.errorCodes.BAD_REQUEST,
+                                                            "Session is active: 1");
+    }
+
+    @Test
+    public void require_get_response_error_when_session_id_does_not_exist() throws Exception {
+        MockSession session = new MockSession(1, null);
+        localRepo.addSession(session);
+        SessionHandler sessHandler = createHandler(fromLocalSessionRepo(localRepo));
+        HttpResponse getResponse = sessHandler.handle(SessionHandlerTest.createTestRequest(pathPrefix, HttpRequest.Method.GET, Cmd.PREPARED, 9999L));
+        HandlerTest.assertHttpStatusCodeErrorCodeAndMessage(getResponse, NOT_FOUND,
+                                                            HttpErrorResponse.errorCodes.NOT_FOUND,
+                                                            "Session 9999 was not found");
+    }
+
+
+    @Test
     public void require_that_tenant_is_in_response() throws Exception {
         MockSession session = new MockSession(1, null);
         localRepo.addSession(session);
@@ -130,7 +275,7 @@ public class SessionPrepareHandlerTest extends SessionPrepareHandlerTestBase {
     @Test
     public void require_that_config_change_actions_are_logged_if_existing() throws Exception {
         List<ServiceInfo> services = Collections.singletonList(new ServiceInfo("serviceName", "serviceType", null,
-                ImmutableMap.of("clustername", "foo", "clustertype", "bar"), "configId", "hostName"));
+                                                                               ImmutableMap.of("clustername", "foo", "clustertype", "bar"), "configId", "hostName"));
         ConfigChangeActions actions = new ConfigChangeActions(Arrays.asList(
                 new MockRestartAction("change", services),
                 new MockRefeedAction("change-id", false, "other change", services, "test")));
@@ -165,13 +310,26 @@ public class SessionPrepareHandlerTest extends SessionPrepareHandlerTestBase {
         assertThat(data.get().field("message").asString(), is(message));
     }
 
-    @Override
-    public SessionHandler createHandler() {
+    private static void assertResponse(HttpResponse response, String activateString) throws IOException {
+        // TODO Test when more logging is added
+        //assertThat(baos.toString(), startsWith("{\"log\":[{\"time\":"));
+        assertThat(SessionHandlerTest.getRenderedString(response), endsWith(activateString));
+    }
+
+    private static void assertResponseContains(HttpResponse response, String string) throws IOException {
+        assertThat(SessionHandlerTest.getRenderedString(response), containsString(string));
+    }
+
+    private static void assertResponseNotContains(HttpResponse response, String string) throws IOException {
+        assertThat(SessionHandlerTest.getRenderedString(response), not(containsString(string)));
+    }
+
+
+    private SessionHandler createHandler() {
         return createHandler(addTestTenant());
     }
 
-    @Override
-    public SessionHandler createHandler(RemoteSessionRepo remoteSessionRepo) {
+    private SessionHandler createHandler(RemoteSessionRepo remoteSessionRepo) {
         return createHandler(addTenant(tenant, localRepo, remoteSessionRepo, new MockSessionFactory()));
     }
 
@@ -179,7 +337,7 @@ public class SessionPrepareHandlerTest extends SessionPrepareHandlerTestBase {
         return addTenant(tenant, localRepo, new RemoteSessionRepo(), new MockSessionFactory());
     }
 
-    static SessionHandler createHandler(TestTenantBuilder builder) {
+    private static SessionHandler createHandler(TestTenantBuilder builder) {
         final ConfigserverConfig configserverConfig = new ConfigserverConfig(new ConfigserverConfig.Builder());
         return new SessionPrepareHandler(new Executor() {
             @SuppressWarnings("NullableProblems")