package azkaban.test.utils;

import java.io.IOException;
import java.io.StringWriter;
import java.util.HashMap;
import java.util.Map;

import junit.framework.Assert;

import org.junit.Test;

import azkaban.utils.JSONUtils;

public class JsonUtilsTest {
	@Test
	public void writePropsNoJarDependencyTest1() throws IOException {
		Map<String, String> test = new HashMap<String,String>();
		test.put("\"myTest\n\b", "myValue\t\\");
		test.put("normalKey", "Other key");
		
		StringWriter writer = new StringWriter();
		JSONUtils.writePropsNoJarDependency(test, writer);
		
		String jsonStr = writer.toString();
		System.out.println(writer.toString());
		
		@SuppressWarnings("unchecked")
		Map<String,String> result = (Map<String,String>)JSONUtils.parseJSONFromString(jsonStr);
		checkInAndOut(test, result);
	}
	
	@Test
	public void writePropsNoJarDependencyTest2() throws IOException {
		Map<String, String> test = new HashMap<String,String>();
		test.put("\"myTest\n\b", "myValue\t\\");
		
		StringWriter writer = new StringWriter();
		JSONUtils.writePropsNoJarDependency(test, writer);
		
		String jsonStr = writer.toString();
		System.out.println(writer.toString());
		
		@SuppressWarnings("unchecked")
		Map<String,String> result = (Map<String,String>)JSONUtils.parseJSONFromString(jsonStr);
		checkInAndOut(test, result);
	}
	
	private static void checkInAndOut(Map<String, String> before, Map<String, String> after) {
		for (Map.Entry<String, String> entry: before.entrySet()) {
			String key = entry.getKey();
			String value = entry.getValue();
			
			String retValue = after.get(key);
			Assert.assertEquals(value, retValue);
		}
	}
	
}