| OLD | NEW |
| 1 // Protocol Buffers - Google's data interchange format | 1 // Protocol Buffers - Google's data interchange format |
| 2 // Copyright 2008 Google Inc. All rights reserved. | 2 // Copyright 2008 Google Inc. All rights reserved. |
| 3 // https://developers.google.com/protocol-buffers/ | 3 // https://developers.google.com/protocol-buffers/ |
| 4 // | 4 // |
| 5 // Redistribution and use in source and binary forms, with or without | 5 // Redistribution and use in source and binary forms, with or without |
| 6 // modification, are permitted provided that the following conditions are | 6 // modification, are permitted provided that the following conditions are |
| 7 // met: | 7 // met: |
| 8 // | 8 // |
| 9 // * Redistributions of source code must retain the above copyright | 9 // * Redistributions of source code must retain the above copyright |
| 10 // notice, this list of conditions and the following disclaimer. | 10 // notice, this list of conditions and the following disclaimer. |
| (...skipping 10 matching lines...) Expand all Loading... |
| 21 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR | 21 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| 22 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT | 22 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| 23 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, | 23 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| 24 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT | 24 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| 25 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, | 25 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| 26 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY | 26 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| 27 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | 27 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| 28 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | 28 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| 29 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | 29 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| 30 | 30 |
| 31 package com.google.protobuf; | 31 /* This code will be inserted into generated code for |
| 32 | 32 * google/protobuf/any.proto. */ |
| 33 import junit.framework.TestCase; | |
| 34 | |
| 35 import java.io.ByteArrayOutputStream; | |
| 36 import java.io.IOException; | |
| 37 import java.nio.ByteBuffer; | |
| 38 import java.util.Arrays; | |
| 39 import java.util.Random; | |
| 40 | 33 |
| 41 /** | 34 /** |
| 42 * Tests for {@link ByteBufferWriter}. | 35 * Returns the type name contained in this instance, if any. |
| 36 * @return {string|undefined} |
| 43 */ | 37 */ |
| 44 public class ByteBufferWriterTest extends TestCase { | 38 proto.google.protobuf.Any.prototype.getTypeName = function() { |
| 39 return this.getTypeUrl().split('/').pop(); |
| 40 }; |
| 45 | 41 |
| 46 public void testHeapBuffer() throws IOException { | 42 |
| 47 // Test a small and large buffer. | 43 /** |
| 48 testWrite(ByteBuffer.allocate(100)); | 44 * Packs the given message instance into this Any. |
| 49 testWrite(ByteBuffer.allocate(1024 * 100)); | 45 * @param {!Uint8Array} serialized The serialized data to pack. |
| 46 * @param {string} name The type name of this message object. |
| 47 * @param {string=} opt_typeUrlPrefix the type URL prefix. |
| 48 */ |
| 49 proto.google.protobuf.Any.prototype.pack = function(serialized, name, |
| 50 opt_typeUrlPrefix) { |
| 51 if (!opt_typeUrlPrefix) { |
| 52 opt_typeUrlPrefix = 'type.googleapis.com/'; |
| 50 } | 53 } |
| 51 | 54 |
| 52 public void testDirectBuffer() throws IOException { | 55 if (opt_typeUrlPrefix.substr(-1) != '/') { |
| 53 // Test a small and large buffer. | 56 this.setTypeUrl(opt_typeUrlPrefix + '/' + name); |
| 54 testWrite(ByteBuffer.allocateDirect(100)); | 57 } else { |
| 55 testWrite(ByteBuffer.allocateDirect(1024 * 100)); | 58 this.setTypeUrl(opt_typeUrlPrefix + name); |
| 56 } | 59 } |
| 57 | 60 |
| 58 private void testWrite(ByteBuffer buffer) throws IOException { | 61 this.setValue(serialized); |
| 59 fillRandom(buffer); | 62 }; |
| 60 ByteArrayOutputStream os = new ByteArrayOutputStream(buffer.remaining()); | 63 |
| 61 ByteBufferWriter.write(buffer, os); | 64 |
| 62 assertEquals(0, buffer.position()); | 65 /** |
| 63 assertTrue(Arrays.equals(toArray(buffer), os.toByteArray())); | 66 * @template T |
| 67 * Unpacks this Any into the given message object. |
| 68 * @param {function(Uint8Array):T} deserialize Function that will deserialize |
| 69 * the binary data properly. |
| 70 * @param {string} name The expected type name of this message object. |
| 71 * @return {?T} If the name matched the expected name, returns the deserialized |
| 72 * object, otherwise returns undefined. |
| 73 */ |
| 74 proto.google.protobuf.Any.prototype.unpack = function(deserialize, name) { |
| 75 if (this.getTypeName() == name) { |
| 76 return deserialize(this.getValue_asU8()); |
| 77 } else { |
| 78 return null; |
| 64 } | 79 } |
| 65 | 80 }; |
| 66 private void fillRandom(ByteBuffer buf) { | |
| 67 byte[] bytes = new byte[buf.remaining()]; | |
| 68 new Random().nextBytes(bytes); | |
| 69 buf.put(bytes); | |
| 70 buf.flip(); | |
| 71 return; | |
| 72 } | |
| 73 | |
| 74 private byte[] toArray(ByteBuffer buf) { | |
| 75 int originalPosition = buf.position(); | |
| 76 byte[] bytes = new byte[buf.remaining()]; | |
| 77 buf.get(bytes); | |
| 78 buf.position(originalPosition); | |
| 79 return bytes; | |
| 80 } | |
| 81 } | |
| OLD | NEW |