OLD | NEW |
1 // Copyright (c) 2008 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2008 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 "base/data_pack.h" | 5 #include "base/data_pack.h" |
6 | 6 |
7 #include "base/file_path.h" | 7 #include "base/file_path.h" |
| 8 #include "base/file_util.h" |
8 #include "base/path_service.h" | 9 #include "base/path_service.h" |
| 10 #include "base/scoped_temp_dir.h" |
9 #include "base/string_piece.h" | 11 #include "base/string_piece.h" |
10 #include "testing/gtest/include/gtest/gtest.h" | 12 #include "testing/gtest/include/gtest/gtest.h" |
11 | 13 |
12 class DataPackTest : public testing::Test { | 14 TEST(DataPackTest, Load) { |
13 public: | 15 FilePath data_path; |
14 DataPackTest() { | 16 PathService::Get(base::DIR_SOURCE_ROOT, &data_path); |
15 PathService::Get(base::DIR_SOURCE_ROOT, &data_path_); | 17 data_path = data_path.Append( |
16 data_path_ = data_path_.Append( | |
17 FILE_PATH_LITERAL("base/data/data_pack_unittest/sample.pak")); | 18 FILE_PATH_LITERAL("base/data/data_pack_unittest/sample.pak")); |
18 } | |
19 | 19 |
20 FilePath data_path_; | |
21 }; | |
22 | |
23 TEST_F(DataPackTest, Load) { | |
24 base::DataPack pack; | 20 base::DataPack pack; |
25 ASSERT_TRUE(pack.Load(data_path_)); | 21 ASSERT_TRUE(pack.Load(data_path)); |
26 | 22 |
27 base::StringPiece data; | 23 base::StringPiece data; |
28 ASSERT_TRUE(pack.GetStringPiece(4, &data)); | 24 ASSERT_TRUE(pack.GetStringPiece(4, &data)); |
29 EXPECT_EQ("this is id 4", data); | 25 EXPECT_EQ("this is id 4", data); |
30 ASSERT_TRUE(pack.GetStringPiece(6, &data)); | 26 ASSERT_TRUE(pack.GetStringPiece(6, &data)); |
31 EXPECT_EQ("this is id 6", data); | 27 EXPECT_EQ("this is id 6", data); |
32 | 28 |
33 // Try reading zero-length data blobs, just in case. | 29 // Try reading zero-length data blobs, just in case. |
34 ASSERT_TRUE(pack.GetStringPiece(1, &data)); | 30 ASSERT_TRUE(pack.GetStringPiece(1, &data)); |
35 EXPECT_EQ(0U, data.length()); | 31 EXPECT_EQ(0U, data.length()); |
36 ASSERT_TRUE(pack.GetStringPiece(10, &data)); | 32 ASSERT_TRUE(pack.GetStringPiece(10, &data)); |
37 EXPECT_EQ(0U, data.length()); | 33 EXPECT_EQ(0U, data.length()); |
38 | 34 |
39 // Try looking up an invalid key. | 35 // Try looking up an invalid key. |
40 ASSERT_FALSE(pack.GetStringPiece(140, &data)); | 36 ASSERT_FALSE(pack.GetStringPiece(140, &data)); |
41 } | 37 } |
| 38 |
| 39 TEST(DataPackTest, Write) { |
| 40 ScopedTempDir dir; |
| 41 ASSERT_TRUE(dir.CreateUniqueTempDir()); |
| 42 FilePath file = dir.path().Append(FILE_PATH_LITERAL("data.pak")); |
| 43 |
| 44 std::string one("one"); |
| 45 std::string two("two"); |
| 46 std::string three("three"); |
| 47 std::string four("four"); |
| 48 std::string fifteen("fifteen"); |
| 49 |
| 50 std::map<uint32, base::StringPiece> resources; |
| 51 resources.insert(std::make_pair(1, base::StringPiece(one))); |
| 52 resources.insert(std::make_pair(2, base::StringPiece(two))); |
| 53 resources.insert(std::make_pair(15, base::StringPiece(fifteen))); |
| 54 resources.insert(std::make_pair(3, base::StringPiece(three))); |
| 55 resources.insert(std::make_pair(4, base::StringPiece(four))); |
| 56 ASSERT_TRUE(base::DataPack::WritePack(file, resources)); |
| 57 |
| 58 // Now try to read the data back in. |
| 59 base::DataPack pack; |
| 60 ASSERT_TRUE(pack.Load(file)); |
| 61 |
| 62 base::StringPiece data; |
| 63 ASSERT_TRUE(pack.GetStringPiece(1, &data)); |
| 64 EXPECT_EQ(one, data); |
| 65 ASSERT_TRUE(pack.GetStringPiece(2, &data)); |
| 66 EXPECT_EQ(two, data); |
| 67 ASSERT_TRUE(pack.GetStringPiece(3, &data)); |
| 68 EXPECT_EQ(three, data); |
| 69 ASSERT_TRUE(pack.GetStringPiece(4, &data)); |
| 70 EXPECT_EQ(four, data); |
| 71 ASSERT_TRUE(pack.GetStringPiece(15, &data)); |
| 72 EXPECT_EQ(fifteen, data); |
| 73 } |
OLD | NEW |