alluxio-master-2.x/job/common/src/test/java/alluxio/job/workflow/composite/CompositeConfigTest.java
/*
 * The Alluxio Open Foundation licenses this work under the Apache License, version 2.0
 * (the "License"). You may not use this work except in compliance with the License, which is
 * available at www.apache.org/licenses/LICENSE-2.0
 *
 * This software is distributed on an "AS IS" basis, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
 * either express or implied, as more fully set forth in the License.
 *
 * See the NOTICE file distributed with this work for information regarding copyright ownership.
 */

package alluxio.job.workflow.composite;

import static junit.framework.TestCase.assertNotNull;
import static org.junit.Assert.assertEquals;

import alluxio.job.JobConfig;
import alluxio.job.plan.load.LoadConfig;
import alluxio.job.plan.transform.CompactConfig;
import alluxio.job.plan.transform.PartitionInfo;
import alluxio.job.util.SerializationUtils;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.google.common.collect.Lists;
import org.apache.commons.io.FileUtils;
import org.junit.Test;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;

/**
 * Tests {@link CompositeConfig}.
 */
public final class CompositeConfigTest {
  private static final CompositeConfig CONFIG;

  static {
    PartitionInfo pInfo = new PartitionInfo("serde", "inputformat", new HashMap<>(),
        new HashMap<>(), new ArrayList<>());
    ArrayList<JobConfig> jobs = new ArrayList<>();
    jobs.add(new CompositeConfig(new ArrayList<>(), true));
    jobs.add(new CompositeConfig(new ArrayList<>(), false));
    jobs.add(new CompositeConfig(Lists.newArrayList(new LoadConfig("/", 1, Collections.EMPTY_SET,
        Collections.EMPTY_SET, Collections.EMPTY_SET, Collections.EMPTY_SET, true)), true));
    jobs.add(new CompactConfig(pInfo, "/input", pInfo, "/output", 100, FileUtils.ONE_GB));
    CONFIG = new CompositeConfig(jobs, true);
  }

  @Test
  public void jsonTest() throws Exception {
    ObjectMapper mapper = new ObjectMapper();
    CompositeConfig other = mapper.readValue(mapper.writeValueAsString(CONFIG),
        CompositeConfig.class);
    assertEquals(CONFIG, other);
  }

  @Test
  public void javaSerializationTest() throws Exception {
    byte[] bytes = SerializationUtils.serialize(CONFIG);
    assertNotNull(bytes);
    Serializable deserialized = SerializationUtils.deserialize(bytes);
    assertEquals(CompositeConfig.class, deserialized.getClass());
    assertEquals(CONFIG, deserialized);
  }
}
