slider-core/src/test/java/org/apache/slider/providers/agent/TestAgentClientProvider.java

/*
 * 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.providers.agent;

import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.fs.FileSystem;
import org.apache.slider.common.tools.SliderFileSystem;
import org.apache.slider.core.conf.AggregateConf;
import org.apache.slider.core.exceptions.BadConfigException;
import org.apache.slider.tools.TestUtility;
import org.junit.Assert;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.TemporaryFolder;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.Set;

/**
 *
 */
public class TestAgentClientProvider {
  protected static final Logger log =
      LoggerFactory.getLogger(TestAgentClientProvider.class);
  @Rule
  public TemporaryFolder folder = new TemporaryFolder();

  @Test
  public void testGetApplicationTags() throws Exception {
    Configuration configuration = new Configuration();
    FileSystem fs = FileSystem.getLocal(configuration);
    SliderFileSystem sliderFileSystem = new SliderFileSystem(fs, configuration);

    AgentClientProvider provider = new AgentClientProvider(null);
    String zipFileName = TestUtility.createAppPackage(
        folder,
        "testpkg",
        "test.zip",
        "target/test-classes/org/apache/slider/common/tools/test");
    Set<String> tags = provider.getApplicationTags(sliderFileSystem, zipFileName);
    assert tags != null;
    assert !tags.isEmpty();
    assert tags.contains("Name: STORM");
    assert tags.contains("Description: Apache Hadoop Stream processing framework");
    assert tags.contains("Version: 0.9.1.2.1");

  }

  @Test
  public void testValidateInstanceDefinition() throws Exception {
    AgentClientProvider provider = new AgentClientProvider(null);
    AggregateConf instanceDefinition = new AggregateConf();

    try {
      provider.validateInstanceDefinition(instanceDefinition, null);
      Assert.assertFalse("Should fail with BadConfigException", true);
    } catch (BadConfigException e) {
      log.info(e.toString());
      Assert.assertTrue(e.getMessage().contains("Application definition must be provided"));
    }
  }
}