package ca.uhn.hl7v2.model;

import junit.framework.Assert;
import junit.framework.TestCase;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import ca.uhn.hl7v2.HL7Exception;
import ca.uhn.hl7v2.model.v26.datatype.AD;
import ca.uhn.hl7v2.model.v26.datatype.CD;
import ca.uhn.hl7v2.model.v26.datatype.ST;
import ca.uhn.hl7v2.model.v24.datatype.CE;
import ca.uhn.hl7v2.model.v26.message.ORU_R01;
import ca.uhn.hl7v2.model.v26.segment.OBX;
import ca.uhn.hl7v2.parser.DefaultXMLParser;
import ca.uhn.hl7v2.parser.EncodingCharacters;
import ca.uhn.hl7v2.parser.EncodingNotSupportedException;
import ca.uhn.hl7v2.parser.GenericParser;
import ca.uhn.hl7v2.parser.Parser;
import ca.uhn.hl7v2.parser.PipeParser;
import ca.uhn.hl7v2.util.Terser;

/**
 * A JUnit test harness for Varies
 * 
 * @author Bryan Tripp
 */
public class VariesTest extends TestCase {

	public static final String MSH_TEXT = "MSH|^~\\&|QVDX|WP|CCG-PDC-Workflow|CCG-PDC-Workflow|19990927110328||ORU^R01|19990927110328|P|2.4\r";
	private static final Log ourLog = LogFactory.getLog(VariesTest.class);
	private EncodingCharacters ec = null;
	private final EncodingCharacters encoders;
	private final Message mshOnly;
	private Parser parser = null;

	/**
	 * Creates a new instance of VariesTest
	 */
	public VariesTest(String arg) throws EncodingNotSupportedException,
			HL7Exception {
		super(arg);

		mshOnly = new PipeParser().parse(MSH_TEXT);
		mshOnly.setParser(new GenericParser());
		encoders = EncodingCharacters.getInstance(mshOnly);

	}

	public void setUp() throws Exception {
		this.parser = new DefaultXMLParser();
		ec = new EncodingCharacters('|', "^~\\&");
	}

	/**
	 * AD = ST, ST,...
	 */
	public void testObx5WithExpectedComponentExpectedComponent()
			throws HL7Exception {

		// Message is stripped down
		String msgString = "MSH|^~\\&\r" // -
				+ "OBR|\r" // -
				+ "OBX||AD|||F1C1^F2C1\r";

		ORU_R01 msg = new ORU_R01();
		msg.parse(msgString);

		String encode = msg.encode();
		ourLog.info("\n\n" + encode);

		Varies observationValue = msg.getPATIENT_RESULT(0)
				.getORDER_OBSERVATION(0).getOBSERVATION().getOBX()
				.getObx5_ObservationValue(0);
		AD obx5 = (AD) observationValue.getData();
		assertEquals("F1C1", obx5.getAd1_StreetAddress().getValue());
		assertEquals("F2C1", obx5.getAd2_OtherDesignation().getValue());

		assertEquals(msgString.trim(), encode.trim());

	}

	/**
	 * CD = WVI, WVS,...<br/>
	 * WVI = NM, ST
	 */
	public void testObx5WithExpectedComponentExpectedSubcomponent()
			throws HL7Exception {

		// Message is stripped down
		String msgString = "MSH|^~\\&\r" // -
				+ "OBR|\r" // -
				+ "OBX||CD|||F1C1&F1C2\r";

		ORU_R01 msg = new ORU_R01();
		msg.parse(msgString);

		String encode = msg.encode();
		ourLog.info("\n\n" + encode);

		Varies observationValue = msg.getPATIENT_RESULT(0)
				.getORDER_OBSERVATION(0).getOBSERVATION().getOBX()
				.getObx5_ObservationValue(0);
		CD obx5 = (CD) observationValue.getData();
		assertEquals("F1C1", obx5.getCd1_ChannelIdentifier()
				.getWvi1_ChannelNumber().getValue());
		assertEquals("F1C2", obx5.getCd1_ChannelIdentifier()
				.getWvi2_ChannelName().getValue());

		assertEquals(msgString.trim(), encode.trim());

	}

	/**
	 * AD = ST, ST,...
	 */
	public void testObx5WithExpectedComponentUnpexpectedSubcomponentExpectedComponent()
			throws HL7Exception {

		// Message is stripped down
		String msgString = "MSH|^~\\&\r" // -
				+ "OBR|\r" // -
				+ "OBX||AD|||F1C1&F1C2^F2C1\r";

		ORU_R01 msg = new ORU_R01();
		msg.parse(msgString);

		String encode = msg.encode();
		ourLog.info("\n\n" + encode);

		Varies observationValue = msg.getPATIENT_RESULT(0)
				.getORDER_OBSERVATION(0).getOBSERVATION().getOBX()
				.getObx5_ObservationValue(0);
		AD obx5 = (AD) observationValue.getData();
		assertEquals("F1C1", obx5.getAd1_StreetAddress().getValue());
		assertEquals("F1C2", obx5.getAd1_StreetAddress().getExtraComponents().getComponent(0).encode());
		assertEquals("F2C1", obx5.getAd2_OtherDesignation().getValue());

		assertEquals(msgString.trim(), encode.trim());

	}

	/**
	 * AD = ST, ST,...
	 */
	public void testObx5WithExpectedComponentUnpexpectedSubcomponentWithinPrimitive()
			throws HL7Exception {

		// Message is stripped down
		String msgString = "MSH|^~\\&\r" // -
				+ "OBR|\r" // -
				+ "OBX||ST|||F1C1&F1C2\r";

		System.setProperty(Varies.ESCAPE_SUBCOMPONENT_DELIM_IN_PRIMITIVE, "TRUE");
		
		ORU_R01 msg = new ORU_R01();
		msg.parse(msgString);

		String encode = msg.encode();
		ourLog.info("\n\n" + encode);

		Varies observationValue = msg.getPATIENT_RESULT(0)
				.getORDER_OBSERVATION(0).getOBSERVATION().getOBX()
				.getObx5_ObservationValue(0);
		ST obx5 = (ST) observationValue.getData();
		assertEquals("F1C1&F1C2", obx5.getValue());

		String expected = "MSH|^~\\&\r" // -
			+ "OBR|\r" // -
			+ "OBX||ST|||F1C1\\T\\F1C2\r";
		
		assertEquals(expected.trim(), encode.trim());

	}

	
	/**
	 * AD = ST, ST,...
	 */
	public void testObx5WithExpectedComponentUnpexpectedSubcomponentWithinPrimitiveWithoutEscaping()
			throws HL7Exception {

		// Message is stripped down
		String msgString = "MSH|^~\\&\r" // -
				+ "OBR|\r" // -
				+ "OBX||ST|||F1C1&F1C2\r";

		System.setProperty(Varies.ESCAPE_SUBCOMPONENT_DELIM_IN_PRIMITIVE, "FALSE");
		
		ORU_R01 msg = new ORU_R01();
		msg.parse(msgString);

		String encode = msg.encode();
		ourLog.info("\n\n" + encode);

		Varies observationValue = msg.getPATIENT_RESULT(0)
				.getORDER_OBSERVATION(0).getOBSERVATION().getOBX()
				.getObx5_ObservationValue(0);
		ST obx5 = (ST) observationValue.getData();
		String actual = obx5.getValue();
		assertEquals("F1C1", actual);

		actual = obx5.encode();
		ourLog.info("Actual: " + actual);
		assertEquals("F1C1^F1C2", actual);
		
		String expected = "MSH|^~\\&\r" // -
			+ "OBR|\r" // -
			+ "OBX||ST|||F1C1^F1C2\r";

		String trim = encode.trim();
		ourLog.info("Encoded: " + trim.replace("\r", "\n"));
		
		assertEquals(expected.trim(), trim);

	}

	
	public void testParse() throws HL7Exception {
		Varies varies = new Varies(null);
		CE data = new CE(null);
		data.getIdentifier().setValue("foo");
		varies.setData(data);

		// existing primitive will NOT copy over "foo" as of Nov 2002
		assertEquals("foo", ((CE) varies.getData()).getIdentifier().getValue());
		data.getIdentifier().setValue("foo");
		assertEquals("foo", ((CE) varies.getData()).getIdentifier().getValue());
	}

	public void testParser_VariesCompositeSecondField() throws HL7Exception {
		final Varies variesField = new Varies(mshOnly);
		final Parser parser = mshOnly.getParser();
		final String hl7Text = "^A3456&2Rozwick&don&FURTHER&JR&MR&DR^200701010933^200702031022";
		parser.parse(variesField, hl7Text, encoders);
		final Composite data = (Composite) variesField.getData();
		assertEquals(4, data.getComponents().length);

		final Varies component = (Varies) data.getComponent(1);
		assertEquals(GenericComposite.class, component.getData().getClass());
		assertEquals(7,
				((Composite) component.getData()).getComponents().length);
		assertEquals(hl7Text, PipeParser.encode(variesField, encoders));
	}

	public void testParserParse_VariesCompositeFirstComponent()
			throws HL7Exception {

		final Varies variesField = new Varies(mshOnly);
		final Parser parser = mshOnly.getParser();
		final String hl7Text = "A3456&2Rozwick&don&FURTHER&JR&MR&DR^200701010933^200702031022";
		parser.parse(variesField, hl7Text, encoders);
		final Composite data = (Composite) variesField.getData();
		assertEquals(3, data.getComponents().length);

		final Varies component = (Varies) data.getComponent(0);
		assertEquals(GenericComposite.class, component.getData().getClass());
		assertEquals(7,
				((Composite) component.getData()).getComponents().length);
		assertEquals(hl7Text, PipeParser.encode(variesField, encoders));
	}

	public void testTerserGetPrimitive_VariesCompositeFirstComponent()
			throws HL7Exception {
		final Varies variesField = new Varies(mshOnly);
		final Primitive comp1Sub1 = Terser.getPrimitive(variesField, 1, 1);
		comp1Sub1.setValue("comp1Sub1");
		final Primitive comp1Sub2 = Terser.getPrimitive(variesField, 1, 2);
		comp1Sub2.setValue("comp1Sub2");
		final Primitive comp2Sub1 = Terser.getPrimitive(variesField, 2, 1);
		comp2Sub1.setValue("comp2Sub1");

		assertEquals(GenericComposite.class, variesField.getData().getClass());

		final GenericComposite field = (GenericComposite) variesField.getData();
		final Type typeComp1 = field.getComponent(0);
		assertEquals(typeComp1.getClass(), Varies.class);

		final Varies variesComp1 = (Varies) typeComp1;
		assertEquals(GenericComposite.class, variesComp1.getData().getClass());

		final GenericComposite comp1 = (GenericComposite) variesComp1.getData();
		final Type typeSub1 = comp1.getComponent(1);
		assertEquals(typeSub1.getClass(), Varies.class);

		final Varies variesSub1 = (Varies) typeSub1;
		assertEquals(GenericPrimitive.class, variesSub1.getData().getClass());

		assertEquals("comp1Sub1&comp1Sub2^comp2Sub1",
				PipeParser.encode(variesField, encoders));
	}

	public void testTerserGetPrimitive_VariesCompositeSecondComponent()
			throws HL7Exception {
		final Varies variesField = new Varies(mshOnly);
		final Primitive comp1Sub1 = Terser.getPrimitive(variesField, 1, 1);
		comp1Sub1.setValue("comp1Sub1");
		final Primitive comp2Sub1 = Terser.getPrimitive(variesField, 2, 1);
		comp2Sub1.setValue("comp2Sub1");
		final Primitive comp2Sub2 = Terser.getPrimitive(variesField, 2, 2);
		comp2Sub2.setValue("comp2Sub2");

		assertEquals(GenericComposite.class, variesField.getData().getClass());

		final GenericComposite field = (GenericComposite) variesField.getData();
		final Type typeComp1 = field.getComponent(0);
		assertEquals(typeComp1.getClass(), Varies.class);

		final Varies variesComp1 = (Varies) typeComp1;
		assertEquals(GenericPrimitive.class, variesComp1.getData().getClass());

		assertEquals("comp1Sub1^comp2Sub1&comp2Sub2",
				PipeParser.encode(variesField, encoders));
	}

	/**
	 * This is from a reported message that was kind of malformed to begin with,
	 * but became more malformed in processing.
	 * 
	 * Parse method should clear all existing data including extra
	 * components.
	 */
	public void testVariesParseMethodClearsExistingExtraComponents()
			throws HL7Exception {

		// Message is stripped down
		String msgString = "MSH|^~\\&\r" // -
				+ "OBR|\r" // -
				+ "OBX||FN|||OF1C1^OF2C1&OF2C2\r";

		ORU_R01 msg = new ORU_R01();
		msg.parse(msgString);

		OBX obx = msg.getPATIENT_RESULT().getORDER_OBSERVATION().getOBSERVATION().getOBX();
		Varies varies = obx.getObx5_ObservationValue(0);
		varies.parse("NF1^NF2");
		
		String encode = msg.encode();
		ourLog.info("\n\n" + encode);
		
		Assert.assertEquals("NF1^NF2", varies.encode());


	}

	
}