SpaceIndexingInfoTest.java

/*
 * JBoss, Home of Professional Open Source
 * Copyright 2012 Red Hat Inc. and/or its affiliates and other contributors
 * as indicated by the @authors tag. All rights reserved.
 */
package org.jboss.elasticsearch.river.remote;

import java.io.IOException;

import junit.framework.Assert;

import org.elasticsearch.common.xcontent.XContentFactory;
import org.elasticsearch.common.xcontent.XContentType;
import org.jboss.elasticsearch.river.remote.testtools.TestUtils;
import org.junit.Test;

/**
 * Unit test for {@link SpaceIndexingInfo}.
 * 
 * @author Vlastimil Elias (velias at redhat dot com)
 * @author Lukáš Vlček (lvlcek@redhat.com)
 */
public class SpaceIndexingInfoTest {

	@Test
	public void addErrorMessage() {
		SpaceIndexingInfo tested = new SpaceIndexingInfo("KEY", true);

		// case - empty at the begining
		Assert.assertNull(tested.getErrorMessage());

		// case - do not add empty messages
		tested.addErrorMessage(" ");
		Assert.assertNull(tested.getErrorMessage());

		// case - add first row
		tested.addErrorMessage("msg row 1");
		Assert.assertEquals("msg row 1", tested.getErrorMessage());

		// case - do not add null messages
		tested.addErrorMessage(null);
		Assert.assertEquals("msg row 1", tested.getErrorMessage());

		// case - add second row
		tested.addErrorMessage("msg row 2");
		Assert.assertEquals("msg row 1\nmsg row 2", tested.getErrorMessage());

	}

	@Test
	public void buildDocument() throws Exception {

		TestUtils.assertJsonEqual(TestUtils.readStringFromClasspathFile("/asserts/SpaceIndexingInfoTest_1.json"),
				new SpaceIndexingInfo("ORG", true, 10, 1, 1, DateTimeUtils.parseISODateTime("2012-09-10T12:55:58Z"), true,
						1250, null).buildDocument(XContentFactory.jsonBuilder(), null, true, true).string());

		SpaceIndexingInfo sit = new SpaceIndexingInfo("ORG", true, 10, 1, 1,
				DateTimeUtils.parseISODateTime("2012-09-10T12:56:50Z"), false, 125, "Error message");
		sit.documentsWithError = 10;
		TestUtils.assertJsonEqual(TestUtils.readStringFromClasspathFile("/asserts/SpaceIndexingInfoTest_2.json"), sit
				.buildDocument(XContentFactory.jsonBuilder(), "my_river", true, true).string());

		TestUtils.assertJsonEqual(TestUtils.readStringFromClasspathFile("/asserts/SpaceIndexingInfoTest_3.json"),
				new SpaceIndexingInfo("ORG", true, 10, 1, 1, DateTimeUtils.parseISODateTime("2012-09-10T12:55:58Z"), true,
						1250, null).buildDocument(XContentFactory.jsonBuilder(), "my_river", false, true).string());

		TestUtils.assertJsonEqual(TestUtils.readStringFromClasspathFile("/asserts/SpaceIndexingInfoTest_4.json"),
				new SpaceIndexingInfo("ORG", true, 10, 1, 1, DateTimeUtils.parseISODateTime("2012-09-10T12:55:58Z"), true,
						1250, null).buildDocument(XContentFactory.jsonBuilder(), null, false, false).string());
	}

	@Test
	public void readFromDocument() throws IOException {
		readFromDocumentInternalTest(new SpaceIndexingInfo("ORG", true, 10, 1, 1,
				DateTimeUtils.parseISODateTime("2012-09-10T12:55:58Z"), true, 1250, null));
		readFromDocumentInternalTest(new SpaceIndexingInfo("ORGA", false, 10, 0, 1,
				DateTimeUtils.parseISODateTime("2012-09-11T02:55:58Z"), false, 125, "Error"));
	}

	private void readFromDocumentInternalTest(SpaceIndexingInfo src) throws IOException {
		SpaceIndexingInfo result = SpaceIndexingInfo.readFromDocument(XContentFactory.xContent(XContentType.JSON)
				.createParser(src.buildDocument(XContentFactory.jsonBuilder(), null, true, true).string()).mapAndClose());

		Assert.assertEquals(src.spaceKey, result.spaceKey);
		Assert.assertEquals(src.fullUpdate, result.fullUpdate);
		Assert.assertEquals(src.documentsUpdated, result.documentsUpdated);
		Assert.assertEquals(src.documentsDeleted, result.documentsDeleted);
		Assert.assertEquals(src.documentsWithError, result.documentsWithError);
		Assert.assertEquals(src.commentsDeleted, result.commentsDeleted);
		Assert.assertEquals(src.startDate, result.startDate);
		Assert.assertEquals(src.finishedOK, result.finishedOK);
		Assert.assertEquals(src.timeElapsed, result.timeElapsed);
		Assert.assertEquals(src.getErrorMessage(), result.getErrorMessage());
	}

}