e21e038993251498308f26e1c134d8a8bf33fa54
[tools/dynpart-tools.git] /
1 // Protocol Buffers - Google's data interchange format
2 // Copyright 2008 Google Inc.  All rights reserved.
3 // http://code.google.com/p/protobuf/
4 //
5 // Redistribution and use in source and binary forms, with or without
6 // modification, are permitted provided that the following conditions are
7 // met:
8 //
9 //     * Redistributions of source code must retain the above copyright
10 // notice, this list of conditions and the following disclaimer.
11 //     * Redistributions in binary form must reproduce the above
12 // copyright notice, this list of conditions and the following disclaimer
13 // in the documentation and/or other materials provided with the
14 // distribution.
15 //     * Neither the name of Google Inc. nor the names of its
16 // contributors may be used to endorse or promote products derived from
17 // this software without specific prior written permission.
18 //
19 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
22 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
23 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
24 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
25 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
26 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
27 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
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.
30
31 package com.google.protobuf.test;
32 import com.google.protobuf.*;
33
34
35 import protobuf_unittest.UnittestProto;
36
37 import junit.framework.TestCase;
38
39 import java.io.IOException;
40
41 /**
42  * Tests to make sure the lazy conversion of UTF8-encoded byte arrays to
43  * strings works correctly.
44  *
45  * @author jonp@google.com (Jon Perlow)
46  */
47 public class LazyStringEndToEndTest extends TestCase {
48
49   private static ByteString TEST_ALL_TYPES_SERIALIZED_WITH_ILLEGAL_UTF8 =
50       ByteString.copyFrom(new byte[] {
51           114, 4, -1, 0, -1, 0, -30, 2, 4, -1,
52           0, -1, 0, -30, 2, 4, -1, 0, -1, 0, });
53
54   private ByteString encodedTestAllTypes;
55
56   @Override
57   protected void setUp() throws Exception {
58     super.setUp();
59     this.encodedTestAllTypes = UnittestProto.TestAllTypes.newBuilder()
60         .setOptionalString("foo")
61         .addRepeatedString("bar")
62         .addRepeatedString("baz")
63         .build()
64         .toByteString();
65   }
66
67   /**
68    * Tests that an invalid UTF8 string will roundtrip through a parse
69    * and serialization.
70    */
71   public void testParseAndSerialize() throws InvalidProtocolBufferException {
72     UnittestProto.TestAllTypes tV2 = UnittestProto.TestAllTypes.parseFrom(
73         TEST_ALL_TYPES_SERIALIZED_WITH_ILLEGAL_UTF8);
74     ByteString bytes = tV2.toByteString();
75     assertEquals(TEST_ALL_TYPES_SERIALIZED_WITH_ILLEGAL_UTF8, bytes);
76
77     tV2.getOptionalString();
78     bytes = tV2.toByteString();
79     assertEquals(TEST_ALL_TYPES_SERIALIZED_WITH_ILLEGAL_UTF8, bytes);
80   }
81
82   public void testParseAndWrite() throws IOException {
83     UnittestProto.TestAllTypes tV2 = UnittestProto.TestAllTypes.parseFrom(
84         TEST_ALL_TYPES_SERIALIZED_WITH_ILLEGAL_UTF8);
85     byte[] sink = new byte[TEST_ALL_TYPES_SERIALIZED_WITH_ILLEGAL_UTF8.size()];
86     CodedOutputStream outputStream = CodedOutputStream.newInstance(sink);
87     tV2.writeTo(outputStream);
88     outputStream.flush();
89     assertEquals(
90         TEST_ALL_TYPES_SERIALIZED_WITH_ILLEGAL_UTF8,
91         ByteString.copyFrom(sink));
92   }
93
94   public void testNoStringCachingIfOnlyBytesAccessed() throws Exception {
95     UnittestProto.TestAllTypes proto =
96         UnittestProto.TestAllTypes.parseFrom(encodedTestAllTypes);
97     ByteString optional = proto.getOptionalStringBytes();
98     assertSame(optional, proto.getOptionalStringBytes());
99     assertSame(optional, proto.toBuilder().getOptionalStringBytes());
100
101     ByteString repeated0 = proto.getRepeatedStringBytes(0);
102     ByteString repeated1 = proto.getRepeatedStringBytes(1);
103     assertSame(repeated0, proto.getRepeatedStringBytes(0));
104     assertSame(repeated1, proto.getRepeatedStringBytes(1));
105     assertSame(repeated0, proto.toBuilder().getRepeatedStringBytes(0));
106     assertSame(repeated1, proto.toBuilder().getRepeatedStringBytes(1));
107   }
108 }