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 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