/** * Copyright 2015 StreamSets Inc. * * Licensed under the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you under the Apache License, Version 2.0 (the * "License"); you may not use this file except in compliance * with the License. You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package com.streamsets.pipeline.lib.parser.binary; import com.streamsets.pipeline.api.OnRecordError; import com.streamsets.pipeline.api.Record; import com.streamsets.pipeline.api.Stage; import com.streamsets.pipeline.lib.parser.DataParser; import com.streamsets.pipeline.lib.parser.DataParserException; import com.streamsets.pipeline.sdk.ContextInfoCreator; import org.junit.Assert; import org.junit.Test; import java.io.ByteArrayInputStream; import java.io.IOException; import java.util.Arrays; import java.util.Collections; public class TestBinaryDataParser { private static final String TEST_STRING_251 = "StreamSets was founded in June 2014 by business and engineering " + "leaders in the data integration space with a history of bringing successful products to market. We're a " + "team that is laser-focused on solving hard problems so our customers don't have to."; private Stage.Context getContext() { return ContextInfoCreator.createSourceContext("i", false, OnRecordError.TO_ERROR, Collections.EMPTY_LIST); } private byte[] getTestBytes() throws Exception { return TEST_STRING_251.getBytes(); } @Test public void testParse() throws Exception { byte[] data = getTestBytes(); DataParser parser = new BinaryDataParser(getContext(), new ByteArrayInputStream(data), "myId", 1000); Assert.assertEquals(0, Long.parseLong(parser.getOffset())); Record record = parser.parse(); Assert.assertNotNull(record); Assert.assertTrue(Arrays.equals(TEST_STRING_251.getBytes(), record.get().getValueAsByteArray())); Assert.assertEquals(251, record.get().getValueAsByteArray().length); long offset = Long.parseLong(parser.getOffset()); Assert.assertEquals(251, offset); record = parser.parse(); Assert.assertNull(record); Assert.assertEquals(251, Long.parseLong(parser.getOffset())); parser.close(); } @Test(expected = DataParserException.class) public void testParseExceedsMaxDataLength() throws Exception { byte[] data = getTestBytes(); DataParser parser = new BinaryDataParser(getContext(), new ByteArrayInputStream(data), "myId", 159); Assert.assertEquals(0, Long.parseLong(parser.getOffset())); parser.parse(); } @Test(expected = IOException.class) public void testClose() throws Exception { byte[] data = getTestBytes(); DataParser parser = new BinaryDataParser(getContext(), new ByteArrayInputStream(data), "myId", 1000); parser.close(); parser.parse(); } }