1 // Copyright (c) 2010 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 "chrome/browser/ui/shell_dialogs.h"
6 #include "testing/gtest/include/gtest/gtest.h"
7
TEST(ShellDialogsWin,AppendExtensionIfNeeded)8 TEST(ShellDialogsWin, AppendExtensionIfNeeded) {
9 struct AppendExtensionTestCase {
10 wchar_t* filename;
11 wchar_t* filter_selected;
12 wchar_t* suggested_ext;
13 wchar_t* expected_filename;
14 } test_cases[] = {
15 // Known extensions, with or without associated MIME types, should not get
16 // an extension appended.
17 { L"sample.html", L"*.txt", L"txt", L"sample.html" },
18 { L"sample.reg", L"*.txt", L"txt", L"sample.reg" },
19
20 // An unknown extension, or no extension, should get the default extension
21 // appended.
22 { L"sample.unknown", L"*.txt", L"txt", L"sample.unknown.txt" },
23 { L"sample", L"*.txt", L"txt", L"sample.txt" },
24 // ...unless the unknown and default extensions match.
25 { L"sample.unknown", L"*.unknown", L"unknown", L"sample.unknown" },
26
27 // The extension alone should be treated like a filename with no extension.
28 { L"txt", L"*.txt", L"txt", L"txt.txt" },
29
30 // Trailing dots should cause us to append an extension.
31 { L"sample.txt.", L"*.txt", L"txt", L"sample.txt.txt" },
32 { L"...", L"*.txt", L"txt", L"...txt" },
33
34 // If the filter is changed to "All files", we allow any filename.
35 { L"sample.unknown", L"*.*", L"", L"sample.unknown" },
36 };
37
38 for (size_t i = 0; i < arraysize(test_cases); ++i) {
39 EXPECT_EQ(std::wstring(test_cases[i].expected_filename),
40 AppendExtensionIfNeeded(test_cases[i].filename,
41 test_cases[i].filter_selected,
42 test_cases[i].suggested_ext));
43 }
44 }
45