blob: daf091c46300dbef448d008cb428ff952a844fd0 [file] [log] [blame]
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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 org.apache.slider.funtest.lifecycle
import groovy.util.logging.Slf4j
import org.apache.hadoop.fs.Path
import org.apache.slider.common.SliderExitCodes
import org.apache.slider.common.params.Arguments
import org.apache.slider.common.params.SliderActions
import org.apache.slider.funtest.framework.CommandTestBase
import org.apache.slider.funtest.framework.FuntestProperties
import org.apache.slider.funtest.framework.SliderShell
import org.junit.Before
import org.junit.BeforeClass
@Slf4j
class AgentCommandTestBase extends CommandTestBase
implements FuntestProperties, Arguments, SliderExitCodes, SliderActions {
public static final boolean AGENTTESTS_ENABLED
protected static String APP_RESOURCE = "../slider-core/src/test/app_packages/test_command_log/resources.json"
protected static String APP_TEMPLATE = "../slider-core/src/test/app_packages/test_command_log/appConfig.json"
protected static String APP_PKG = "../slider-core/src/test/app_packages/test_command_log/cmd_log_app_pkg.zip"
protected static String AGENT_CONF = "../slider-agent/conf/agent.ini"
protected static final File LOCAL_SLIDER_AGENT_TARGZ
protected static final File LOCAL_APP_PKZ
protected static final File LOCAL_AGENT_CONF
protected static Path agentTarballPath;
protected static Path appPkgPath;
protected static Path agtIniPath;
static {
AGENTTESTS_ENABLED = SLIDER_CONFIG.getBoolean(KEY_TEST_AGENT_ENABLED, false)
LOCAL_SLIDER_AGENT_TARGZ = new File(
SLIDER_BIN_DIRECTORY,
AGENT_SLIDER_GZ).canonicalFile
LOCAL_APP_PKZ = new File(APP_PKG).canonicalFile
LOCAL_AGENT_CONF = new File(AGENT_CONF).canonicalFile
}
@BeforeClass
public static void setupAgent() {
assumeAgentTestsEnabled()
// Upload the agent tarball
assume(LOCAL_SLIDER_AGENT_TARGZ.exists(), "Slider agent not found at $LOCAL_SLIDER_AGENT_TARGZ")
agentTarballPath = new Path(clusterFS.homeDirectory, "/slider/agent/slider-agent.tar.gz")
Path localTarball = new Path(LOCAL_SLIDER_AGENT_TARGZ.toURI());
clusterFS.copyFromLocalFile(false, true, localTarball, agentTarballPath)
// Upload the app pkg
assume(LOCAL_APP_PKZ.exists(), "App pkg not found at $LOCAL_APP_PKZ")
appPkgPath = new Path(clusterFS.homeDirectory, "/slider/cmd_log_app_pkg.zip")
Path localAppPkg = new Path(LOCAL_APP_PKZ.toURI());
clusterFS.copyFromLocalFile(false, true, localAppPkg, appPkgPath)
// Upload the agent.ini
assume(LOCAL_AGENT_CONF.exists(), "Agent config not found at $LOCAL_AGENT_CONF")
agtIniPath = new Path(clusterFS.homeDirectory, "/slider/agent/conf/agent.ini")
Path localAgtIni = new Path(LOCAL_AGENT_CONF.toURI());
clusterFS.copyFromLocalFile(false, true, localAgtIni, agtIniPath)
}
public static void assumeAgentTestsEnabled() {
assumeFunctionalTestsEnabled()
assume(AGENTTESTS_ENABLED, "Agent tests disabled")
}
public static void logShell(SliderShell shell) {
for (String str in shell.out) {
log.info str
}
}
public static void assertComponentCount(String component, int count, SliderShell shell) {
log.info("Asserting component count.")
String entry = findLineEntry(shell, ["instances", component] as String[])
log.info(entry)
assert entry != null
int instanceCount = 0
int index = entry.indexOf("container_")
while (index != -1) {
instanceCount++;
index = entry.indexOf("container_", index + 1)
}
assert instanceCount == count, 'Instance count for component did not match expected. Parsed: ' + entry
}
public static String findLineEntry(SliderShell shell, String[] locators) {
int index = 0;
for (String str in shell.out) {
if (str.contains("\"" + locators[index] + "\"")) {
if (locators.size() == index + 1) {
return str;
} else {
index++;
}
}
}
return null;
}
public static boolean isAppRunning(String text, SliderShell shell) {
boolean exists = false
for (String str in shell.out) {
if (str.contains(text)) {
exists = true
}
}
return exists
}
}