Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* Copyright (c) 2014 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.system;
import com.google.common.collect.ImmutableList;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.spotify.helios.client.HeliosClient;
import com.spotify.helios.common.descriptors.Deployment;
import com.spotify.helios.common.descriptors.Goal;
import com.spotify.helios.common.descriptors.HostStatus;
import com.spotify.helios.common.descriptors.Job;
import com.spotify.helios.common.descriptors.JobId;
import com.spotify.helios.common.descriptors.JobStatus;
import com.spotify.helios.common.descriptors.TaskStatus.State;
import org.junit.Test;
import java.util.Map;
import java.util.concurrent.TimeUnit;
import static org.junit.Assert.assertEquals;
import static com.spotify.helios.common.descriptors.HostStatus.Status.UP;
import static java.util.concurrent.TimeUnit.SECONDS;
import static org.junit.Assert.assertTrue;
public class MultipleHostsTest extends SystemTestBase {
@Test
public void testHostStatuses() throws Exception {
final String aHost = testHost() + "a";
final String bHost = testHost() + "b";
startDefaultMaster();
startDefaultAgent(aHost);
startDefaultAgent(bHost);
awaitHostStatus(aHost, UP, LONG_WAIT_SECONDS, SECONDS);
awaitHostStatus(bHost, UP, LONG_WAIT_SECONDS, SECONDS);
final Map cliStatuses = new ObjectMapper().readValue(cli("hosts", "--json"),
new TypeReference>(){});
assertTrue("status must contain key for " + aHost, cliStatuses.containsKey(aHost));
assertTrue("status must contain key for " + bHost, cliStatuses.containsKey(bHost));
final HeliosClient client = defaultClient();
final Map clientStatuses = client.hostStatuses(
ImmutableList.of(aHost, bHost)).get();
assertTrue("status must contain key for " + aHost, clientStatuses.containsKey(aHost));
assertTrue("status must contain key for " + bHost, clientStatuses.containsKey(bHost));
}
@Test
public void testFilteringJobAndHostStatuses() throws Exception {
final String aHost = testHost() + "a";
final String bHost = testHost() + "b";
startDefaultMaster();
startDefaultAgent(aHost);
startDefaultAgent(bHost);
awaitHostStatus(aHost, UP, LONG_WAIT_SECONDS, SECONDS);
awaitHostStatus(bHost, UP, LONG_WAIT_SECONDS, SECONDS);
final HeliosClient client = defaultClient();
final Job job = Job.newBuilder()
.setName(testJobName + "I_WANT_THIS_ONE")
.setVersion(testJobVersion)
.setImage(BUSYBOX)
.setCommand(IDLE_COMMAND)
.setCreatingUser(TEST_USER)
.build();
final JobId jobId = job.getId();
client.createJob(job).get();
final Job job2 = Job.newBuilder()
.setName(testJobName)
.setVersion(testJobVersion)
.setImage(BUSYBOX)
.setCommand(IDLE_COMMAND)
.setCreatingUser(TEST_USER)
.build();
final JobId jobId2 = job2.getId();
client.createJob(job2).get();
final Deployment deployment = Deployment.of(jobId, Goal.START);
client.deploy(deployment, aHost);
client.deploy(deployment, bHost);
client.deploy(Deployment.of(jobId2, Goal.START), aHost);
awaitJobState(client, aHost, jobId, State.RUNNING, LONG_WAIT_SECONDS, TimeUnit.SECONDS);
awaitJobState(client, bHost, jobId, State.RUNNING, LONG_WAIT_SECONDS, TimeUnit.SECONDS);
awaitJobState(client, aHost, jobId2, State.RUNNING, LONG_WAIT_SECONDS, TimeUnit.SECONDS);
final Map cliStatuses = new ObjectMapper().readValue(
cli("status", "--job", "I_WANT_THIS_ONE", "--host", aHost, "--json"),
new TypeReference>(){});
assertEquals("status should only have one job", 1, cliStatuses.size());
assertTrue(cliStatuses.containsKey(jobId));
final JobStatus status = cliStatuses.get(jobId);
assertEquals("deployments should have only one item", 1, status.getDeployments().size());
assertTrue("should only have deployment info for aHost",
status.getDeployments().containsKey(aHost));
assertEquals("Task statuses should only have one item", 1, status.getTaskStatuses().size());
assertTrue("should only have status info for aHost",
status.getTaskStatuses().containsKey(aHost));
}
}
