ComponentTest.java

package com.sequenceiq.cloudbreak.orchestrator.yarn.model;

import static org.junit.Assert.assertTrue;
import static org.junit.Assert.assertEquals;

import java.util.ArrayList;
import java.util.List;

import org.junit.Test;

import com.sequenceiq.cloudbreak.orchestrator.yarn.model.core.Artifact;
import com.sequenceiq.cloudbreak.orchestrator.yarn.model.core.YarnComponent;
import com.sequenceiq.cloudbreak.orchestrator.yarn.model.core.Dependency;
import com.sequenceiq.cloudbreak.orchestrator.yarn.model.core.Resource;

public class ComponentTest {

    private static final String NAME = "testComp";

    private static final String LAUNCH_COMMAND = "/bin/foo";

    private static final int NUM_OF_CONTAINERS = 1;

    @Test
    public void testResource() throws Exception {
        Resource resource = new Resource();
        YarnComponent component = new YarnComponent();
        component.setResource(resource);
        assertTrue(component.getResource() != null);
    }

    @Test
    public void testArtifact() throws Exception {
        Artifact artifact = new Artifact();
        YarnComponent component = new YarnComponent();
        component.setArtifact(artifact);
        assertTrue(component.getArtifact() != null);
    }

    @Test
    public void testLaunchCommand() throws Exception {
        YarnComponent component = new YarnComponent();
        component.setLaunchCommand(LAUNCH_COMMAND);
        assertEquals(LAUNCH_COMMAND, component.getLaunchCommand());
    }

    @Test
    public void testNumberOfContainers() throws Exception {
        YarnComponent component = new YarnComponent();
        component.setNumberOfContainers(NUM_OF_CONTAINERS);
        assertEquals(NUM_OF_CONTAINERS, component.getNumberOfContainers());
    }

    @Test
    public void testName() throws Exception {
        YarnComponent component = new YarnComponent();
        component.setName(NAME);
        assertEquals(NAME, component.getName());
    }

    @Test
    public void testDependencies() throws Exception {
        Dependency dependency = new Dependency();
        YarnComponent component = new YarnComponent();
        List<Dependency> dependencies = new ArrayList<>();
        dependencies.add(dependency);
        component.setDependencies(dependencies);
        assertEquals(1, component.getDependencies().size());
    }
}