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