| 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 |