--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/test/org/tmatesoft/hg/test/TestKeywordFilter.java	Fri Oct 11 21:35:41 2013 +0200
@@ -0,0 +1,140 @@
+/*
+ * Copyright (c) 2013 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.File;
+import java.nio.ByteBuffer;
+
+import org.junit.Assert;
+import org.junit.Test;
+import org.tmatesoft.hg.internal.ByteArrayChannel;
+import org.tmatesoft.hg.internal.Filter;
+import org.tmatesoft.hg.internal.Filter.Direction;
+import org.tmatesoft.hg.internal.KeywordFilter;
+import org.tmatesoft.hg.repo.HgLookup;
+import org.tmatesoft.hg.repo.HgRepository;
+import org.tmatesoft.hg.repo.HgRepositoryFiles;
+import org.tmatesoft.hg.util.Path;
+
+/**
+ * 
+ * @author Artem Tikhomirov
+ * @author TMate Software Ltd.
+ */
+public class TestKeywordFilter {
+	
+	private HgRepository repo;
+
+	@Test
+	public void testSmallBuffer() throws Exception {
+		initRepo();
+		final Filter kwFilter = createFilter(Filter.Direction.ToRepo);
+		final byte[] in = "1\n2\n3\n".getBytes();
+		ByteBuffer bb = kwFilter.filter(ByteBuffer.wrap(in));
+		final byte[] out = new byte[bb.remaining()];
+		bb.get(out);
+		Assert.assertArrayEquals(in, out);
+	}
+	
+	@Test
+	public void testKeywordDrop() throws Exception {
+		initRepo();
+		final Filter kwFilter = createFilter(Filter.Direction.ToRepo);
+		final byte[] in = "1\n$Revision: cf200271439a7ec256151b30bc4360b21db3542e$\n3\n".getBytes();
+		ByteBuffer bb = kwFilter.filter(ByteBuffer.wrap(in));
+		final byte[] out = new byte[bb.remaining()];
+		bb.get(out);
+		Assert.assertArrayEquals("1\n$Revision$\n3\n".getBytes(), out);
+	}
+
+	@Test
+	public void testKeywordExpansion() throws Exception {
+		Assert.fail("Need real repo with changeset to check kw expansion");
+	}
+
+	/**
+	 * what if keyword is split between two input buffers  
+	 */
+	@Test
+	public void testKeywordSplitInBuffer() throws Exception {
+		initRepo();
+		final byte[] in1 = "1\n$Id:whatever".getBytes();
+		final byte[] in2 = " continues here$\n3\n".getBytes();
+		ByteArrayChannel out = new ByteArrayChannel();
+		final Filter kwFilter = createFilter(Filter.Direction.ToRepo);
+		out.write(kwFilter.filter(ByteBuffer.wrap(in1)));
+		out.write(kwFilter.filter(ByteBuffer.wrap(in2)));
+		Assert.assertEquals("1\n$Id$\n3\n", new String(out.toArray()));
+		// Same as above, to extreme - only $ in the first buffer
+		final Filter kwFilter2 = createFilter(Filter.Direction.ToRepo);
+		out = new ByteArrayChannel();
+		out.write(kwFilter2.filter(ByteBuffer.wrap("1\n$".getBytes())));
+		out.write(kwFilter2.filter(ByteBuffer.wrap("Id:whatever continues here$\n3\n".getBytes())));
+		Assert.assertEquals("1\n$Id$\n3\n", new String(out.toArray()));
+	}
+	
+	/**
+	 * what if input contains smth similar to keyword but unless the second part of the buffer
+	 * comes in, it's impossible to tell
+	 */
+	@Test
+	public void testIncompleteKeyword() throws Exception {
+		initRepo();
+		final byte[] in1 = "1\n$Id:whatever".getBytes();
+		final byte[] in2 = " id doesn't close here\n3\n".getBytes();
+		ByteArrayChannel out = new ByteArrayChannel();
+		final Filter kwFilter = createFilter(Filter.Direction.ToRepo);
+		out.write(kwFilter.filter(ByteBuffer.wrap(in1)));
+		out.write(kwFilter.filter(ByteBuffer.wrap(in2)));
+		byte[] expected = new byte[in1.length + in2.length];
+		System.arraycopy(in1, 0, expected, 0, in1.length);
+		System.arraycopy(in2, 0, expected, in1.length, in2.length);
+		Assert.assertEquals(new String(expected), new String(out.toArray()));
+	}
+	
+	@Test
+	public void testIncompleteKeywordAtEOF() throws Exception {
+		initRepo();
+		final byte[] in = "1\n$Id:whatever\n".getBytes();
+		final Filter kwFilter = createFilter(Filter.Direction.ToRepo);
+		ByteBuffer outBuf = kwFilter.filter(ByteBuffer.wrap(in));
+		byte[] out = new byte[outBuf.remaining()];
+		outBuf.get(out);
+		Assert.assertEquals(new String(in), new String(out));
+		//
+		// incomplete $kw is stripped of in case of EOF
+		final Filter kwFilter2 = createFilter(Filter.Direction.ToRepo);
+		outBuf = kwFilter2.filter(ByteBuffer.wrap("1\n$Id:whatever".getBytes()));
+		out = new byte[outBuf.remaining()];
+		outBuf.get(out);
+		Assert.assertEquals("1\n", new String(out));
+	}
+
+	private Filter createFilter(Direction dir) {
+		final KeywordFilter.Factory kwFactory = new KeywordFilter.Factory();
+		kwFactory.initialize(repo);
+		return kwFactory.create(Path.create("a/b"), new Filter.Options(dir));
+	}
+
+	private HgRepository initRepo() throws Exception {
+		final File repoLoc = RepoUtils.initEmptyTempRepo("test-kw-filter");
+		final File hgrc = new File(repoLoc, HgRepositoryFiles.RepoConfig.getPath());
+		RepoUtils.createFile(hgrc, "[extensions]\nkeyword=\n\n[keyword]\n**.*=\n");
+		repo = new HgLookup().detect(repoLoc);
+		return repo;
+	}
+}