OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/sessions/compress_data_helper.h" | 5 #include "chrome/browser/sessions/compress_data_helper.h" |
6 | 6 |
7 #include "base/pickle.h" | 7 #include "base/pickle.h" |
8 #include "testing/gtest/include/gtest/gtest.h" | 8 #include "testing/gtest/include/gtest/gtest.h" |
9 | 9 |
10 typedef testing::Test CompressDataHelperTest; | 10 typedef testing::Test CompressDataHelperTest; |
11 | 11 |
12 TEST_F(CompressDataHelperTest, CompressAndDecompressData) { | 12 TEST_F(CompressDataHelperTest, CompressAndDecompressData) { |
13 std::string data = "test data for compression and decompression"; | 13 std::string data = "test data for compression and decompression"; |
14 // Add NULL bytes. | 14 // Add NULL bytes. |
15 data[3] = data[8] = 0; | 15 data[3] = data[8] = 0; |
16 | 16 |
17 Pickle pickle; | 17 Pickle pickle; |
18 int bytes_written = 0; | 18 int bytes_written = 0; |
19 int max_bytes = 100; | 19 int max_bytes = 100; |
20 CompressDataHelper::CompressAndWriteStringToPickle( | 20 CompressDataHelper::CompressAndWriteStringToPickle( |
21 data, max_bytes, &pickle, &bytes_written); | 21 data, max_bytes, &pickle, &bytes_written); |
22 | 22 |
23 EXPECT_GT(bytes_written, 0); | 23 EXPECT_GT(bytes_written, 0); |
24 EXPECT_LT(bytes_written, max_bytes + 1); | 24 EXPECT_LT(bytes_written, max_bytes + 1); |
25 | 25 |
26 void* it = NULL; | 26 PickleIterator it(pickle); |
27 std::string data_out; | 27 std::string data_out; |
28 | 28 |
29 ASSERT_TRUE(CompressDataHelper::ReadAndDecompressStringFromPickle( | 29 ASSERT_TRUE(CompressDataHelper::ReadAndDecompressStringFromPickle( |
30 pickle, &it, &data_out)); | 30 pickle, &it, &data_out)); |
31 | 31 |
32 ASSERT_EQ(data.size(), data_out.size()); | 32 ASSERT_EQ(data.size(), data_out.size()); |
33 for (size_t i = 0; i < data_out.size(); ++i) | 33 for (size_t i = 0; i < data_out.size(); ++i) |
34 EXPECT_EQ(data_out[i], data[i]); | 34 EXPECT_EQ(data_out[i], data[i]); |
35 } | 35 } |
36 | 36 |
37 TEST_F(CompressDataHelperTest, CompressAndDecompressEmptyData) { | 37 TEST_F(CompressDataHelperTest, CompressAndDecompressEmptyData) { |
38 std::string empty; | 38 std::string empty; |
39 | 39 |
40 Pickle pickle; | 40 Pickle pickle; |
41 int bytes_written = 0; | 41 int bytes_written = 0; |
42 int max_bytes = 100; | 42 int max_bytes = 100; |
43 CompressDataHelper::CompressAndWriteStringToPickle( | 43 CompressDataHelper::CompressAndWriteStringToPickle( |
44 empty, max_bytes, &pickle, &bytes_written); | 44 empty, max_bytes, &pickle, &bytes_written); |
45 | 45 |
46 EXPECT_EQ(0, bytes_written); | 46 EXPECT_EQ(0, bytes_written); |
47 | 47 |
48 void* it = NULL; | 48 PickleIterator it(pickle); |
49 std::string data_out; | 49 std::string data_out; |
50 ASSERT_TRUE(CompressDataHelper::ReadAndDecompressStringFromPickle( | 50 ASSERT_TRUE(CompressDataHelper::ReadAndDecompressStringFromPickle( |
51 pickle, &it, &data_out)); | 51 pickle, &it, &data_out)); |
52 | 52 |
53 EXPECT_EQ(0U, data_out.size()); | 53 EXPECT_EQ(0U, data_out.size()); |
54 } | 54 } |
55 | 55 |
56 TEST_F(CompressDataHelperTest, TooMuchData) { | 56 TEST_F(CompressDataHelperTest, TooMuchData) { |
57 std::string data = "Some data that clearly cannot be compressed to 2 bytes"; | 57 std::string data = "Some data that clearly cannot be compressed to 2 bytes"; |
58 | 58 |
59 Pickle pickle; | 59 Pickle pickle; |
60 int bytes_written = 0; | 60 int bytes_written = 0; |
61 int max_bytes = 2; | 61 int max_bytes = 2; |
62 CompressDataHelper::CompressAndWriteStringToPickle( | 62 CompressDataHelper::CompressAndWriteStringToPickle( |
63 data, max_bytes, &pickle, &bytes_written); | 63 data, max_bytes, &pickle, &bytes_written); |
64 | 64 |
65 EXPECT_EQ(0, bytes_written); | 65 EXPECT_EQ(0, bytes_written); |
66 | 66 |
67 // When the data is read, we get an empty string back. | 67 // When the data is read, we get an empty string back. |
68 void* it = NULL; | 68 PickleIterator it(pickle); |
69 std::string data_out; | 69 std::string data_out; |
70 ASSERT_TRUE(CompressDataHelper::ReadAndDecompressStringFromPickle( | 70 ASSERT_TRUE(CompressDataHelper::ReadAndDecompressStringFromPickle( |
71 pickle, &it, &data_out)); | 71 pickle, &it, &data_out)); |
72 | 72 |
73 EXPECT_EQ(0U, data_out.size()); | 73 EXPECT_EQ(0U, data_out.size()); |
74 } | 74 } |
OLD | NEW |