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 |