MarkupUtilsTest.java

//Dstl (c) Crown Copyright 2017
package uk.gov.dstl.baleen.contentmanipulators.helpers;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

import org.jsoup.nodes.Element;
import org.jsoup.parser.Tag;
import org.junit.Test;

public class MarkupUtilsTest {

	@Test
	public void testAdditionallyAnnotateAsType() {
		Element e = new Element(Tag.valueOf("p"), "");
		MarkupUtils.additionallyAnnotateAsType(e, "testtype");

		assertEquals(MarkupUtils.getAttribute(e, "types"), "testtype");

		assertTrue(MarkupUtils.getTypes(e).contains("testtype"));

	}

	@Test
	public void testSetAttribute() {
		Element e = new Element(Tag.valueOf("p"), "");
		MarkupUtils.setAttribute(e, "key", "value");
		assert (MarkupUtils.getAttribute(e, "key").equals("value"));

		MarkupUtils.setAttribute(e, "key", "value2");
		assert (MarkupUtils.getAttribute(e, "key").equals("value2"));

	}

	@Test
	public void testAddAttribute() {
		Element e = new Element(Tag.valueOf("p"), "");
		MarkupUtils.addAttribute(e, "key", "value1");

		assert (MarkupUtils.getAttribute(e, "key").contains("value1"));

		MarkupUtils.addAttribute(e, "key", "value2");

		assert (MarkupUtils.getAttribute(e, "key").contains("value1"));
		assert (MarkupUtils.getAttribute(e, "key").contains("value2"));

		assert (MarkupUtils.getAttributes(e, "key").contains("value1"));
		assert (MarkupUtils.getAttributes(e, "key").contains("value2"));
	}
}