diff options
| author | Keith Wall <kwall@apache.org> | 2012-01-07 09:44:58 +0000 |
|---|---|---|
| committer | Keith Wall <kwall@apache.org> | 2012-01-07 09:44:58 +0000 |
| commit | c44ae41105668027766723033c6f5325366524f1 (patch) | |
| tree | b966e092913dcaf2e53199608a5020a240ec7878 /qpid/java/common | |
| parent | a6bf49a9c8b50a48afbee851edb5fe51e573af03 (diff) | |
| download | qpid-python-c44ae41105668027766723033c6f5325366524f1.tar.gz | |
QPID-3715: Fix the receiving of an empty ObjectMessage (without a body)
Applied patch from Oleksandr Rudyy<orudyy@gmail.com>
git-svn-id: https://svn.apache.org/repos/asf/qpid/trunk@1228583 13f79535-47bb-0310-9956-ffa450edef68
Diffstat (limited to 'qpid/java/common')
| -rw-r--r-- | qpid/java/common/src/main/java/org/apache/qpid/util/ByteBufferInputStream.java | 95 | ||||
| -rw-r--r-- | qpid/java/common/src/test/java/org/apache/qpid/util/ByteBufferInputStreamTest.java | 111 |
2 files changed, 206 insertions, 0 deletions
diff --git a/qpid/java/common/src/main/java/org/apache/qpid/util/ByteBufferInputStream.java b/qpid/java/common/src/main/java/org/apache/qpid/util/ByteBufferInputStream.java new file mode 100644 index 0000000000..b72b342187 --- /dev/null +++ b/qpid/java/common/src/main/java/org/apache/qpid/util/ByteBufferInputStream.java @@ -0,0 +1,95 @@ +/* + * + * Licensed to 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 org.apache.qpid.util; + +import java.io.IOException; +import java.io.InputStream; +import java.nio.ByteBuffer; + +/** + * Wraps @link {@link ByteBuffer} into {@link InputStream} + */ +public class ByteBufferInputStream extends InputStream +{ + private final ByteBuffer _buffer; + + public ByteBufferInputStream(ByteBuffer buffer) + { + _buffer = buffer; + } + + @Override + public int read() throws IOException + { + if (_buffer.hasRemaining()) + { + return _buffer.get() & 0xFF; + } + return -1; + } + + @Override + public int read(byte[] b, int off, int len) throws IOException + { + if (!_buffer.hasRemaining()) + { + return -1; + } + if(_buffer.remaining() < len) + { + len = _buffer.remaining(); + } + _buffer.get(b, off, len); + + return len; + } + + @Override + public void mark(int readlimit) + { + _buffer.mark(); + } + + @Override + public void reset() throws IOException + { + _buffer.reset(); + } + + @Override + public boolean markSupported() + { + return true; + } + + @Override + public long skip(long n) throws IOException + { + _buffer.position(_buffer.position()+(int)n); + return n; + } + + @Override + public int available() throws IOException + { + return _buffer.remaining(); + } +} diff --git a/qpid/java/common/src/test/java/org/apache/qpid/util/ByteBufferInputStreamTest.java b/qpid/java/common/src/test/java/org/apache/qpid/util/ByteBufferInputStreamTest.java new file mode 100644 index 0000000000..0b393a489f --- /dev/null +++ b/qpid/java/common/src/test/java/org/apache/qpid/util/ByteBufferInputStreamTest.java @@ -0,0 +1,111 @@ +/* + * + * Licensed to 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 org.apache.qpid.util; + +import java.io.IOException; +import java.nio.ByteBuffer; +import java.util.Arrays; + +import junit.framework.TestCase; + +public class ByteBufferInputStreamTest extends TestCase +{ + private byte[] _data = {2, 1, 5, 3, 4}; + private ByteBufferInputStream _inputStream; + + public void setUp() throws Exception + { + _inputStream = new ByteBufferInputStream(ByteBuffer.wrap(_data)); + } + + public void testRead() throws IOException + { + for (int i = 0; i < _data.length; i++) + { + assertEquals("Unexpected byte at position " + i, _data[i], _inputStream.read()); + } + assertEquals("EOF not reached", -1, _inputStream.read()); + } + + public void testReadByteArray() throws IOException + { + byte[] readBytes = new byte[_data.length]; + int length = _inputStream.read(readBytes, 0, 2); + + byte[] expected = new byte[_data.length]; + System.arraycopy(_data, 0, expected, 0, 2); + + assertTrue("Unexpected data", Arrays.equals(expected, readBytes)); + assertEquals("Unexpected length", 2, length); + + length = _inputStream.read(readBytes, 2, 3); + + assertTrue("Unexpected data", Arrays.equals(_data, readBytes)); + assertEquals("Unexpected length", 3, length); + + length = _inputStream.read(readBytes); + assertEquals("EOF not reached", -1, length); + } + + public void testSkip() throws IOException + { + _inputStream.skip(3); + byte[] readBytes = new byte[_data.length - 3]; + int length = _inputStream.read(readBytes); + + byte[] expected = new byte[_data.length - 3]; + System.arraycopy(_data, 3, expected, 0, _data.length - 3); + + assertTrue("Unexpected data", Arrays.equals(expected, readBytes)); + assertEquals("Unexpected length", _data.length - 3, length); + } + + public void testAvailable() throws IOException + { + int available = _inputStream.available(); + assertEquals("Unexpected number of available bytes", _data.length, available); + byte[] readBytes = new byte[_data.length]; + _inputStream.read(readBytes); + available = _inputStream.available(); + assertEquals("Unexpected number of available bytes", 0, available); + } + + public void testMarkReset() throws IOException + { + _inputStream.mark(0); + byte[] readBytes = new byte[_data.length]; + int length = _inputStream.read(readBytes); + assertEquals("Unexpected length", _data.length, length); + assertEquals("Unexpected number of available bytes", 0, _inputStream.available()); + + _inputStream.reset(); + readBytes = new byte[_data.length]; + length = _inputStream.read(readBytes); + assertEquals("Unexpected length", _data.length, length); + assertEquals("Unexpected number of available bytes", 0, _inputStream.available()); + } + + public void testMarkSupported() throws IOException + { + assertTrue("Unexpected mark supported", _inputStream.markSupported()); + } + +} |
