BufferedSegmentReaderTest.java

/*
	Milyn - Copyright (C) 2006 - 2010

	This library is free software; you can redistribute it and/or
	modify it under the terms of the GNU Lesser General Public
	License (version 2.1) as published by the Free Software 
	Foundation.

	This library is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  
    
	See the GNU Lesser General Public License for more details:    
	http://www.gnu.org/licenses/lgpl.txt
*/

package org.milyn.edisax;

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

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.util.Arrays;

import org.apache.commons.lang.StringUtils;
import org.junit.Test;
import org.milyn.edisax.model.internal.Delimiters;
import org.xml.sax.InputSource;

/**
 * @author tfennelly
 */
public class BufferedSegmentReaderTest {

    @Test	
    public void test() throws IOException {
		test("111111111ff22222222222ff333333333ff4444444444f4444444fff5555555555", "ff", "|",
			new String[] {"111111111", "22222222222", "333333333", "4444444444f4444444", "f5555555555"});

		test("a", "ff", "|", new String[] {"a"});

		test("ff", "ff", "*",  new String[] {});

		test("111111111\n22222222222\n333333333\n4444444444f4444444\nf5555555555", "\n", "*",
				new String[] {"111111111", "22222222222", "333333333", "4444444444f4444444", "f5555555555"});
	}

    @Test
    public void test_ignore_cr_lf() throws IOException {
        String fieldDelimiter = "*";
        String segmentDelimiter = "'!$";
        String edi1 = "SEG0*1*2**4*5'\nSEG1*1*2*3'\r\nSEG2*1*2*3*4'";
        String edi2 = "SEG0*1*2**4*5'SEG1*1*2*3'SEG2*1*2*3*4'";

        // Check that BufferedSegmentReader reads all segments when !$ exists at end of segmentdelimiter.
        BufferedSegmentReader reader = createSegmentReader(edi1, segmentDelimiter, fieldDelimiter);
        int segIndex = 0;
        while(reader.moveToNextSegment()) {
			assertEquals("Segment comparison failure.", "SEG" + segIndex, reader.getCurrentSegmentFields()[0]);
			segIndex++;
		}
        assertTrue("The number of segments read should be three", segIndex-1 == 2);

        // Check that BufferedSegmentReader reads all segments when !$ exists at end of segmentdelimiter and there are no newlines.
        reader = createSegmentReader(edi2, segmentDelimiter, fieldDelimiter);
        segIndex = 0;
        while(reader.moveToNextSegment()) {
			assertEquals("Segment comparison failure.", "SEG" + segIndex, reader.getCurrentSegmentFields()[0]);
			segIndex++;
		}
        assertTrue("The number of segments read should be three", segIndex-1 == 2);

        // Check that BufferedSegmentReader can peek through \r or \n characters
        reader = createSegmentReader(edi1, segmentDelimiter, fieldDelimiter);
        segIndex = 0;
        String peek = reader.peek(edi1.length());
        assertTrue("Peek should return all characters but \n and \r", peek.equals(edi2));
        
    }

    @Test
    public void test_not_ignore_cr_lf() throws IOException {
        String fieldDelimiter = "*";
        String segmentDelimiter = "'";
        String edi1 = "SEG0*1*2**4*5'\nSEG1*1*2*3'\r\nSEG2*1*2*3*4'";
        String edi2 = "SEG0*1*2**4*5'SEG1*1*2*3'SEG2*1*2*3*4'";

        // Check that BufferedSegmentReader reads all segments when !$ exists at end of segmentdelimiter.
        BufferedSegmentReader reader = createSegmentReader(edi1, segmentDelimiter, fieldDelimiter);
        int segIndex = 0;
        while(reader.moveToNextSegment()) {
            if (segIndex == 0) {
			    assertEquals("Segment comparison failure.", "SEG" + segIndex, reader.getCurrentSegmentFields()[0]);
            } else if (segIndex == 1) {
                assertEquals("Segment comparison failure.", "SEG" + segIndex, reader.getCurrentSegmentFields()[0]);
            } else if (segIndex == 2) {
                assertEquals("Segment comparison failure.", "SEG" + segIndex, reader.getCurrentSegmentFields()[0]);
            } else {
                assertTrue("More segments than expected in test case.", false);
            }
			segIndex++;
		}
        assertTrue("The number of segments read should be three", segIndex-1 == 2);

        // Check that BufferedSegmentReader reads all segments when !$ exists at end of segmentdelimiter and there are no newlines.
        reader = createSegmentReader(edi2, segmentDelimiter, fieldDelimiter);
        segIndex = 0;
        while(reader.moveToNextSegment()) {
			assertEquals("Segment comparison failure.", "SEG" + segIndex, reader.getCurrentSegmentFields()[0]);
			segIndex++;
		}
        assertTrue("The number of segments read should be three", segIndex-1 == 2);

    }

	private void test(String input, String segmentDelim, String fieldDelim, String[] segments) throws IOException {
        BufferedSegmentReader reader = createSegmentReader(input, segmentDelim, fieldDelim);
		int segIndex = 0;
		
		while(segIndex < segments.length && reader.moveToNextSegment()) {
			String segment = reader.getSegmentBuffer().toString();
			assertEquals("Segment comparison failure.", segments[segIndex], segment);
			segIndex++;
		}
		
		assertEquals("All segments not read.", segments.length, segIndex);
	}

    private BufferedSegmentReader createSegmentReader(String input, String segmentDelim, String fieldDelim) {
        InputSource inputSource = new InputSource(new ByteArrayInputStream(input.getBytes()));
        Delimiters delimiters = new Delimiters().setSegment(segmentDelim).setField(fieldDelim);
        delimiters.setEscape("?");
        BufferedSegmentReader reader = new BufferedSegmentReader(inputSource, delimiters);
        return reader;
    }

    public void test_split() {
		Arrays.asList(StringUtils.splitPreserveAllTokens("a*b***C*d", "*"));
	}
}