HttpFieldsTest.java

package test.codec.http2.model;

import static org.hamcrest.Matchers.containsString;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertThat;
import static org.junit.Assert.assertTrue;

import java.nio.ByteBuffer;
import java.util.Collections;
import java.util.Enumeration;
import java.util.List;
import java.util.Locale;
import java.util.NoSuchElementException;

import org.hamcrest.Matchers;
import org.junit.Assert;
import org.junit.Test;

import com.firefly.codec.http2.encode.HttpGenerator;
import com.firefly.codec.http2.model.HttpField;
import com.firefly.codec.http2.model.HttpFields;
import com.firefly.codec.http2.model.HttpHeader;
import com.firefly.codec.http2.model.HttpHeaderValue;
import com.firefly.utils.io.BufferUtils;

public class HttpFieldsTest {
	@Test
	public void testPut() throws Exception {
		HttpFields header = new HttpFields();

		header.put("name0", "value:0");
		header.put("name1", "value1");

		assertEquals(2, header.size());
		assertEquals("value:0", header.get("name0"));
		assertEquals("value1", header.get("name1"));
		assertNull(header.get("name2"));

		int matches = 0;
		Enumeration<String> e = header.getFieldNames();
		while (e.hasMoreElements()) {
			Object o = e.nextElement();
			if ("name0".equals(o))
				matches++;
			if ("name1".equals(o))
				matches++;
		}
		assertEquals(2, matches);

		e = header.getValues("name0");
		assertEquals(true, e.hasMoreElements());
		assertEquals(e.nextElement(), "value:0");
		assertEquals(false, e.hasMoreElements());
	}

	@Test
	public void testPutTo() throws Exception {
		HttpFields header = new HttpFields();

		header.put("name0", "value0");
		header.put("name1", "value:A");
		header.add("name1", "value:B");
		header.add("name2", "");

		ByteBuffer buffer = BufferUtils.allocate(1024);
		BufferUtils.flipToFill(buffer);
		HttpGenerator.putTo(header, buffer);
		BufferUtils.flipToFlush(buffer, 0);
		String result = BufferUtils.toString(buffer);

		assertThat(result, Matchers.containsString("name0: value0"));
		assertThat(result, Matchers.containsString("name1: value:A"));
		assertThat(result, Matchers.containsString("name1: value:B"));
	}

	@Test
	public void testGet() throws Exception {
		HttpFields header = new HttpFields();

		header.put("name0", "value0");
		header.put("name1", "value1");

		assertEquals("value0", header.get("name0"));
		assertEquals("value0", header.get("Name0"));
		assertEquals("value1", header.get("name1"));
		assertEquals("value1", header.get("Name1"));
		assertEquals(null, header.get("Name2"));

		assertEquals("value0", header.getField("name0").getValue());
		assertEquals("value0", header.getField("Name0").getValue());
		assertEquals("value1", header.getField("name1").getValue());
		assertEquals("value1", header.getField("Name1").getValue());
		assertEquals(null, header.getField("Name2"));

		assertEquals("value0", header.getField(0).getValue());
		assertEquals("value1", header.getField(1).getValue());
		try {
			header.getField(2);
			Assert.fail();
		} catch (NoSuchElementException e) {
		}
	}

	@Test
	public void testGetKnown() throws Exception {
		HttpFields header = new HttpFields();

		header.put("Connection", "value0");
		header.put(HttpHeader.ACCEPT, "value1");

		assertEquals("value0", header.get(HttpHeader.CONNECTION));
		assertEquals("value1", header.get(HttpHeader.ACCEPT));

		assertEquals("value0", header.getField(HttpHeader.CONNECTION).getValue());
		assertEquals("value1", header.getField(HttpHeader.ACCEPT).getValue());

		assertEquals(null, header.getField(HttpHeader.AGE));
		assertEquals(null, header.get(HttpHeader.AGE));
	}

	@Test
	public void testCRLF() throws Exception {
		HttpFields header = new HttpFields();

		header.put("name0", "value\r\n0");
		header.put("name\r\n1", "value1");
		header.put("name:2", "value:\r\n2");

		ByteBuffer buffer = BufferUtils.allocate(1024);
		BufferUtils.flipToFill(buffer);
		HttpGenerator.putTo(header, buffer);
		BufferUtils.flipToFlush(buffer, 0);
		String out = BufferUtils.toString(buffer);
		assertThat(out, containsString("name0: value  0"));
		assertThat(out, containsString("name??1: value1"));
		assertThat(out, containsString("name?2: value:  2"));
	}

	@Test
	public void testCachedPut() throws Exception {
		HttpFields header = new HttpFields();

		header.put("Connection", "Keep-Alive");
		header.put("tRansfer-EncOding", "CHUNKED");
		header.put("CONTENT-ENCODING", "gZIP");

		ByteBuffer buffer = BufferUtils.allocate(1024);
		BufferUtils.flipToFill(buffer);
		HttpGenerator.putTo(header, buffer);
		BufferUtils.flipToFlush(buffer, 0);
		String out = BufferUtils.toString(buffer).toLowerCase(Locale.ENGLISH);

		Assert.assertThat(out, Matchers.containsString(
				(HttpHeader.CONNECTION + ": " + HttpHeaderValue.KEEP_ALIVE).toLowerCase(Locale.ENGLISH)));
		Assert.assertThat(out, Matchers.containsString(
				(HttpHeader.TRANSFER_ENCODING + ": " + HttpHeaderValue.CHUNKED).toLowerCase(Locale.ENGLISH)));
		Assert.assertThat(out, Matchers.containsString(
				(HttpHeader.CONTENT_ENCODING + ": " + HttpHeaderValue.GZIP).toLowerCase(Locale.ENGLISH)));
	}

	@Test
	public void testRePut() throws Exception {
		HttpFields header = new HttpFields();

		header.put("name0", "value0");
		header.put("name1", "xxxxxx");
		header.put("name2", "value2");

		assertEquals("value0", header.get("name0"));
		assertEquals("xxxxxx", header.get("name1"));
		assertEquals("value2", header.get("name2"));

		header.put("name1", "value1");

		assertEquals("value0", header.get("name0"));
		assertEquals("value1", header.get("name1"));
		assertEquals("value2", header.get("name2"));
		assertNull(header.get("name3"));

		int matches = 0;
		Enumeration<String> e = header.getFieldNames();
		while (e.hasMoreElements()) {
			String o = e.nextElement();
			if ("name0".equals(o))
				matches++;
			if ("name1".equals(o))
				matches++;
			if ("name2".equals(o))
				matches++;
		}
		assertEquals(3, matches);

		e = header.getValues("name1");
		assertEquals(true, e.hasMoreElements());
		assertEquals(e.nextElement(), "value1");
		assertEquals(false, e.hasMoreElements());
	}

	@Test
	public void testRemovePut() throws Exception {
		HttpFields header = new HttpFields(1);

		header.put("name0", "value0");
		header.put("name1", "value1");
		header.put("name2", "value2");

		assertEquals("value0", header.get("name0"));
		assertEquals("value1", header.get("name1"));
		assertEquals("value2", header.get("name2"));

		header.remove("name1");

		assertEquals("value0", header.get("name0"));
		assertNull(header.get("name1"));
		assertEquals("value2", header.get("name2"));
		assertNull(header.get("name3"));

		int matches = 0;
		Enumeration<String> e = header.getFieldNames();
		while (e.hasMoreElements()) {
			Object o = e.nextElement();
			if ("name0".equals(o))
				matches++;
			if ("name1".equals(o))
				matches++;
			if ("name2".equals(o))
				matches++;
		}
		assertEquals(2, matches);

		e = header.getValues("name1");
		assertEquals(false, e.hasMoreElements());
	}

	@Test
	public void testAdd() throws Exception {
		HttpFields fields = new HttpFields();

		fields.add("name0", "value0");
		fields.add("name1", "valueA");
		fields.add("name2", "value2");

		assertEquals("value0", fields.get("name0"));
		assertEquals("valueA", fields.get("name1"));
		assertEquals("value2", fields.get("name2"));

		fields.add("name1", "valueB");

		assertEquals("value0", fields.get("name0"));
		assertEquals("valueA", fields.get("name1"));
		assertEquals("value2", fields.get("name2"));
		assertNull(fields.get("name3"));

		int matches = 0;
		Enumeration<String> e = fields.getFieldNames();
		while (e.hasMoreElements()) {
			Object o = e.nextElement();
			if ("name0".equals(o))
				matches++;
			if ("name1".equals(o))
				matches++;
			if ("name2".equals(o))
				matches++;
		}
		assertEquals(3, matches);

		e = fields.getValues("name1");
		assertEquals(true, e.hasMoreElements());
		assertEquals(e.nextElement(), "valueA");
		assertEquals(true, e.hasMoreElements());
		assertEquals(e.nextElement(), "valueB");
		assertEquals(false, e.hasMoreElements());
	}

	@Test
	public void testGetValues() throws Exception {
		HttpFields fields = new HttpFields();

		fields.put("name0", "value0A,value0B");
		fields.add("name0", "value0C,value0D");
		fields.put("name1", "value1A, \"value\t, 1B\" ");
		fields.add("name1", "\"value1C\",\tvalue1D");

		Enumeration<String> e = fields.getValues("name0");
		assertEquals(true, e.hasMoreElements());
		assertEquals(e.nextElement(), "value0A,value0B");
		assertEquals(true, e.hasMoreElements());
		assertEquals(e.nextElement(), "value0C,value0D");
		assertEquals(false, e.hasMoreElements());

		// e = fields.getValues("name0",",");
		// assertEquals(true, e.hasMoreElements());
		// assertEquals(e.nextElement(), "value0A");
		// assertEquals(true, e.hasMoreElements());
		// assertEquals(e.nextElement(), "value0B");
		// assertEquals(true, e.hasMoreElements());
		// assertEquals(e.nextElement(), "value0C");
		// assertEquals(true, e.hasMoreElements());
		// assertEquals(e.nextElement(), "value0D");
		// assertEquals(false, e.hasMoreElements());
		//
		// e = fields.getValues("name1",",");
		// assertEquals(true, e.hasMoreElements());
		// assertEquals(e.nextElement(), "value1A");
		// assertEquals(true, e.hasMoreElements());
		// assertEquals(e.nextElement(), "value\t, 1B");
		// assertEquals(true, e.hasMoreElements());
		// assertEquals(e.nextElement(), "value1C");
		// assertEquals(true, e.hasMoreElements());
		// assertEquals(e.nextElement(), "value1D");
		// assertEquals(false, e.hasMoreElements());
	}

	@Test
	public void testGetCSV() throws Exception {
		HttpFields fields = new HttpFields();

		fields.put("name0", "value0A,value0B");
		fields.add("name0", "value0C,value0D");
		fields.put("name1", "value1A, \"value\t, 1B\" ");
		fields.add("name1", "\"value1C\",\tvalue1D");

		Enumeration<String> e = fields.getValues("name0");
		assertEquals(true, e.hasMoreElements());
		assertEquals(e.nextElement(), "value0A,value0B");
		assertEquals(true, e.hasMoreElements());
		assertEquals(e.nextElement(), "value0C,value0D");
		assertEquals(false, e.hasMoreElements());

		e = Collections.enumeration(fields.getCSV("name0", false));
		assertEquals(true, e.hasMoreElements());
		assertEquals(e.nextElement(), "value0A");
		assertEquals(true, e.hasMoreElements());
		assertEquals(e.nextElement(), "value0B");
		assertEquals(true, e.hasMoreElements());
		assertEquals(e.nextElement(), "value0C");
		assertEquals(true, e.hasMoreElements());
		assertEquals(e.nextElement(), "value0D");
		assertEquals(false, e.hasMoreElements());

		e = Collections.enumeration(fields.getCSV("name1", false));
		assertEquals(true, e.hasMoreElements());
		assertEquals(e.nextElement(), "value1A");
		assertEquals(true, e.hasMoreElements());
		assertEquals(e.nextElement(), "value\t, 1B");
		assertEquals(true, e.hasMoreElements());
		assertEquals(e.nextElement(), "value1C");
		assertEquals(true, e.hasMoreElements());
		assertEquals(e.nextElement(), "value1D");
		assertEquals(false, e.hasMoreElements());
	}

	@Test
	public void testAddQuotedCSV() throws Exception {
		HttpFields fields = new HttpFields();

		fields.put("some", "value");
		fields.add("name", "\"zero\"");
		fields.add("name", "one, \"1 + 1\"");
		fields.put("other", "value");
		fields.add("name", "three");
		fields.add("name", "four, I V");

		List<String> list = fields.getCSV("name", false);
		assertEquals("zero", HttpFields.valueParameters(list.get(0), null));
		assertEquals("one", HttpFields.valueParameters(list.get(1), null));
		assertEquals("1 + 1", HttpFields.valueParameters(list.get(2), null));
		assertEquals("three", HttpFields.valueParameters(list.get(3), null));
		assertEquals("four", HttpFields.valueParameters(list.get(4), null));
		assertEquals("I V", HttpFields.valueParameters(list.get(5), null));

		fields.addCSV("name", "six");
		list = fields.getCSV("name", false);
		assertEquals("zero", HttpFields.valueParameters(list.get(0), null));
		assertEquals("one", HttpFields.valueParameters(list.get(1), null));
		assertEquals("1 + 1", HttpFields.valueParameters(list.get(2), null));
		assertEquals("three", HttpFields.valueParameters(list.get(3), null));
		assertEquals("four", HttpFields.valueParameters(list.get(4), null));
		assertEquals("I V", HttpFields.valueParameters(list.get(5), null));
		assertEquals("six", HttpFields.valueParameters(list.get(6), null));

		fields.addCSV("name", "1 + 1", "7", "zero");
		list = fields.getCSV("name", false);
		assertEquals("zero", HttpFields.valueParameters(list.get(0), null));
		assertEquals("one", HttpFields.valueParameters(list.get(1), null));
		assertEquals("1 + 1", HttpFields.valueParameters(list.get(2), null));
		assertEquals("three", HttpFields.valueParameters(list.get(3), null));
		assertEquals("four", HttpFields.valueParameters(list.get(4), null));
		assertEquals("I V", HttpFields.valueParameters(list.get(5), null));
		assertEquals("six", HttpFields.valueParameters(list.get(6), null));
		assertEquals("7", HttpFields.valueParameters(list.get(7), null));
	}

	@Test
	public void testGetQualityCSV() throws Exception {
		HttpFields fields = new HttpFields();

		fields.put("some", "value");
		fields.add("name", "zero;q=0.9,four;q=0.1");
		fields.put("other", "value");
		fields.add("name", "nothing;q=0");
		fields.add("name", "one;q=0.4");
		fields.add("name", "three;x=y;q=0.2;a=b,two;q=0.3");

		List<String> list = fields.getQualityCSV("name");
		assertEquals("zero", HttpFields.valueParameters(list.get(0), null));
		assertEquals("one", HttpFields.valueParameters(list.get(1), null));
		assertEquals("two", HttpFields.valueParameters(list.get(2), null));
		assertEquals("three", HttpFields.valueParameters(list.get(3), null));
		assertEquals("four", HttpFields.valueParameters(list.get(4), null));
	}

	@Test
	public void testDateFields() throws Exception {
		HttpFields fields = new HttpFields();

		fields.put("D0", "Wed, 31 Dec 1969 23:59:59 GMT");
		fields.put("D1", "Fri, 31 Dec 1999 23:59:59 GMT");
		fields.put("D2", "Friday, 31-Dec-99 23:59:59 GMT");
		fields.put("D3", "Fri Dec 31 23:59:59 1999");
		fields.put("D4", "Mon Jan 1 2000 00:00:01");
		fields.put("D5", "Tue Feb 29 2000 12:00:00");

		long d1 = fields.getDateField("D1");
		long d0 = fields.getDateField("D0");
		long d2 = fields.getDateField("D2");
		long d3 = fields.getDateField("D3");
		long d4 = fields.getDateField("D4");
		long d5 = fields.getDateField("D5");
		assertTrue(d0 != -1);
		assertTrue(d1 > 0);
		assertTrue(d2 > 0);
		assertEquals(d1, d2);
		assertEquals(d2, d3);
		assertEquals(d3 + 2000, d4);
		assertEquals(951825600000L, d5);

		d1 = fields.getDateField("D1");
		d2 = fields.getDateField("D2");
		d3 = fields.getDateField("D3");
		d4 = fields.getDateField("D4");
		d5 = fields.getDateField("D5");
		assertTrue(d1 > 0);
		assertTrue(d2 > 0);
		assertEquals(d1, d2);
		assertEquals(d2, d3);
		assertEquals(d3 + 2000, d4);
		assertEquals(951825600000L, d5);

		fields.putDateField("D2", d1);
		assertEquals("Fri, 31 Dec 1999 23:59:59 GMT", fields.get("D2"));
	}

	@Test
	public void testNegDateFields() throws Exception {
		HttpFields fields = new HttpFields();

		fields.putDateField("Dzero", 0);
		assertEquals("Thu, 01 Jan 1970 00:00:00 GMT", fields.get("Dzero"));

		fields.putDateField("Dminus", -1);
		assertEquals("Wed, 31 Dec 1969 23:59:59 GMT", fields.get("Dminus"));

		fields.putDateField("Dminus", -1000);
		assertEquals("Wed, 31 Dec 1969 23:59:59 GMT", fields.get("Dminus"));

		fields.putDateField("Dancient", Long.MIN_VALUE);
		assertEquals("Sun, 02 Dec 55 16:47:04 GMT", fields.get("Dancient"));
	}

	@Test
	public void testLongFields() throws Exception {
		HttpFields header = new HttpFields();

		header.put("I1", "42");
		header.put("I2", " 43 99");
		header.put("I3", "-44");
		header.put("I4", " - 45abc");
		header.put("N1", " - ");
		header.put("N2", "xx");

		long i1 = header.getLongField("I1");
		try {
			header.getLongField("I2");
			assertTrue(false);
		} catch (NumberFormatException e) {
			assertTrue(true);
		}

		long i3 = header.getLongField("I3");

		try {
			header.getLongField("I4");
			assertTrue(false);
		} catch (NumberFormatException e) {
			assertTrue(true);
		}

		try {
			header.getLongField("N1");
			assertTrue(false);
		} catch (NumberFormatException e) {
			assertTrue(true);
		}

		try {
			header.getLongField("N2");
			assertTrue(false);
		} catch (NumberFormatException e) {
			assertTrue(true);
		}

		assertEquals(42, i1);
		assertEquals(-44, i3);

		header.putLongField("I5", 46);
		header.putLongField("I6", -47);
		assertEquals("46", header.get("I5"));
		assertEquals("-47", header.get("I6"));
	}

	@Test
	public void testContains() throws Exception {
		HttpFields header = new HttpFields();

		header.add("n0", "");
		header.add("n1", ",");
		header.add("n2", ",,");
		header.add("N3", "abc");
		header.add("N4", "def");
		header.add("n5", "abc,def,hig");
		header.add("N6", "abc");
		header.add("n6", "def");
		header.add("N6", "hig");
		header.add("n7", "abc ,  def;q=0.9  ,  hig");
		header.add("n8", "abc ,  def;q=0  ,  hig");
		header.add(HttpHeader.ACCEPT, "abc ,  def;q=0  ,  hig");

		for (int i = 0; i < 8; i++) {
			assertTrue(header.containsKey("n" + i));
			assertTrue(header.containsKey("N" + i));
			assertFalse("" + i, header.contains("n" + i, "xyz"));
			assertEquals("" + i, i >= 4, header.contains("n" + i, "def"));
		}

		assertTrue(header.contains(new HttpField("N5", "def")));
		assertTrue(header.contains(new HttpField("accept", "abc")));
		assertTrue(header.contains(HttpHeader.ACCEPT, "abc"));
		assertFalse(header.contains(new HttpField("N5", "xyz")));
		assertFalse(header.contains(new HttpField("N8", "def")));
		assertFalse(header.contains(HttpHeader.ACCEPT, "def"));
		assertFalse(header.contains(HttpHeader.AGE, "abc"));

		assertFalse(header.containsKey("n11"));
	}
}