OLD | NEW |
| (Empty) |
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 | |
3 // found in the LICENSE file. | |
4 | |
5 #include "base/data_pack.h" | |
6 | |
7 #include "base/file_path.h" | |
8 #include "base/file_util.h" | |
9 #include "base/path_service.h" | |
10 #include "base/scoped_temp_dir.h" | |
11 #include "base/string_piece.h" | |
12 #include "testing/gtest/include/gtest/gtest.h" | |
13 | |
14 TEST(DataPackTest, Load) { | |
15 FilePath data_path; | |
16 PathService::Get(base::DIR_SOURCE_ROOT, &data_path); | |
17 data_path = data_path.Append( | |
18 FILE_PATH_LITERAL("base/data/data_pack_unittest/sample.pak")); | |
19 | |
20 base::DataPack pack; | |
21 ASSERT_TRUE(pack.Load(data_path)); | |
22 | |
23 base::StringPiece data; | |
24 ASSERT_TRUE(pack.GetStringPiece(4, &data)); | |
25 EXPECT_EQ("this is id 4", data); | |
26 ASSERT_TRUE(pack.GetStringPiece(6, &data)); | |
27 EXPECT_EQ("this is id 6", data); | |
28 | |
29 // Try reading zero-length data blobs, just in case. | |
30 ASSERT_TRUE(pack.GetStringPiece(1, &data)); | |
31 EXPECT_EQ(0U, data.length()); | |
32 ASSERT_TRUE(pack.GetStringPiece(10, &data)); | |
33 EXPECT_EQ(0U, data.length()); | |
34 | |
35 // Try looking up an invalid key. | |
36 ASSERT_FALSE(pack.GetStringPiece(140, &data)); | |
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 |