/*-
 * -\-\-
 * Helios Services
 * --
 * Copyright (C) 2016 Spotify AB
 * --
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * -/-/-
 */

package com.spotify.helios.agent;

import static org.hamcrest.Matchers.is;
import static org.junit.Assert.assertThat;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import com.codahale.metrics.health.HealthCheck;
import com.spotify.docker.client.DockerClient;
import com.spotify.docker.client.exceptions.DockerRequestException;

import java.net.URI;

import org.junit.Before;
import org.junit.Test;

public class DockerDaemonHealthCheckerTest {

  private DockerClient dockerClient;

  @Before
  public void setUp() throws Exception {
    dockerClient = mock(DockerClient.class);
  }

  @Test
  public void testHealthy() throws Exception {
    final DockerDaemonHealthChecker checker = new DockerDaemonHealthChecker(dockerClient);
    final HealthCheck.Result result = checker.check();
    assertThat(result.isHealthy(), is(true));
  }

  @Test
  public void testUnhealthy() throws Exception {
    when(dockerClient.ping()).thenThrow(new DockerRequestException("GET", new URI("/ping")));
    final DockerDaemonHealthChecker checker = new DockerDaemonHealthChecker(dockerClient);
    final HealthCheck.Result result = checker.check();
    assertThat(result.isHealthy(), is(false));
  }
}

@@ -1031,33 +1031,8 @@
 I;%0A%0A
-import org.junit.Before;%0A
 impo
@@ -1111,195 +1111,77 @@
 ate 
-DockerClient dockerClient;%0A%0A  @Before%0A  public void setUp() throws Exception %7B%0A    dockerClient = mock(DockerClient.class);%0A  %7D%0A%0A  @Test%0A  public void testHealthy() throws Exception %7B%0A   
+final DockerClient dockerClient = mock(DockerClient.class);%0A  private
  fin
@@ -1201,39 +1201,35 @@
 onHealthChecker 
-checker
+sut
  = new DockerDae
@@ -1251,32 +1251,88 @@
 r(dockerClient);
+%0A%0A  @Test%0A  public void testHealthy() throws Exception %7B
 %0A    final Healt
@@ -1346,39 +1346,35 @@
 Result result = 
-checker
+sut
 .check();%0A    as
@@ -1573,99 +1573,8 @@
 ));%0A
-    final DockerDaemonHealthChecker checker = new DockerDaemonHealthChecker(dockerClient);%0A
     
@@ -1611,15 +1611,11 @@
 t = 
-checker
+sut
 .che

Clean up DockerDaemonHealthCheckerTest a bit