Java tutorial
/* * Copyright 2019 ThoughtWorks, Inc. * * 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.thoughtworks.go.agent.launcher; import com.googlecode.junit.ext.checkers.OSChecker; import com.thoughtworks.cruise.agent.common.launcher.AgentLaunchDescriptor; import com.thoughtworks.cruise.agent.common.launcher.AgentLauncher; import com.thoughtworks.go.CurrentGoCDVersion; import com.thoughtworks.go.agent.ServerUrlGenerator; import com.thoughtworks.go.agent.common.AgentBootstrapperArgs; import com.thoughtworks.go.agent.testhelper.FakeGoServer; import org.apache.commons.io.FileUtils; import org.junit.After; import org.junit.Before; import org.junit.Rule; import org.junit.Test; import java.io.File; import java.io.IOException; import java.nio.charset.StandardCharsets; import java.util.ArrayList; import java.util.List; import java.util.Map; import java.util.concurrent.ConcurrentHashMap; import static com.thoughtworks.go.agent.common.util.Downloader.*; import static com.thoughtworks.go.agent.testhelper.FakeGoServer.TestResource.*; import static org.hamcrest.Matchers.is; import static org.hamcrest.Matchers.not; import static org.junit.Assert.*; import static org.mockito.Mockito.mock; import static org.mockito.Mockito.when; public class AgentLauncherImplTest { @Rule public FakeGoServer server = new FakeGoServer(); public static final OSChecker OS_CHECKER = new OSChecker(OSChecker.WINDOWS); @Before public void setUp() throws IOException { cleanup(); } @After public void tearDown() { cleanup(); } private void cleanup() { FileUtils.deleteQuietly(AGENT_PLUGINS_ZIP); FileUtils.deleteQuietly(AGENT_BINARY_JAR); FileUtils.deleteQuietly(AGENT_LAUNCHER_JAR); FileUtils.deleteQuietly(TFS_IMPL_JAR); new Lockfile(new File(AgentLauncherImpl.AGENT_BOOTSTRAPPER_LOCK_FILE)).delete(); } @Test public void shouldPassLauncherVersionToAgent() throws InterruptedException, IOException { final List<String> actualVersion = new ArrayList<>(); final AgentLauncher launcher = new AgentLauncherImpl(new AgentLauncherImpl.AgentProcessParentRunner() { public int run(String launcherVersion, String launcherMd5, ServerUrlGenerator urlConstructor, Map<String, String> environmentVariables, Map context) { actualVersion.add(launcherVersion); return 0; } }); TEST_AGENT_LAUNCHER.copyTo(AGENT_LAUNCHER_JAR); launcher.launch(launchDescriptor()); assertThat(actualVersion.size(), is(1)); assertThat(actualVersion.get(0), is(CurrentGoCDVersion.getInstance().fullVersion())); } @Test public void shouldNotThrowException_instedReturnAppropriateErrorCode_whenSomethingGoesWrongInLaunch() { AgentLaunchDescriptor launchDesc = mock(AgentLaunchDescriptor.class); when((String) launchDesc.context().get(AgentBootstrapperArgs.SERVER_URL)) .thenThrow(new RuntimeException("Ouch!")); try { assertThat(new AgentLauncherImpl().launch(launchDesc), is(-273)); } catch (Exception e) { fail("should not have blown up, because it directly interfaces with bootstrapper"); } } private AgentLaunchDescriptor launchDescriptor() { AgentLaunchDescriptor launchDescriptor = mock(AgentLaunchDescriptor.class); Map contextMap = new ConcurrentHashMap(); contextMap.put(AgentBootstrapperArgs.SERVER_URL, "http://localhost:" + server.getPort() + "/go"); contextMap.put(AgentBootstrapperArgs.SSL_VERIFICATION_MODE, "NONE"); when(launchDescriptor.context()).thenReturn(contextMap); return launchDescriptor; } @Test public void shouldDownloadLauncherJarIfLocalCopyIsStale() throws IOException { //because new invocation will take care of pulling latest agent down, and will then operate on it with the latest launcher -jj File staleJar = randomFile(AGENT_LAUNCHER_JAR); long original = staleJar.length(); new AgentLauncherImpl().launch(launchDescriptor()); assertThat(staleJar.length(), not(original)); } @Test public void shouldDownload_AgentJar_IfTheCurrentJarIsStale() throws Exception { if (!OS_CHECKER.satisfy()) { TEST_AGENT_LAUNCHER.copyTo(AGENT_LAUNCHER_JAR); File staleJar = randomFile(AGENT_BINARY_JAR); long original = staleJar.length(); new AgentLauncherImpl().launch(launchDescriptor()); assertThat(staleJar.length(), not(original)); } } @Test public void should_NOT_Download_AgentJar_IfTheCurrentJarIsUpToDate() throws Exception { if (!OS_CHECKER.satisfy()) { TEST_AGENT_LAUNCHER.copyTo(AGENT_LAUNCHER_JAR); TEST_AGENT.copyTo(AGENT_BINARY_JAR); assertTrue(AGENT_BINARY_JAR.setLastModified(0)); new AgentLauncherImpl().launch(launchDescriptor()); assertThat(AGENT_BINARY_JAR.lastModified(), is(0L)); } } @Test public void should_NOT_Download_TfsImplJar_IfTheCurrentJarIsUpToDate() throws Exception { if (!OS_CHECKER.satisfy()) { TEST_AGENT_LAUNCHER.copyTo(AGENT_LAUNCHER_JAR); TEST_AGENT.copyTo(AGENT_BINARY_JAR); TEST_TFS_IMPL.copyTo(TFS_IMPL_JAR); assertTrue(TFS_IMPL_JAR.setLastModified(0)); new AgentLauncherImpl().launch(launchDescriptor()); assertThat(TFS_IMPL_JAR.lastModified(), is(0L)); } } @Test public void shouldDownloadLauncherJarIfLocalCopyIsStale_butShouldReturnWithoutDownloadingOrLaunchingAgent() throws Exception { File launcher = randomFile(AGENT_LAUNCHER_JAR); long original = launcher.length(); File agentFile = randomFile(AGENT_BINARY_JAR); long originalAgentLength = agentFile.length(); new AgentLauncherImpl().launch(launchDescriptor()); assertThat(launcher.length(), not(original)); assertThat(agentFile.length(), is(originalAgentLength)); } private File randomFile(final File pathname) throws IOException { FileUtils.write(pathname, "some rubbish", StandardCharsets.UTF_8); return pathname; } }