1 files changed, 16 insertions, 10 deletions
diff --git a/base/pickle_unittest.cc b/base/pickle_unittest.cc
index 6780e21..4c38677 100644
--- a/base/pickle_unittest.cc
+++ b/base/pickle_unittest.cc
@@ -26,14 +26,13 @@
 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
-#include <windows.h>
-#include <string.h>
+
+#include <string>
 
 #include "base/basictypes.h"
-#include "base/check_handler.h"
 #include "base/logging.h"
-#include "base/scoped_ptr.h"
 #include "base/pickle.h"
+#include "base/scoped_ptr.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
@@ -94,10 +93,13 @@ TEST(PickleTest, EncodeDecode) {
   EXPECT_TRUE(pickle.WriteBool(testbool2));
   EXPECT_TRUE(pickle.WriteData(testdata, testdatalen));
 
-  char* dest = pickle.BeginWriteData(testdatalen);
+  // Over allocate BeginWriteData so we can test TrimWriteData.
+  char* dest = pickle.BeginWriteData(testdatalen + 100);
   EXPECT_TRUE(dest);
   memcpy(dest, testdata, testdatalen);
 
+  pickle.TrimWriteData(testdatalen);
+
   VerifyResult(pickle);
 
   // test copy constructor
@@ -206,11 +208,15 @@ TEST(PickleTest, Resize) {
   EXPECT_EQ(cur_payload, pickle.payload_size());
 }
 
-TEST(PickleTest, HeaderPadding) {
-  struct CustomHeader : Pickle::Header {
-    int blah;
-  };
+namespace {
 
+struct CustomHeader : Pickle::Header {
+  int blah;
+};
+
+}  // namespace
+
+TEST(PickleTest, HeaderPadding) {
   const uint32 kMagic = 0x12345678;
 
   Pickle pickle(sizeof(CustomHeader));
@@ -235,4 +241,4 @@ TEST(PickleTest, EqualsOperator) {
   Pickle copy;
   copy = copy_refs_source_buffer;
   ASSERT_EQ(source.size(), copy.size());
-}
-\ No newline at end of file
+}