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
45
46
47
48
49
50
51
52
|
// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifdef UNSAFE_BUFFERS_BUILD
// TODO(crbug.com/351564777): Remove this and convert code to safer constructs.
#pragma allow_unsafe_buffers
#endif
#include "ui/shell_dialogs/select_file_utils_win.h"
#include <stddef.h>
#include "base/strings/stringprintf.h"
#include "testing/gtest/include/gtest/gtest.h"
TEST(SelectFileUtilsWin, RemoveEnvVarFromFileName) {
struct RemoveEnvFromFileNameTestCase {
const wchar_t* filename;
const wchar_t* sanitized_filename;
} test_cases[] = {
{L"", L""},
{L"a.txt", L"a.txt"},
// Only 1 "%" in file name.
{L"%", L"%"},
{L"%.txt", L"%.txt"},
{L"ab%c.txt", L"ab%c.txt"},
{L"abc.t%", L"abc.t%"},
// 2 "%" in file name.
{L"%%", L""},
{L"%c%", L""},
{L"%c%d", L"d"},
{L"d%c%.txt", L"d.txt"},
{L"ab%c.t%", L"ab"},
{L"abc.%t%", L"abc."},
{L"*.%txt%", L"*."},
// More than 2 "%" in file name.
{L"%ab%c%.txt", L"c%.txt"},
{L"%abc%.%txt%", L"."},
{L"%ab%c%.%txt%", L"ctxt%"},
};
for (size_t i = 0; i < std::size(test_cases); ++i) {
SCOPED_TRACE(base::StringPrintf("i=%zu", i));
std::wstring sanitized = ui::RemoveEnvVarFromFileName<wchar_t>(
std::wstring(test_cases[i].filename), std::wstring(L"%"));
EXPECT_EQ(std::wstring(test_cases[i].sanitized_filename), sanitized);
}
}
|
