third_party/WebKit/Source/wtf/ArrayBufferBuilderTest.cpp

   1 /*
   2  * Copyright (C) 2013 Google Inc. All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions are
   6  * met:
   7  *
   8  *     * Redistributions of source code must retain the above copyright
   9  * notice, this list of conditions and the following disclaimer.
  10  *     * Redistributions in binary form must reproduce the above
  11  * copyright notice, this list of conditions and the following disclaimer
  12  * in the documentation and/or other materials provided with the
  13  * distribution.
  14  *     * Neither the name of Google Inc. nor the names of its
  15  * contributors may be used to endorse or promote products derived from
  16  * this software without specific prior written permission.
  17  *
  18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  22  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  24  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  29  */
  30
  31 #include "config.h"
  32 #include "wtf/ArrayBufferBuilder.h"
  33
  34 #include "wtf/Assertions.h"
  35 #include <gtest/gtest.h>
  36 #include <limits.h>
  37 #include <string.h>
  38
  39 namespace WTF {
  40
  41 TEST(ArrayBufferBuilderTest, Constructor)
  42 {
  43     ArrayBufferBuilder zeroBuilder(0);
  44     EXPECT_EQ(0u, zeroBuilder.byteLength());
  45     EXPECT_EQ(0u, zeroBuilder.capacity());
  46
  47     ArrayBufferBuilder smallBuilder(1024);
  48     EXPECT_EQ(0u, zeroBuilder.byteLength());
  49     EXPECT_EQ(1024u, smallBuilder.capacity());
  50
  51     ArrayBufferBuilder bigBuilder(2048);
  52     EXPECT_EQ(0u, zeroBuilder.byteLength());
  53     EXPECT_EQ(2048u, bigBuilder.capacity());
  54 }
  55
  56 TEST(ArrayBufferBuilderTest, Append)
  57 {
  58     const char data[] = "HelloWorld";
  59     size_t dataSize = sizeof(data) - 1;
  60
  61     ArrayBufferBuilder builder(2 * dataSize);
  62
  63     EXPECT_EQ(dataSize, builder.append(data, dataSize));
  64     EXPECT_EQ(dataSize, builder.byteLength());
  65     EXPECT_EQ(dataSize * 2, builder.capacity());
  66
  67     EXPECT_EQ(dataSize, builder.append(data, dataSize));
  68     EXPECT_EQ(dataSize * 2, builder.byteLength());
  69     EXPECT_EQ(dataSize * 2, builder.capacity());
  70
  71     EXPECT_EQ(dataSize, builder.append(data, dataSize));
  72     EXPECT_EQ(dataSize * 3, builder.byteLength());
  73     EXPECT_GE(builder.capacity(), dataSize * 3);
  74 }
  75
  76 TEST(ArrayBufferBuilderTest, AppendRepeatedly)
  77 {
  78     const char data[] = "HelloWorld";
  79     size_t dataSize = sizeof(data) - 1;
  80
  81     ArrayBufferBuilder builder(37); // Some number coprime with dataSize.
  82
  83     for (size_t i = 1; i < 1000U; ++i) {
  84         EXPECT_EQ(dataSize, builder.append(data, dataSize));
  85         EXPECT_EQ(dataSize * i, builder.byteLength());
  86         EXPECT_GE(builder.capacity(), dataSize * i);
  87     }
  88 }
  89
  90 TEST(ArrayBufferBuilderTest, DefaultConstructorAndAppendRepeatedly)
  91 {
  92     const char data[] = "HelloWorld";
  93     size_t dataSize = sizeof(data) - 1;
  94
  95     ArrayBufferBuilder builder;
  96
  97     for (size_t i = 1; i < 4000U; ++i) {
  98         EXPECT_EQ(dataSize, builder.append(data, dataSize));
  99         EXPECT_EQ(dataSize * i, builder.byteLength());
 100         EXPECT_GE(builder.capacity(), dataSize * i);
 101     }
 102 }
 103
 104 TEST(ArrayBufferBuilderTest, AppendFixedCapacity)
 105 {
 106     const char data[] = "HelloWorld";
 107     size_t dataSize = sizeof(data) - 1;
 108
 109     ArrayBufferBuilder builder(15);
 110     builder.setVariableCapacity(false);
 111
 112     EXPECT_EQ(dataSize, builder.append(data, dataSize));
 113     EXPECT_EQ(dataSize, builder.byteLength());
 114     EXPECT_EQ(15u, builder.capacity());
 115
 116     EXPECT_EQ(5u, builder.append(data, dataSize));
 117     EXPECT_EQ(15u, builder.byteLength());
 118     EXPECT_EQ(15u, builder.capacity());
 119
 120     EXPECT_EQ(0u, builder.append(data, dataSize));
 121     EXPECT_EQ(15u, builder.byteLength());
 122     EXPECT_EQ(15u, builder.capacity());
 123 }
 124
 125 TEST(ArrayBufferBuilderTest, ToArrayBuffer)
 126 {
 127     const char data1[] = "HelloWorld";
 128     size_t data1Size = sizeof(data1) - 1;
 129
 130     const char data2[] = "GoodbyeWorld";
 131     size_t data2Size = sizeof(data2) - 1;
 132
 133     ArrayBufferBuilder builder(1024);
 134     builder.append(data1, data1Size);
 135     builder.append(data2, data2Size);
 136
 137     const char expected[] = "HelloWorldGoodbyeWorld";
 138     size_t expectedSize = sizeof(expected) - 1;
 139
 140     RefPtr<ArrayBuffer> result = builder.toArrayBuffer();
 141     ASSERT_EQ(data1Size + data2Size, result->byteLength());
 142     ASSERT_EQ(expectedSize, result->byteLength());
 143     EXPECT_EQ(0, memcmp(expected, result->data(), expectedSize));
 144 }
 145
 146 TEST(ArrayBufferBuilderTest, ToArrayBufferSameAddressIfExactCapacity)
 147 {
 148     const char data[] = "HelloWorld";
 149     size_t dataSize = sizeof(data) - 1;
 150
 151     ArrayBufferBuilder builder(dataSize);
 152     builder.append(data, dataSize);
 153
 154     RefPtr<ArrayBuffer> result1 = builder.toArrayBuffer();
 155     RefPtr<ArrayBuffer> result2 = builder.toArrayBuffer();
 156     EXPECT_EQ(result1.get(), result2.get());
 157 }
 158
 159 TEST(ArrayBufferBuilderTest, ToString)
 160 {
 161     const char data1[] = "HelloWorld";
 162     size_t data1Size = sizeof(data1) - 1;
 163
 164     const char data2[] = "GoodbyeWorld";
 165     size_t data2Size = sizeof(data2) - 1;
 166
 167     ArrayBufferBuilder builder(1024);
 168     builder.append(data1, data1Size);
 169     builder.append(data2, data2Size);
 170
 171     const char expected[] = "HelloWorldGoodbyeWorld";
 172     size_t expectedSize = sizeof(expected) - 1;
 173
 174     String result = builder.toString();
 175     EXPECT_EQ(expectedSize, result.length());
 176     for (unsigned i = 0; i < result.length(); ++i)
 177         EXPECT_EQ(expected[i], result[i]);
 178 }
 179
 180 TEST(ArrayBufferBuilderTest, ShrinkToFitNoAppend)
 181 {
 182     ArrayBufferBuilder builder(1024);
 183     EXPECT_EQ(1024u, builder.capacity());
 184     builder.shrinkToFit();
 185     EXPECT_EQ(0u, builder.byteLength());
 186     EXPECT_EQ(0u, builder.capacity());
 187 }
 188
 189 TEST(ArrayBufferBuilderTest, ShrinkToFit)
 190 {
 191     const char data[] = "HelloWorld";
 192     size_t dataSize = sizeof(data) - 1;
 193
 194     ArrayBufferBuilder builder(32);
 195
 196     EXPECT_EQ(dataSize, builder.append(data, dataSize));
 197     EXPECT_EQ(dataSize, builder.byteLength());
 198     EXPECT_EQ(32u, builder.capacity());
 199
 200     builder.shrinkToFit();
 201     EXPECT_EQ(dataSize, builder.byteLength());
 202     EXPECT_EQ(dataSize, builder.capacity());
 203 }
 204
 205 TEST(ArrayBufferBuilderTest, ShrinkToFitFullyUsed)
 206 {
 207     const char data[] = "HelloWorld";
 208     size_t dataSize = sizeof(data) - 1;
 209
 210     ArrayBufferBuilder builder(dataSize);
 211     const void* internalAddress = builder.data();
 212
 213     EXPECT_EQ(dataSize, builder.append(data, dataSize));
 214     EXPECT_EQ(dataSize, builder.byteLength());
 215     EXPECT_EQ(dataSize, builder.capacity());
 216
 217     builder.shrinkToFit();
 218     // Reallocation should not happen.
 219     EXPECT_EQ(internalAddress, builder.data());
 220     EXPECT_EQ(dataSize, builder.byteLength());
 221     EXPECT_EQ(dataSize, builder.capacity());
 222 }
 223
 224 TEST(ArrayBufferBuilderTest, ShrinkToFitAfterGrowth)
 225 {
 226     const char data[] = "HelloWorld";
 227     size_t dataSize = sizeof(data) - 1;
 228
 229     ArrayBufferBuilder builder(5);
 230
 231     EXPECT_EQ(dataSize, builder.append(data, dataSize));
 232     EXPECT_GE(builder.capacity(), dataSize);
 233     builder.shrinkToFit();
 234     EXPECT_EQ(dataSize, builder.byteLength());
 235     EXPECT_EQ(dataSize, builder.capacity());
 236 }
 237
 238 } // namespace WTF