OLD | NEW |
| (Empty) |
1 // Copyright 2015 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 "ui/base/resource/resource_bundle.h" | |
6 | |
7 #import <AppKit/AppKit.h> | |
8 #include <stddef.h> | |
9 #include <stdint.h> | |
10 | |
11 #include "base/base_paths.h" | |
12 #include "base/big_endian.h" | |
13 #include "base/files/file_path.h" | |
14 #include "base/files/file_util.h" | |
15 #include "base/files/scoped_temp_dir.h" | |
16 #include "base/logging.h" | |
17 #include "base/macros.h" | |
18 #include "base/memory/ref_counted_memory.h" | |
19 #include "testing/gmock/include/gmock/gmock.h" | |
20 #include "testing/gtest/include/gtest/gtest.h" | |
21 #include "third_party/skia/include/core/SkBitmap.h" | |
22 #include "ui/base/resource/data_pack.h" | |
23 #include "ui/gfx/codec/png_codec.h" | |
24 #include "ui/gfx/image/image_skia.h" | |
25 #include "ui/resources/grit/ui_resources.h" | |
26 #include "ui/strings/grit/app_locale_settings.h" | |
27 | |
28 namespace ui { | |
29 | |
30 extern const char kEmptyPakContents[]; | |
31 extern const size_t kEmptyPakSize; | |
32 | |
33 namespace { | |
34 | |
35 const unsigned char kPngMagic[8] = { 0x89, 'P', 'N', 'G', 13, 10, 26, 10 }; | |
36 const size_t kPngChunkMetadataSize = 12; | |
37 const unsigned char kPngIHDRChunkType[4] = { 'I', 'H', 'D', 'R' }; | |
38 | |
39 // Returns |bitmap_data| with |custom_chunk| inserted after the IHDR chunk. | |
40 void AddCustomChunk(const base::StringPiece& custom_chunk, | |
41 std::vector<unsigned char>* bitmap_data) { | |
42 EXPECT_LT(arraysize(kPngMagic) + kPngChunkMetadataSize, bitmap_data->size()); | |
43 EXPECT_TRUE(std::equal( | |
44 bitmap_data->begin(), | |
45 bitmap_data->begin() + arraysize(kPngMagic), | |
46 kPngMagic)); | |
47 std::vector<unsigned char>::iterator ihdr_start = | |
48 bitmap_data->begin() + arraysize(kPngMagic); | |
49 char ihdr_length_data[sizeof(uint32_t)]; | |
50 for (size_t i = 0; i < sizeof(uint32_t); ++i) | |
51 ihdr_length_data[i] = *(ihdr_start + i); | |
52 uint32_t ihdr_chunk_length = 0; | |
53 base::ReadBigEndian(reinterpret_cast<char*>(ihdr_length_data), | |
54 &ihdr_chunk_length); | |
55 EXPECT_TRUE( | |
56 std::equal(ihdr_start + sizeof(uint32_t), | |
57 ihdr_start + sizeof(uint32_t) + sizeof(kPngIHDRChunkType), | |
58 kPngIHDRChunkType)); | |
59 | |
60 bitmap_data->insert(ihdr_start + kPngChunkMetadataSize + ihdr_chunk_length, | |
61 custom_chunk.begin(), custom_chunk.end()); | |
62 } | |
63 | |
64 // Creates datapack at |path| with a single bitmap at resource ID 3 | |
65 // which is |edge_size|x|edge_size| pixels. | |
66 // If |custom_chunk| is non empty, adds it after the IHDR chunk | |
67 // in the encoded bitmap data. | |
68 void CreateDataPackWithSingleBitmap(const base::FilePath& path, | |
69 int edge_size, | |
70 const base::StringPiece& custom_chunk) { | |
71 SkBitmap bitmap; | |
72 bitmap.allocN32Pixels(edge_size, edge_size); | |
73 bitmap.eraseColor(SK_ColorWHITE); | |
74 std::vector<unsigned char> bitmap_data; | |
75 EXPECT_TRUE(gfx::PNGCodec::EncodeBGRASkBitmap(bitmap, false, &bitmap_data)); | |
76 | |
77 if (custom_chunk.size() > 0) | |
78 AddCustomChunk(custom_chunk, &bitmap_data); | |
79 | |
80 std::map<uint16_t, base::StringPiece> resources; | |
81 resources[3u] = base::StringPiece( | |
82 reinterpret_cast<const char*>(&bitmap_data[0]), bitmap_data.size()); | |
83 DataPack::WritePack(path, resources, ui::DataPack::BINARY); | |
84 } | |
85 | |
86 } // namespace | |
87 | |
88 class ResourceBundleMacImageTest : public testing::Test { | |
89 public: | |
90 ResourceBundleMacImageTest() : resource_bundle_(NULL) {} | |
91 | |
92 ~ResourceBundleMacImageTest() override {} | |
93 | |
94 void SetUp() override { | |
95 // Create a temporary directory to write test resource bundles to. | |
96 ASSERT_TRUE(dir_.CreateUniqueTempDir()); | |
97 } | |
98 | |
99 // Overridden from testing::Test: | |
100 void TearDown() override { delete resource_bundle_; } | |
101 | |
102 // Returns new ResoureBundle with the specified |delegate|. The | |
103 // ResourceBundleTest class manages the lifetime of the returned | |
104 // ResourceBundle. | |
105 ResourceBundle* CreateResourceBundle(ResourceBundle::Delegate* delegate) { | |
106 DCHECK(!resource_bundle_); | |
107 | |
108 resource_bundle_ = new ResourceBundle(delegate); | |
109 return resource_bundle_; | |
110 } | |
111 | |
112 // Returns resource bundle which uses an empty data pak for locale data. | |
113 ui::ResourceBundle* CreateResourceBundleWithEmptyLocalePak() { | |
114 // Write an empty data pak for locale data. | |
115 const base::FilePath& locale_path = dir_path().Append( | |
116 FILE_PATH_LITERAL("locale.pak")); | |
117 EXPECT_EQ(base::WriteFile(locale_path, kEmptyPakContents, kEmptyPakSize), | |
118 static_cast<int>(kEmptyPakSize)); | |
119 | |
120 ui::ResourceBundle* resource_bundle = CreateResourceBundle(NULL); | |
121 | |
122 // Load the empty locale data pak. | |
123 resource_bundle->LoadTestResources(base::FilePath(), locale_path); | |
124 return resource_bundle; | |
125 } | |
126 | |
127 // Returns the path of temporary directory to write test data packs into. | |
128 const base::FilePath& dir_path() { return dir_.path(); } | |
129 | |
130 // Returns the number of DataPacks managed by |resource_bundle| which are | |
131 // flagged as containing only material design resources. | |
132 size_t NumberOfMaterialDesignDataPacksInResourceBundle( | |
133 ResourceBundle* resource_bundle) { | |
134 DCHECK(resource_bundle); | |
135 size_t num_material_packs = 0; | |
136 for (size_t i = 0; i < resource_bundle->data_packs_.size(); i++) { | |
137 if (resource_bundle->data_packs_[i]->HasOnlyMaterialDesignAssets()) | |
138 num_material_packs++; | |
139 } | |
140 | |
141 return num_material_packs; | |
142 } | |
143 | |
144 protected: | |
145 ResourceBundle* resource_bundle_; | |
146 | |
147 private: | |
148 std::unique_ptr<DataPack> locale_pack_; | |
149 base::ScopedTempDir dir_; | |
150 | |
151 DISALLOW_COPY_AND_ASSIGN(ResourceBundleMacImageTest); | |
152 }; | |
153 | |
154 // Verifies that ResourceBundle searches the Material Design data pack before | |
155 // the default data pack, and that the returned image contains only the | |
156 // representation from the Material Design pack. | |
157 TEST_F(ResourceBundleMacImageTest, CheckImageFromMaterialDesign) { | |
158 // Create two .pak files, each containing a single image with the | |
159 // same asset ID but different sizes (note that the images must be | |
160 // different sizes in this test in order to correctly determine | |
161 // from which data pack the asset was pulled). Note also that the value | |
162 // of |material_size| was chosen to be divisible by 3, since iOS may | |
163 // use this scale factor. | |
164 const int default_size = 10; | |
165 const int material_size = 48; | |
166 ASSERT_NE(default_size, material_size); | |
167 base::FilePath default_path = dir_path().AppendASCII("default.pak"); | |
168 base::FilePath material_path = dir_path().AppendASCII("material.pak"); | |
169 CreateDataPackWithSingleBitmap(default_path, | |
170 default_size, | |
171 base::StringPiece()); | |
172 CreateDataPackWithSingleBitmap(material_path, | |
173 material_size, | |
174 base::StringPiece()); | |
175 | |
176 ScaleFactor scale_factor = SCALE_FACTOR_100P; | |
177 ResourceBundle* resource_bundle = CreateResourceBundleWithEmptyLocalePak(); | |
178 | |
179 // Load the 'material' data pack into ResourceBundle first. | |
180 resource_bundle->AddMaterialDesignDataPackFromPath(material_path, | |
181 scale_factor); | |
182 resource_bundle->AddDataPackFromPath(default_path, scale_factor); | |
183 | |
184 // Confirm that there are two data packs available. | |
185 const unsigned int data_packs_size = resource_bundle->data_packs_.size(); | |
186 const unsigned int expected_size = 2; | |
187 EXPECT_EQ(expected_size, data_packs_size); | |
188 | |
189 const size_t md_data_pack_count = | |
190 NumberOfMaterialDesignDataPacksInResourceBundle(resource_bundle); | |
191 const size_t expected_pack_count = 1; | |
192 EXPECT_EQ(expected_pack_count, md_data_pack_count); | |
193 | |
194 // Normally with two packs containing the same image, GetNativeImageNamed() | |
195 // returns an NSImage that contains both representations. With the Material | |
196 // Design pack, any images it contains should override the ones in the default | |
197 // pack, so if both packs contain the same image, the returned NSImage should | |
198 // contain just a single representation. | |
199 NSImage* icon = resource_bundle->GetNativeImageNamed(3).ToNSImage(); | |
200 const unsigned long representations_count = [[icon representations] count]; | |
201 const unsigned long expected_count = 1; | |
202 EXPECT_EQ(expected_count, representations_count); | |
203 } | |
204 | |
205 } // namespace ui | |
OLD | NEW |