wherehows.common.utils.JobsUtilTest.java Source code

Introduction

Here is the source code for wherehows.common.utils.JobsUtilTest.java

Source

/**
 * Copyright 2015 LinkedIn Corp. All rights reserved.
 *
 * 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.
 */
package wherehows.common.utils;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.Map;
import java.util.Properties;
import org.apache.commons.configuration.ConfigurationException;
import org.testng.Assert;
import org.testng.annotations.Test;
import wherehows.common.Constant;

import static wherehows.common.utils.JobsUtil.*;

public class JobsUtilTest {

    @Test
    public void testEnvVarResolution() throws IOException, ConfigurationException {
        String propertyStr = "var1=foo\n" + "var2=$JAVA_HOME\n" + "var3=${JAVA_HOME}";

        Path path = createPropertiesFile(propertyStr);

        String fileName = jobNameFromPath(path);
        String dir = path.getParent().toString();

        Properties props = getJobConfigProperties(path.toFile());

        Assert.assertNotEquals(props, null);
        Assert.assertEquals(props.getProperty("var1", ""), "foo");
        Assert.assertTrue(props.getProperty("var2").length() > 0);
        Assert.assertEquals(props.getProperty("var2"), props.getProperty("var3"));
        Files.deleteIfExists(path);
    }

    @Test
    public void testGetScheduledJobs() throws IOException, ConfigurationException {
        String propertyStr1 = "job.class=test\n" + "job.cron.expr=0 0 1 * * ? *\n" + "#job.disabled=1\n"
                + "job.type=TEST1";
        String propertyStr2 = "job.class=test\n" + "job.cron.expr=0 0 1 * * ? *\n" + "job.disabled=1\n"
                + "job.type=TEST2";
        String propertyStr3 = "job.class=test\n" + "#job.disabled=1\n" + "job.type=TEST3";

        Path path1 = createPropertiesFile(propertyStr1);
        Path path2 = createPropertiesFile(propertyStr2);
        Path path3 = createPropertiesFile(propertyStr3);

        String filename1 = jobNameFromPath(path1);

        String dir = path1.getParent().toString();

        Map<String, Properties> jobs = getScheduledJobs(dir);

        Assert.assertEquals(jobs.size(), 1);
        Assert.assertEquals(jobs.get(filename1).getProperty(Constant.JOB_TYPE_KEY), "TEST1");
        Assert.assertEquals(jobs.get(filename1).getProperty("job.class"), "test");

        Files.deleteIfExists(path1);
        Files.deleteIfExists(path2);
        Files.deleteIfExists(path3);
    }

    @Test
    public void testGetEnabledJobs() throws IOException, ConfigurationException {
        String propertyStr1 = "job.class=test\n" + "job.cron.expr=0 0 1 * * ? *\n" + "#job.disabled=1\n"
                + "job.type=TEST1";
        String propertyStr2 = "job.class=test\n" + "#job.disabled=1\n";

        Path path1 = createPropertiesFile(propertyStr1);
        Path path2 = createPropertiesFile(propertyStr2);

        String filename1 = jobNameFromPath(path1);
        String filename2 = jobNameFromPath(path2);

        String dir = path1.getParent().toString();

        Map<String, Properties> jobs = getEnabledJobs(dir);

        Assert.assertEquals(jobs.size(), 2);
        Assert.assertEquals(jobs.get(filename1).getProperty("job.class"), "test");
        Assert.assertEquals(jobs.get(filename1).getProperty("job.disabled", ""), "");
        Assert.assertEquals(jobs.get(filename2).getProperty("job.class"), "test");

        Files.deleteIfExists(path1);
        Files.deleteIfExists(path2);
    }

    @Test
    public void testGetEnabledJobsByType() throws IOException, ConfigurationException {
        String propertyStr1 = "job.class=test\n" + "job.cron.expr=0 0 1 * * ? *\n" + "#job.disabled=1\n"
                + "job.type=TEST1";
        String propertyStr2 = "job.class=test\n" + "job.cron.expr=0 0 1 * * ? *\n" + "#job.disabled=1\n"
                + "job.type=TEST2,TEST3";

        Path path1 = createPropertiesFile(propertyStr1);
        Path path2 = createPropertiesFile(propertyStr2);

        String filename1 = jobNameFromPath(path1);

        String dir = path1.getParent().toString();

        Map<String, Properties> jobs = getEnabledJobsByType(dir, "TEST1");

        Assert.assertEquals(jobs.size(), 1);
        Assert.assertEquals(jobs.get(filename1).getProperty("job.class"), "test");
        Assert.assertEquals(jobs.get(filename1).getProperty("job.disabled", ""), "");

        Files.deleteIfExists(path1);
        Files.deleteIfExists(path2);
    }

    @Test
    public void testMultipleLinesSameProperties() throws IOException, ConfigurationException {
        String dir = "/tmp/";
        String propertyStr = "var1=com.mysql.jdbc.driver\n" + "var1=username\n" + "var1=password\n" + "var2=3301\n"
                + "var2=1000\n" + "var3=http://wherehows.com\n" + "var1=mysql\n" + "var4=wherehows\n"
                + "var5=1000,10,3\n" + "var5=true\n";
        ;
        Properties props = new Properties();
        Path path = createPropertiesFile(propertyStr);

        props = getJobConfigProperties(path.toFile());

        Assert.assertNotEquals(props, null);
        Assert.assertEquals(props.getProperty("var1", ""), "com.mysql.jdbc.driver,username,password,mysql");
        Assert.assertEquals(props.getProperty("var2"), "3301,1000");
        Assert.assertEquals(props.getProperty("var3"), "http://wherehows.com");
        Assert.assertEquals(props.getProperty("var4"), "wherehows");
        Assert.assertEquals(props.getProperty("var5"), "1000,10,3,true");

        Files.deleteIfExists(path);
    }

    private Path createPropertiesFile(String content) throws IOException {
        File propertyFile = File.createTempFile("temp", ".job");
        FileWriter writer = new FileWriter(propertyFile);
        writer.write(content);
        writer.close();
        return propertyFile.toPath();
    }
}