org.ebayopensource.turmeric.eclipse.utils.test.io.TestPropertiesFileUtil.java Source code

Introduction

Here is the source code for org.ebayopensource.turmeric.eclipse.utils.test.io.TestPropertiesFileUtil.java

Source

/*******************************************************************************
 * Copyright (c) 2006-2010 eBay Inc. 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
 *******************************************************************************/
package org.ebayopensource.turmeric.eclipse.utils.test.io;

import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.Map;
import java.util.Properties;
import java.util.concurrent.ConcurrentHashMap;

import org.apache.commons.io.IOUtils;
import org.ebayopensource.turmeric.eclipse.utils.io.PropertiesFileUtil;
import org.junit.Assert;
import org.junit.Test;

/**
 * @author yayu
 *
 */
public class TestPropertiesFileUtil {

    /**
     * Test method for {@link org.ebayopensource.turmeric.eclipse.utils.io.PropertiesFileUtil#isEqual(java.util.Properties, java.util.Properties)}.
     */
    @Test
    public void testIsEqual() {
        Map<String, String> data = new ConcurrentHashMap<String, String>();
        data.put("nikon", "D3x");
        data.put("canon", "1D Mark III");
        data.put("pentax", "K7");

        Properties props1 = new Properties();
        props1.putAll(data);

        Properties props2 = new Properties();
        props2.putAll(new ConcurrentHashMap<String, String>(data));
        Assert.assertTrue(PropertiesFileUtil.isEqual(props1, props2));

        props2.put("sony", "a900");
        Assert.assertFalse(PropertiesFileUtil.isEqual(props1, props2));

        props2.remove("sony");
        props2.put("canon", "sucks");
        System.out.println(props1);
        System.out.println(props2);
        Assert.assertFalse(PropertiesFileUtil.isEqual(props1, props2));

    }

    /**
     * Test method for {@link org.ebayopensource.turmeric.eclipse.utils.io.PropertiesFileUtil#getPropertyValueByKey(java.io.InputStream, java.lang.String)}.
     * @throws IOException 
     */
    @Test
    public void testGetPropertyValueByKey() throws IOException {
        File tempFile = null;
        InputStream input = null;
        OutputStream output = null;
        try {
            tempFile = File.createTempFile("nikon", "d80");
            output = new FileOutputStream(tempFile);
            Properties props = new Properties();
            props.put("Nikon", "d80,d90,d700,d3s,d3x");
            props.store(output, "Comments");
            IOUtils.closeQuietly(output);

            input = new FileInputStream(tempFile);
            String value = PropertiesFileUtil.getPropertyValueByKey(input, "Nikon");
            Assert.assertNotNull(value);
            Assert.assertEquals("d80,d90,d700,d3s,d3x", value);
        } finally {
            IOUtils.closeQuietly(output);
            IOUtils.closeQuietly(input);
            tempFile.delete();
        }

    }

    /**
     * Test method for {@link org.ebayopensource.turmeric.eclipse.utils.io.PropertiesFileUtil#removePropertyByKey(java.io.InputStream, java.io.OutputStream, java.lang.String[])}.
     * @throws IOException 
     */
    @Test
    public void testRemovePropertyByKey() throws IOException {
        File tempFile = null;
        InputStream input = null;
        OutputStream output = null;
        try {
            tempFile = File.createTempFile("nikon", "d80");
            output = new FileOutputStream(tempFile);
            Properties props = new Properties();
            props.put("Nikon", "d80,d90,d700,d3s,d3x");
            props.put("Canon", "5D Mark II");
            props.store(output, "Comments");
            IOUtils.closeQuietly(output);

            input = new FileInputStream(tempFile);
            String value = PropertiesFileUtil.getPropertyValueByKey(input, "Nikon");
            Assert.assertNotNull(value);
            Assert.assertEquals("d80,d90,d700,d3s,d3x", value);

            output = new ByteArrayOutputStream();
            PropertiesFileUtil.removePropertyByKey(input, output, new String[] { "Nikon" });

            System.out.println("byte output: " + output.toString());
        } finally {
            IOUtils.closeQuietly(output);
            IOUtils.closeQuietly(input);
            tempFile.delete();
        }
    }

}