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
3 // found in the LICENSE file.
4
5 #include "base/memory/ref_counted_memory.h"
6 #include "base/message_loop/message_loop.h"
7 #include "chrome/browser/icon_manager.h"
8 #include "chrome/browser/profiles/profile.h"
9 #include "chrome/browser/ui/webui/fileicon_source.h"
10 #include "chrome/test/base/testing_profile.h"
11 #include "content/public/test/test_browser_thread.h"
12 #include "testing/gmock/include/gmock/gmock.h"
13 #include "testing/gtest/include/gtest/gtest.h"
14
15 using content::BrowserThread;
16
17 namespace {
18
19 class TestFileIconSource : public FileIconSource {
20 public:
TestFileIconSource()21 explicit TestFileIconSource() {}
22
23 MOCK_METHOD4(FetchFileIcon,
24 void(const base::FilePath& path,
25 float scale_factor,
26 IconLoader::IconSize icon_size,
27 const content::URLDataSource::GotDataCallback& callback));
28
~TestFileIconSource()29 virtual ~TestFileIconSource() {}
30 };
31
32 class FileIconSourceTest : public testing::Test {
33 public:
FileIconSourceTest()34 FileIconSourceTest()
35 : ui_thread_(BrowserThread::UI, base::MessageLoop::current()),
36 file_thread_(BrowserThread::FILE, base::MessageLoop::current()) {}
37
CreateFileIconSource()38 static TestFileIconSource* CreateFileIconSource() {
39 return new TestFileIconSource();
40 }
41
42 private:
43 base::MessageLoopForUI loop_;
44 content::TestBrowserThread ui_thread_;
45 content::TestBrowserThread file_thread_;
46 };
47
48 const struct FetchFileIconExpectation {
49 const char* request_path;
50 const base::FilePath::CharType* unescaped_path;
51 float scale_factor;
52 IconLoader::IconSize size;
53 } kBasicExpectations[] = {
54 { "foo?bar", FILE_PATH_LITERAL("foo"), 1.0f, IconLoader::NORMAL },
55 { "foo?bar&scale=2x", FILE_PATH_LITERAL("foo"), 2.0f, IconLoader::NORMAL },
56 { "foo?iconsize=small", FILE_PATH_LITERAL("foo"), 1.0f, IconLoader::SMALL },
57 { "foo?iconsize=normal", FILE_PATH_LITERAL("foo"), 1.0f, IconLoader::NORMAL },
58 { "foo?iconsize=large", FILE_PATH_LITERAL("foo"), 1.0f, IconLoader::LARGE },
59 { "foo?iconsize=asdf", FILE_PATH_LITERAL("foo"), 1.0f, IconLoader::NORMAL },
60 { "foo?blah=b&iconsize=small", FILE_PATH_LITERAL("foo"), 1.0f,
61 IconLoader::SMALL },
62 { "foo?blah&iconsize=small", FILE_PATH_LITERAL("foo"), 1.0f,
63 IconLoader::SMALL },
64 { "a%3Fb%26c%3Dd.txt?iconsize=small", FILE_PATH_LITERAL("a?b&c=d.txt"), 1.0f,
65 IconLoader::SMALL },
66 { "a%3Ficonsize%3Dsmall?iconsize=large",
67 FILE_PATH_LITERAL("a?iconsize=small"), 1.0f, IconLoader::LARGE },
68 { "o%40%23%24%25%26*()%20%2B%3D%3F%2C%3A%3B%22.jpg",
69 FILE_PATH_LITERAL("o@#$%&*() +=?,:;\".jpg"), 1.0f, IconLoader::NORMAL },
70 #if defined(OS_WIN)
71 { "c:/foo/bar/baz", FILE_PATH_LITERAL("c:\\foo\\bar\\baz"), 1.0f,
72 IconLoader::NORMAL },
73 { "/foo?bar=asdf&asdf", FILE_PATH_LITERAL("\\foo"), 1.0f,
74 IconLoader::NORMAL },
75 { "c%3A%2Fusers%2Ffoo%20user%2Fbar.txt",
76 FILE_PATH_LITERAL("c:\\users\\foo user\\bar.txt"), 1.0f,
77 IconLoader::NORMAL },
78 { "c%3A%2Fusers%2F%C2%A9%202000.pdf",
79 FILE_PATH_LITERAL("c:\\users\\\xa9 2000.pdf"), 1.0f, IconLoader::NORMAL },
80 { "%E0%B6%9A%E0%B6%BB%E0%B7%9D%E0%B6%B8%E0%B7%8A",
81 FILE_PATH_LITERAL("\x0d9a\x0dbb\x0ddd\x0db8\x0dca"), 1.0f,
82 IconLoader::NORMAL },
83 { "%2Ffoo%2Fbar", FILE_PATH_LITERAL("\\foo\\bar"), 1.0f, IconLoader::NORMAL },
84 { "%2Fbaz%20(1).txt?iconsize=small", FILE_PATH_LITERAL("\\baz (1).txt"),
85 1.0f, IconLoader::SMALL },
86 #else
87 { "/foo/bar/baz", FILE_PATH_LITERAL("/foo/bar/baz"), 1.0f,
88 IconLoader::NORMAL },
89 { "/foo?bar", FILE_PATH_LITERAL("/foo"), 1.0f, IconLoader::NORMAL },
90 { "%2Ffoo%2f%E0%B6%9A%E0%B6%BB%E0%B7%9D%E0%B6%B8%E0%B7%8A",
91 FILE_PATH_LITERAL("/foo/\xe0\xb6\x9a\xe0\xb6\xbb\xe0\xb7\x9d")
92 FILE_PATH_LITERAL("\xe0\xb6\xb8\xe0\xb7\x8a"), 1.0f, IconLoader::NORMAL },
93 { "%2Ffoo%2Fbar", FILE_PATH_LITERAL("/foo/bar"), 1.0f, IconLoader::NORMAL },
94 { "%2Fbaz%20(1).txt?iconsize=small", FILE_PATH_LITERAL("/baz (1).txt"), 1.0f,
95 IconLoader::SMALL },
96 #endif
97 };
98
99 // Test that the callback is NULL.
100 MATCHER(CallbackIsNull, "") {
101 return arg.is_null();
102 }
103
104 } // namespace
105
TEST_F(FileIconSourceTest,FileIconSource_Parse)106 TEST_F(FileIconSourceTest, FileIconSource_Parse) {
107 std::vector<ui::ScaleFactor> supported_scale_factors;
108 supported_scale_factors.push_back(ui::SCALE_FACTOR_100P);
109 supported_scale_factors.push_back(ui::SCALE_FACTOR_200P);
110 ui::test::ScopedSetSupportedScaleFactors scoped_supported(
111 supported_scale_factors);
112
113 for (unsigned i = 0; i < arraysize(kBasicExpectations); i++) {
114 scoped_ptr<TestFileIconSource> source(CreateFileIconSource());
115 content::URLDataSource::GotDataCallback callback;
116 EXPECT_CALL(*source.get(),
117 FetchFileIcon(
118 base::FilePath(kBasicExpectations[i].unescaped_path),
119 kBasicExpectations[i].scale_factor,
120 kBasicExpectations[i].size, CallbackIsNull()));
121 source->StartDataRequest(kBasicExpectations[i].request_path, -1, -1,
122 callback);
123 }
124 }
125