shell_dialogs_win_unittest.cc revision 4a5e2dc747d50c653511c68ccb2cfbfb740bd5a7 
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/shell_dialogs.h"
6#include "testing/gtest/include/gtest/gtest.h"
7
8TEST(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