summaryrefslogtreecommitdiffstats
path: root/chrome/browser/views/shell_dialogs_win_unittest.cc
blob: 0c2cad5aeb36a6baca4181da8b67522ee8207761 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
// Copyright (c) 2010 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/shell_dialogs.h"
#include "testing/gtest/include/gtest/gtest.h"

TEST(ShellDialogsWin, AppendExtensionIfNeeded) {
  struct AppendExtensionTestCase {
    wchar_t* filename;
    wchar_t* filter_selected;
    wchar_t* suggested_ext;
    wchar_t* expected_filename;
  } test_cases[] = {
    // Known extensions, with or without associated MIME types, should not get
    // an extension appended.
    { L"sample.html",    L"*.txt",     L"txt",     L"sample.html" },
    { L"sample.reg",     L"*.txt",     L"txt",     L"sample.reg" },

    // An unknown extension, or no extension, should get the default extension
    // appended.
    { L"sample.unknown", L"*.txt",     L"txt",     L"sample.unknown.txt" },
    { L"sample",         L"*.txt",     L"txt",     L"sample.txt" },
    // ...unless the unknown and default extensions match.
    { L"sample.unknown", L"*.unknown", L"unknown", L"sample.unknown" },

    // The extension alone should be treated like a filename with no extension.
    { L"txt",            L"*.txt",     L"txt",     L"txt.txt" },

    // Trailing dots should cause us to append an extension.
    { L"sample.txt.",    L"*.txt",     L"txt",     L"sample.txt.txt" },
    { L"...",            L"*.txt",     L"txt",     L"...txt" },

    // If the filter is changed to "All files", we allow any filename.
    { L"sample.unknown", L"*.*",       L"",        L"sample.unknown" },
  };

  for (size_t i = 0; i < arraysize(test_cases); ++i) {
    EXPECT_EQ(std::wstring(test_cases[i].expected_filename),
              AppendExtensionIfNeeded(test_cases[i].filename,
                                      test_cases[i].filter_selected,
                                      test_cases[i].suggested_ext));
  }
}