com.spotify.helios.system

Class SystemTestBase

java.lang.Object

com.spotify.helios.system.SystemTestBase

Direct Known Subclasses:

AgentReportingTest, AgentRestartTest, AgentZooKeeperDownTolerationTest, BindVolumeTest, CliDeploymentTest, CliJobCreationTest, CliMasterListTest, ClusterDeploymentTest, ConfigFileJobCreationTest, DeploymentTest, DeregisterTest, DnsServerTest, EnvironmentVariableTest, FlappingTest, IdMismatchJobCreateTest, ImageMissingTest, JobExpirationTest, JobHistoryTest, JobListTest, JobWatchExactTest, JobWatchTest, MasterRespondsWithNoZKTest, MultipleHostsTest, MultipleJobsTest, MultiplePortJobTest, NamespaceTest, PortCollisionJobTest, PredefinedPortImageDeploymentTest, QueryFailureTest, ReapingTest, ResourcesTest, ServiceRegistrationTestBase, SyslogRedirectionTest, TerminationTest, TokenTest, UndeployFilteringTest, UndeployRaceTest, VersionCommandTest, VersionResponseFilterTest, VolumeTest, ZooKeeperBadNodeTest, ZooKeeperClusterIdTest, ZooKeeperHeliosFailoverTest, ZooKeeperNamespacingTest, ZooKeeperRestoreTest

public abstract class SystemTestBase
extends Object

Field Summary

Fields

Modifier and Type	Field and Description
static String	ALPINE
static String	BUSYBOX
static DockerHost	DOCKER_HOST
org.junit.rules.ExpectedException	exception
static List<String>	IDLE_COMMAND
static int	INTERNAL_PORT
static int	LONG_WAIT_SECONDS
static String	NGINX
org.junit.rules.TemporaryFolder	temporaryFolder
TemporaryPorts	temporaryPorts
static String	TEST_HOST
static String	TEST_MASTER
static String	TEST_USER
String	testJobName
String	testJobNameAndVersion
String	testJobVersion
String	testTag
static String	UHTTPD
static int	WAIT_TIMEOUT_SECONDS
org.junit.rules.TestRule	watcher
protected String	zkClusterId
protected static String	zooKeeperNamespace

Constructor Summary

Constructors

Constructor and Description
SystemTestBase()

Method Summary

Methods

Modifier and Type	Method and Description
protected void	assertJobEquals(Job expected, Job actual)
protected void	awaitHostRegistered(HeliosClient client, String host, int timeout, TimeUnit timeUnit)
protected void	awaitHostRegistered(String name, long timeout, TimeUnit timeUnit)
protected HostStatus	awaitHostStatus(HeliosClient client, String host, HostStatus.Status status, int timeout, TimeUnit timeUnit)
protected HostStatus	awaitHostStatus(String name, HostStatus.Status status, int timeout, TimeUnit timeUnit)
protected TaskStatus	awaitJobState(HeliosClient client, String host, JobId jobId, TaskStatus.State state, int timeout, TimeUnit timeunit)
protected TaskStatus	awaitJobThrottle(HeliosClient client, String host, JobId jobId, ThrottleState throttled, int timeout, TimeUnit timeunit)
protected void	awaitTaskGone(HeliosClient client, String host, JobId jobId, long timeout, TimeUnit timeunit)
protected TaskStatus	awaitTaskState(JobId jobId, String host, TaskStatus.State state)
void	baseSetup()
void	baseTeardown()
protected String	cli(String command, List<String> args)
protected String	cli(String command, Object... args)
protected String	cli(String command, String... args)
protected HeliosClient	client(String user, String endpoint)
protected <T> T	cliJson(Class<T> klass, String command, List<String> args)
protected <T> T	cliJson(Class<T> klass, String command, String... args)
protected JobId	createJob(Job job)
protected JobId	createJob(String name, String version, String image, List<String> command)
protected JobId	createJob(String name, String version, String image, List<String> command, Date expires)
protected JobId	createJob(String name, String version, String image, List<String> command, com.google.common.collect.ImmutableMap<String,String> env)
protected JobId	createJob(String name, String version, String image, List<String> command, Map<String,String> env, Map<String,PortMapping> ports)
protected JobId	createJob(String name, String version, String image, List<String> command, Map<String,String> env, Map<String,PortMapping> ports, Map<ServiceEndpoint,ServicePorts> registration)
protected JobId	createJob(String name, String version, String image, List<String> command, Map<String,String> env, Map<String,PortMapping> ports, Map<ServiceEndpoint,ServicePorts> registration, Integer gracePeriod, Map<String,String> volumes)
protected JobId	createJob(String name, String version, String image, List<String> command, Map<String,String> env, Map<String,PortMapping> ports, Map<ServiceEndpoint,ServicePorts> registration, Integer gracePeriod, Map<String,String> volumes, Date expires)
protected HeliosClient	defaultClient()
protected void	deployJob(JobId jobId, String host)
protected String	deregisterHost(String host)
com.google.common.collect.Range<Integer>	dockerPortRange()
void	dockerSetup()
protected List<String>	flatten(Iterable<?> values)
protected List<String>	flatten(Object... values)
protected com.spotify.docker.client.DockerClient	getNewDockerClient()
protected <T> T	getOrNull(com.google.common.util.concurrent.ListenableFuture<T> future)
protected boolean	isIntegration()
protected List<com.spotify.docker.client.messages.Container>	listContainers(com.spotify.docker.client.DockerClient dockerClient, String needle)
protected ByteArrayOutputStream	main(Collection<String> args)
protected ByteArrayOutputStream	main(String... args)
protected int	masterAdminPort()
protected String	masterEndpoint()
protected String	masterName()
protected int	masterPort()
protected String	readLogFully(com.sun.jersey.api.client.ClientResponse logs)
protected static void	removeContainer(com.spotify.docker.client.DockerClient dockerClient, String containerId)
protected List<String>	setupDefaultMaster(String... args)
protected AgentMain	startAgent(String... args)
protected AgentMain	startDefaultAgent(String host, String... args)
protected void	startDefaultMaster(String... args)
protected void	startDefaultMasterDontWaitForZK(CuratorClientFactory curatorClientFactory, String... args)
protected String	startJob(JobId jobId, String host)
protected MasterMain	startMaster(String... args)
static void	staticSetup()
protected String	stopJob(JobId jobId, String host)
protected void	tearDownJobs()
protected TemporaryPorts	temporaryPorts()
protected String	testHost()
protected void	undeployJob(JobId jobId, String host)
protected void	waitForMasterToConnectToZK()
protected ZooKeeperTestManager	zk()
protected ZooKeeperTestManager	zooKeeperTestManager()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

WAIT_TIMEOUT_SECONDS

public static final int WAIT_TIMEOUT_SECONDS

See Also:

Constant Field Values

LONG_WAIT_SECONDS

public static final int LONG_WAIT_SECONDS

See Also:

Constant Field Values

INTERNAL_PORT

public static final int INTERNAL_PORT

See Also:

Constant Field Values

BUSYBOX

public static final String BUSYBOX

See Also:

Constant Field Values

NGINX

public static final String NGINX

See Also:

Constant Field Values

UHTTPD

public static final String UHTTPD

See Also:

Constant Field Values

ALPINE

public static final String ALPINE

See Also:

Constant Field Values

IDLE_COMMAND

public static final List<String> IDLE_COMMAND

testTag

public final String testTag

testJobName

public final String testJobName

testJobVersion

public final String testJobVersion

testJobNameAndVersion

public final String testJobNameAndVersion

DOCKER_HOST

public static final DockerHost DOCKER_HOST

TEST_USER

public static final String TEST_USER

See Also:

Constant Field Values

TEST_HOST

public static final String TEST_HOST

See Also:

Constant Field Values

TEST_MASTER

public static final String TEST_MASTER

See Also:

Constant Field Values

temporaryPorts

public final TemporaryPorts temporaryPorts

temporaryFolder

public final org.junit.rules.TemporaryFolder temporaryFolder

exception

public final org.junit.rules.ExpectedException exception

watcher

public final org.junit.rules.TestRule watcher

zooKeeperNamespace

protected static String zooKeeperNamespace

zkClusterId

protected final String zkClusterId

Constructor Detail

SystemTestBase

public SystemTestBase()

Method Detail

staticSetup

public static void staticSetup()

baseSetup

public void baseSetup()
               throws Exception

Throws:

Exception

dockerSetup

public void dockerSetup()
                 throws Exception

Throws:

Exception

getNewDockerClient

protected com.spotify.docker.client.DockerClient getNewDockerClient()
                                                             throws Exception

Throws:

Exception

zooKeeperTestManager

protected ZooKeeperTestManager zooKeeperTestManager()

baseTeardown

public void baseTeardown()
                  throws Exception

Throws:

Exception

tearDownJobs

protected void tearDownJobs()
                     throws InterruptedException,
                            ExecutionException

Throws:

InterruptedException

ExecutionException

isIntegration

protected boolean isIntegration()

temporaryPorts

protected TemporaryPorts temporaryPorts()

zk

protected ZooKeeperTestManager zk()

masterEndpoint

protected String masterEndpoint()

masterName

protected String masterName()
                     throws InterruptedException,
                            ExecutionException

Throws:

InterruptedException

ExecutionException

defaultClient

protected HeliosClient defaultClient()

client

protected HeliosClient client(String user,
                  String endpoint)

masterPort

protected int masterPort()

masterAdminPort

protected int masterAdminPort()

dockerPortRange

public com.google.common.collect.Range<Integer> dockerPortRange()

testHost

protected String testHost()
                   throws InterruptedException,
                          ExecutionException

Throws:

InterruptedException

ExecutionException

setupDefaultMaster

protected List<String> setupDefaultMaster(String... args)
                                   throws Exception

Throws:

Exception

startDefaultMaster

protected void startDefaultMaster(String... args)
                           throws Exception

Throws:

Exception

waitForMasterToConnectToZK

protected void waitForMasterToConnectToZK()
                                   throws Exception

Throws:

Exception

startDefaultMasterDontWaitForZK

protected void startDefaultMasterDontWaitForZK(CuratorClientFactory curatorClientFactory,
                                   String... args)
                                        throws Exception

Throws:

Exception

startDefaultAgent

protected AgentMain startDefaultAgent(String host,
                          String... args)
                               throws Exception

Throws:

Exception

startMaster

protected MasterMain startMaster(String... args)
                          throws Exception

Throws:

Exception

startAgent

protected AgentMain startAgent(String... args)
                        throws Exception

Throws:

Exception

createJob

protected JobId createJob(String name,
              String version,
              String image,
              List<String> command)
                   throws Exception

Throws:

Exception

createJob

protected JobId createJob(String name,
              String version,
              String image,
              List<String> command,
              Date expires)
                   throws Exception

Throws:

Exception

createJob

protected JobId createJob(String name,
              String version,
              String image,
              List<String> command,
              com.google.common.collect.ImmutableMap<String,String> env)
                   throws Exception

Throws:

Exception

createJob

protected JobId createJob(String name,
              String version,
              String image,
              List<String> command,
              Map<String,String> env,
              Map<String,PortMapping> ports)
                   throws Exception

Throws:

Exception

createJob

protected JobId createJob(String name,
              String version,
              String image,
              List<String> command,
              Map<String,String> env,
              Map<String,PortMapping> ports,
              Map<ServiceEndpoint,ServicePorts> registration)
                   throws Exception

Throws:

Exception

createJob

protected JobId createJob(String name,
              String version,
              String image,
              List<String> command,
              Map<String,String> env,
              Map<String,PortMapping> ports,
              Map<ServiceEndpoint,ServicePorts> registration,
              Integer gracePeriod,
              Map<String,String> volumes)
                   throws Exception

Throws:

Exception

createJob

protected JobId createJob(String name,
              String version,
              String image,
              List<String> command,
              Map<String,String> env,
              Map<String,PortMapping> ports,
              Map<ServiceEndpoint,ServicePorts> registration,
              Integer gracePeriod,
              Map<String,String> volumes,
              Date expires)
                   throws Exception

Throws:

Exception

createJob

protected JobId createJob(Job job)
                   throws Exception

Throws:

Exception

deployJob

protected void deployJob(JobId jobId,
             String host)
                  throws Exception

Throws:

Exception

undeployJob

protected void undeployJob(JobId jobId,
               String host)
                    throws Exception

Throws:

Exception

startJob

protected String startJob(JobId jobId,
              String host)
                   throws Exception

Throws:

Exception

stopJob

protected String stopJob(JobId jobId,
             String host)
                  throws Exception

Throws:

Exception

deregisterHost

protected String deregisterHost(String host)
                         throws Exception

Throws:

Exception

cli

protected String cli(String command,
         Object... args)
              throws Exception

Throws:

Exception

cli

protected String cli(String command,
         String... args)
              throws Exception

Throws:

Exception

cli

protected String cli(String command,
         List<String> args)
              throws Exception

Throws:

Exception

cliJson

protected <T> T cliJson(Class<T> klass,
            String command,
            String... args)
             throws Exception

Throws:

Exception

cliJson

protected <T> T cliJson(Class<T> klass,
            String command,
            List<String> args)
             throws Exception

Throws:

Exception

main

protected ByteArrayOutputStream main(String... args)
                              throws Exception

Throws:

Exception

main

protected ByteArrayOutputStream main(Collection<String> args)
                              throws Exception

Throws:

Exception

awaitHostRegistered

protected void awaitHostRegistered(String name,
                       long timeout,
                       TimeUnit timeUnit)
                            throws Exception

Throws:

Exception

awaitHostStatus

protected HostStatus awaitHostStatus(String name,
                         HostStatus.Status status,
                         int timeout,
                         TimeUnit timeUnit)
                              throws Exception

Throws:

Exception

awaitJobState

protected TaskStatus awaitJobState(HeliosClient client,
                       String host,
                       JobId jobId,
                       TaskStatus.State state,
                       int timeout,
                       TimeUnit timeunit)
                            throws Exception

Throws:

Exception

awaitJobThrottle

protected TaskStatus awaitJobThrottle(HeliosClient client,
                          String host,
                          JobId jobId,
                          ThrottleState throttled,
                          int timeout,
                          TimeUnit timeunit)
                               throws Exception

Throws:

Exception

awaitHostRegistered

protected void awaitHostRegistered(HeliosClient client,
                       String host,
                       int timeout,
                       TimeUnit timeUnit)
                            throws Exception

Throws:

Exception

awaitHostStatus

protected HostStatus awaitHostStatus(HeliosClient client,
                         String host,
                         HostStatus.Status status,
                         int timeout,
                         TimeUnit timeUnit)
                              throws Exception

Throws:

Exception

awaitTaskState

protected TaskStatus awaitTaskState(JobId jobId,
                        String host,
                        TaskStatus.State state)
                             throws Exception

Throws:

Exception

awaitTaskGone

protected void awaitTaskGone(HeliosClient client,
                 String host,
                 JobId jobId,
                 long timeout,
                 TimeUnit timeunit)
                      throws Exception

Throws:

Exception

getOrNull

protected <T> T getOrNull(com.google.common.util.concurrent.ListenableFuture<T> future)
               throws ExecutionException,
                      InterruptedException

Throws:

ExecutionException

InterruptedException

readLogFully

protected String readLogFully(com.sun.jersey.api.client.ClientResponse logs)
                       throws IOException

Throws:

IOException

removeContainer

protected static void removeContainer(com.spotify.docker.client.DockerClient dockerClient,
                   String containerId)
                               throws Exception

Throws:

Exception

listContainers

protected List<com.spotify.docker.client.messages.Container> listContainers(com.spotify.docker.client.DockerClient dockerClient,
                                                                String needle)
                                                                     throws com.spotify.docker.client.DockerException,
                                                                            InterruptedException

Throws:

com.spotify.docker.client.DockerException

InterruptedException

flatten

protected List<String> flatten(Object... values)

flatten

protected List<String> flatten(Iterable<?> values)

assertJobEquals

protected void assertJobEquals(Job expected,
                   Job actual)