/*
 * Copyright (c) 2012 TMate Software Ltd
 *  
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; version 2 of the License.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * For information on how to redistribute this software under
 * the terms of a license other than GNU General Public License
 * contact TMate Software at support@hg4j.com
 */
package org.tmatesoft.hg.test;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;

import org.junit.Assert;
import org.junit.Test;
import org.tmatesoft.hg.internal.ConfigFileParser;

/**
 * 
 * @author Artem Tikhomirov
 * @author TMate Software Ltd.
 */
public class TestConfigFileParser {

	@Test
	public void testParseOnly() throws IOException {
		doTest("".getBytes(), new byte[0]);
		// line comments
		byte[] inp = " # line comment \n; and another one".getBytes();
		doTest(inp, inp);
		// comments inside sections
		inp = "[section1]\nkey1 = value\n # line comment\n[section2]\nkey2 = ;just presence\n".getBytes();
		doTest(inp, inp);
		// empty value
		inp = "[section1]\nkey1 = \n".getBytes();
		doTest(inp, inp);
		// multiline values
		inp = "[section1]\nkey1 = a,\n  b,\n  c\nkey2=\n  xyz\n".getBytes();
		doTest(inp, inp);
		// entry without EOL
		inp = "[section1]\nkey1 = value".getBytes();
		doTest(inp, inp);
		// empty section
		inp = "[section1]\nkey1 = value\n[section2]\n".getBytes();
		doTest(inp, inp);
	}
	
	@Test
	public void testLookup() throws IOException {
		byte[] inp = "[section1]\nkey1 = a,\n  b,\n  c\nkey2=\n  xyz\n\n[section2]\nkey3=\n".getBytes();
		doTest(inp, inp, new Inspector() {
			
			public void visit(ConfigFileParser p) {
				Assert.assertTrue(p.exists("section1", "key1"));
				Assert.assertTrue(p.exists("section1", "key2"));
				Assert.assertFalse(p.exists("section1", "key3"));
				
				Assert.assertTrue(p.exists("section2", "key3"));
				Assert.assertFalse(p.exists("section2", "key1"));
				Assert.assertFalse(p.exists("section2", "key2"));
			}
		});
	}

	@Test
	public void testAddChangeEntries() throws IOException {
		byte[] inp = "\n; line comment1\n[sect-a]\nkey1 = value1\n\n[sect-b]\nkey2=value2\n".getBytes();
		byte[] exp = "\n; line comment1\n[sect-a]\nkey1 = value1\nkey3 = value3\n\n[sect-b]\nkey2=valueX\n".getBytes();
		doTest(inp, exp, new Inspector() {
			
			public void visit(ConfigFileParser p) {
				Assert.assertTrue(p.exists("sect-b", "key2"));
				p.add("sect-a", "key3", "value3");
				p.change("sect-b", "key2", "valueX");
			}
		});
	}

	@Test
	public void testAdditionTwoSectionsSameName() throws IOException {
		byte[] inp = "[sect-a]\nkey1=value1\n\n[sect-b]\nkey2=\n\n[sect-a]\nkey3=value3\n".getBytes();
		byte[] exp = "[sect-a]\nkey1=value1\nkey4 = value4\n\n[sect-b]\nkey2=\n\n[sect-a]\nkey3=value3\n".getBytes();
		doTest(inp, exp, new Inspector() {
			
			public void visit(ConfigFileParser p) {
				p.add("sect-a", "key4", "value4");
			}
		});
	}
	
	@Test
	public void testDeleteTwoSubsequentKeys() throws IOException{
		byte[] inp = "# line comment1\n\n[sect-a]\nkey1=value1\nkey2=value2\n#line comment2\nkey3=value3\n".getBytes();
		byte[] exp = "# line comment1\n\n[sect-a]\n\n\n#line comment2\nkey3=value3\n".getBytes();
		doTest(inp, exp, new Inspector() {
			
			public void visit(ConfigFileParser p) {
				p.delete("sect-a", "key1");
				p.delete("sect-a", "key2");
			}
		});
	}
	
	@Test
	public void testDeleteLastKeyInSection() throws IOException {
		String text1 = "[sect-a]\nkey0 = value 0\n%skey1=value1\n%s[sect-b]\nkey3=value3\n";
		String text2 = "[sect-a]\nkey0 = value 0\n%s\n%s[sect-b]\nkey3=value3\n";
		withTwoCommentsDeleteKey1(text1, text2);
	}

	@Test
	public void testDeleteFirstKeyInSection() throws IOException {
		String text1 = "[sect-a]\n%skey1=value1\n%skey2 = value 2\n[sect-b]\nkey3=value3\n";
		String text2 = "[sect-a]\n%s\n%skey2 = value 2\n[sect-b]\nkey3=value3\n";
		withTwoCommentsDeleteKey1(text1, text2);
	}
	
	@Test
	public void testDeleteOnlyKeyInSection() throws IOException {
		String text1 = "[sect-a]\n%skey1=value1\n%s[sect-b]\nkey3=value3\n";
		String text2 = "[sect-a]\n%s\n%s[sect-b]\nkey3=value3\n";
		withTwoCommentsDeleteKey1(text1, text2);
	}
	
	@Test
	public void testAddNewSection() throws IOException {
		byte[] inp = "[sect-a]\nkey1=value1\n".getBytes();
		byte[] exp = "[sect-a]\nkey1=value1\n\n[sect-b]\nkey2 = value2\n".getBytes();
		doTest(inp, exp, new Inspector() {
			
			public void visit(ConfigFileParser p) {
				p.add("sect-b", "key2", "value2");
			}
		});
	}
	
	private void withTwoCommentsDeleteKey1(String text1, String text2) throws IOException {
		String comment = "# line comment\n";
		Inspector insp = new Inspector() {
			
			public void visit(ConfigFileParser p) {
				p.delete("sect-a", "key1");
			}
		};

		byte[] inp = String.format(text1, "", "").getBytes();
		byte[] exp = String.format(text2, "", "").getBytes();
		doTest(inp, exp, insp);
		inp = String.format(text1, comment, "").getBytes();
		exp = String.format(text2, comment, "").getBytes();
		doTest(inp, exp, insp);
		inp = String.format(text1, "", comment).getBytes();
		exp = String.format(text2, "", comment).getBytes();
		doTest(inp, exp, insp);
		inp = String.format(text1, comment, comment).getBytes();
		exp = String.format(text2, comment, comment).getBytes();
		doTest(inp, exp, insp);
	}

	@Test
	public void testAddEntryToEmptySection() throws IOException {
		String text1 = "[sect-a]\n%s[sect-b]\nkey3=value3\n";
		String text2 = "[sect-a]\n%skey1 = value1\n[sect-b]\nkey3=value3\n";
		String comment = "# line comment2\n";
		Inspector insp = new Inspector() {
			
			public void visit(ConfigFileParser p) {
				p.add("sect-a", "key1", "value1");
			}
		};

		byte[] inp = String.format(text1, "").getBytes();
		byte[] exp = String.format(text2, "").getBytes();
		doTest(inp, exp, insp);
		inp = String.format(text1, comment).getBytes();
		exp = String.format(text2, comment).getBytes();
		doTest(inp, exp, insp);
	}
	

	private void doTest(byte[] input, byte[] expected) throws IOException {
		doTest(input, expected, null);
	}

	private void doTest(byte[] input, byte[] expected, Inspector insp) throws IOException {
		ConfigFileParser p = new ConfigFileParser();
		p.parse(new ByteArrayInputStream(input));
		if (insp != null) {
			insp.visit(p);
		}
		ByteArrayOutputStream out = new ByteArrayOutputStream(input.length);
		p.update(out);
		byte[] result = out.toByteArray();
		Assert.assertArrayEquals(expected, result);
	}

	interface Inspector {
		void visit(ConfigFileParser p);
	}
}