package org.commcare.xml;

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

import org.commcare.suite.model.Text;
import org.commcare.suite.model.RemoteQueryDatum;
import org.javarosa.xml.util.InvalidStructureException;
import org.javarosa.xml.util.UnfullfilledRequirementsException;
import org.javarosa.xpath.expr.XPathExpression;
import org.junit.Test;
import org.xmlpull.v1.XmlPullParserException;

import java.io.IOException;
import java.util.Hashtable;
import java.util.List;

/**
 * Low level tests for session parsing
 */
public class SessionDatumParserTest {

    @Test
    public void testSessionDatumParser()
            throws UnfullfilledRequirementsException, InvalidStructureException, XmlPullParserException,
            IOException {
        String query = "<query url=\"https://www.fake.com/patient_search/\" storage-instance=\"patients\">"
                + "<data key=\"device_id\" ref=\"instance('session')/session/context/deviceid\"/>"
                + "</query>";
        SessionDatumParser parser = ParserTestUtils.buildParser(query, SessionDatumParser.class);
        RemoteQueryDatum datum = (RemoteQueryDatum) parser.parse();
        Hashtable<String, XPathExpression> hiddenQueryValues = datum.getHiddenQueryValues();

        assertEquals(1, hiddenQueryValues.size());
        assertTrue(hiddenQueryValues.containsKey("device_id"));
    }

    @Test
    public void testSessionDatumParser__withTitle()
            throws UnfullfilledRequirementsException, InvalidStructureException, XmlPullParserException,
            IOException {
        String query = "<query url=\"https://www.fake.com/patient_search/\" storage-instance=\"patients\">"
                + "<title>"
                + "<text>"
                + "<locale id=\"locale_id\"/>"
                + "</text>"
                + "</title>"
                + "<data key=\"device_id\" ref=\"instance('session')/session/context/deviceid\"/>"
                + "</query>";
        SessionDatumParser parser = ParserTestUtils.buildParser(query, SessionDatumParser.class);
        RemoteQueryDatum datum = (RemoteQueryDatum) parser.parse();
        String title = datum.getTitleText().getArgument();
        Hashtable<String, XPathExpression> hiddenQueryValues = datum.getHiddenQueryValues();

        assertEquals(1, hiddenQueryValues.size());
        assertTrue(hiddenQueryValues.containsKey("device_id"));
        assertEquals("locale_id", title);
    }
}

@@ -141,20 +141,32 @@
 e.model.
-Text
+RemoteQueryDatum
 ;%0Aimport
@@ -191,32 +191,20 @@
 e.model.
-RemoteQueryDatum
+Text
 ;%0Aimport

Update SessionDatumParserTest.java