mirror of
https://github.com/kidfromjupiter/nearby.git
synced 2026-09-15 07:06:11 -04:00
965 lines
31 KiB
C++
965 lines
31 KiB
C++
// Copyright 2022 Google LLC
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// https://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#include "internal/platform/implementation/windows/file_path.h"
|
|
|
|
#include <windows.h>
|
|
#include <knownfolders.h>
|
|
#include <shlobj.h>
|
|
|
|
#include <algorithm>
|
|
#include <fstream>
|
|
#include <sstream>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "gtest/gtest.h"
|
|
#include "connections/implementation/flags/nearby_connections_feature_flags.h"
|
|
#include "internal/flags/nearby_flags.h"
|
|
|
|
namespace nearby {
|
|
namespace windows {
|
|
|
|
namespace {
|
|
const wchar_t* kIllegalPathNames[] = {
|
|
L"CON", L"PRN", L"AUX", L"NUL", L"COM1", L"COM2", L"COM3", L"COM4",
|
|
L"COM5", L"COM6", L"COM7", L"COM8", L"COM9", L"LPT1", L"LPT2", L"LPT3",
|
|
L"LPT4", L"LPT5", L"LPT6", L"LPT7", L"LPT8", L"LPT9"};
|
|
|
|
const wchar_t* kFileName(L"increment_file_test.txt");
|
|
const wchar_t* kFirstIterationFileName(L"/increment_file_test (1).txt");
|
|
const wchar_t* kSecondIterationFileName(L"/increment_file_test (2).txt");
|
|
const wchar_t* kThirdIterationFileName(L"/increment_file_test (3).txt");
|
|
const wchar_t* kFileNameWithNullReplaced(L"/increment_file_test.txt_.txt");
|
|
const wchar_t* kNoDotsFileName(L"incrementfiletesttxt");
|
|
const wchar_t* kOneIterationNoDotsFileName(L"/incrementfiletesttxt (1)");
|
|
const wchar_t* kMultipleDotsFileName(L"increment.file.test.txt");
|
|
const wchar_t* kOneIterationMultipleDotsFileName(
|
|
L"/increment.file.test (1).txt");
|
|
const wchar_t* kImmediateEscape(L"../");
|
|
const wchar_t* kLongEscapeBackSlash(L"..\\test\\..\\..\\test");
|
|
const wchar_t* kTwoLevelFolder(L"/test/test");
|
|
const wchar_t* kLongEscapeSlash(L"../test/../../test");
|
|
const wchar_t* kLongEscapeMixedSlash(L"../test\\..\\../test");
|
|
const wchar_t* kLongEscapeEndingEscape(L"../test/../../test/..");
|
|
const wchar_t* kLongEscapeEndingEscapeWithSlash(
|
|
L"../test/../../test/../../../");
|
|
const wchar_t* kFileNameWithThreeDots(L"...");
|
|
const wchar_t* kFileNameWithFrontTwoDots(L"..file.name.txt");
|
|
} // namespace
|
|
|
|
// Can't run on google 3, I presume the SHGetKnownFolderPath
|
|
// fails.
|
|
class FilePathTests : public testing::Test {
|
|
protected:
|
|
// You can define per-test set-up logic as usual.
|
|
FilePathTests() {
|
|
PWSTR basePath;
|
|
|
|
SHGetKnownFolderPath(
|
|
FOLDERID_Downloads, // rfid: A reference to the KNOWNFOLDERID that
|
|
// identifies the folder.
|
|
0, // dwFlags: Flags that specify special retrieval
|
|
// options.
|
|
nullptr, // hToken: An access token that represents a
|
|
// particular user.
|
|
&basePath); // ppszPath: When this method returns, contains
|
|
// the address of a pointer to a
|
|
// null-terminated Unicode string that
|
|
// specifies the path of the known
|
|
// folder. The calling process is
|
|
// responsible for freeing this resource
|
|
// once it is no longer needed by
|
|
// calling CoTaskMemFree, whether
|
|
// SHGetKnownFolderPath succeeds or not.
|
|
|
|
// size_t bufferSize;
|
|
// wcstombs_s(&bufferSize, nullptr, 0, basePath, 0);
|
|
// default_download_path_.resize(bufferSize - 1, '\0');
|
|
// wcstombs_s(&bufferSize, default_download_path_.data(), bufferSize,
|
|
// basePath,
|
|
// _TRUNCATE);
|
|
default_download_path_ = basePath;
|
|
|
|
std::replace(default_download_path_.begin(), default_download_path_.end(),
|
|
L'\\', L'/');
|
|
}
|
|
|
|
std::wstring default_download_path_;
|
|
};
|
|
|
|
TEST_F(FilePathTests, GetDownloadPathWithEmptyStringArguments\
|
|
ShouldReturnBaseDownloadPath) {
|
|
std::wstring parent_folder(L"");
|
|
std::wstring file_name(L"");
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(actual, default_download_path_);
|
|
} // NOLINT false lint error here
|
|
|
|
TEST_F(FilePathTests, GetDownloadPathWithSlashParent\
|
|
FolderArgumentsShouldReturnBaseDownloadPath) {
|
|
std::wstring parent_folder(L"/");
|
|
std::wstring file_name(L"");
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(actual, default_download_path_);
|
|
} // NOLINT false lint error here
|
|
|
|
TEST_F(FilePathTests, GetDownloadPathWithBackslashParent\
|
|
FolderArgumentsShouldReturnBaseDownloadPath) {
|
|
std::wstring parent_folder(L"\\");
|
|
std::wstring file_name(L"");
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(actual, default_download_path_);
|
|
} // NOLINT false lint error here
|
|
|
|
TEST_F(FilePathTests, GetDownloadPathWithAttemptToEscape\
|
|
UsersDownloadFolderShouldReturnDownloadPathNotEscapingUsersDownloadFolder) {
|
|
std::wstring parent_folder(kImmediateEscape);
|
|
std::wstring file_name(L"");
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(actual, default_download_path_);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPathWithMultiple\
|
|
AttemptsToEscapeUsersDownloadFolderWithBackslashShouldReturnDownloadPath\
|
|
NotEscapingUsersDownloadFolder) {
|
|
std::wstring parent_folder(kLongEscapeBackSlash);
|
|
std::wstring file_name(L"");
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(actual, default_download_path_ + kTwoLevelFolder);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPathWithMultiple\
|
|
AttemptsToEscapeUsersDownloadFolderShouldReturnDownloadPathNotEscapingUsers\
|
|
DownloadFolder) {
|
|
std::wstring parent_folder(kLongEscapeSlash);
|
|
std::wstring file_name(L"");
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(actual, default_download_path_ + kTwoLevelFolder);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPathWithMultiple\
|
|
AttemptsToEscapeUsersDownloadFolderWithMixedSlashShouldReturnDownloadPath\
|
|
NotEscapingUsersDownloadFolder) {
|
|
std::wstring parent_folder(kLongEscapeMixedSlash);
|
|
std::wstring file_name(L"");
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(actual, default_download_path_ + kTwoLevelFolder);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPathWithMultiple\
|
|
AttemptsToEscapeUsersDownloadFolderWithEndingEscapeShouldReturnDownload\
|
|
PathNotEscapingUsersDownloadFolder) {
|
|
std::wstring parent_folder(kLongEscapeEndingEscape);
|
|
std::wstring file_name(L"");
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(actual, default_download_path_ + kTwoLevelFolder);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPathWithMultiple\
|
|
AttemptsToEscapeUsersDownloadFolderWithEndingSlashShouldReturnDownloadPathNot\
|
|
EscapingUsersDownloadFolder) {
|
|
std::wstring parent_folder(kLongEscapeEndingEscapeWithSlash);
|
|
std::wstring file_name(L"");
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(actual, default_download_path_ + kTwoLevelFolder);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPathWithSlashFileName\
|
|
ArgumentsShouldReturnBaseDownloadPath) {
|
|
std::wstring parent_folder(L"");
|
|
std::wstring file_name(L"/");
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(actual, default_download_path_);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPathWithBackslashFile\
|
|
NameArgumentsShouldReturnBaseDownloadPath) {
|
|
std::wstring parent_folder(L"");
|
|
std::wstring file_name(L"\\");
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
auto result_size = actual.size();
|
|
auto default_size = default_download_path_.size();
|
|
|
|
EXPECT_EQ(actual, default_download_path_);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPathWithParentFolder\
|
|
ShouldReturnParentFolderAppendedToBaseDownloadPath) {
|
|
std::wstring parent_folder(L"test_parent_folder");
|
|
std::wstring file_name(L"");
|
|
|
|
std::wstringstream path(L"");
|
|
path << default_download_path_ << L"/" << "test_parent_folder";
|
|
|
|
std::wstring expected = path.str();
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPathWithParentFolder\
|
|
StartingWithSlashArgumentsShouldReturnParentFolderAppendedToBaseDownloadPath) {
|
|
std::wstring parent_folder(L"/test_parent_folder");
|
|
std::wstring file_name(L"");
|
|
|
|
std::wstringstream path(L"");
|
|
path << default_download_path_ << L"/" << "test_parent_folder";
|
|
|
|
std::wstring expected = path.str();
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPathWithParentFolder\
|
|
StartingWithBackslashArgumentsShouldReturnParentFolderAppendedToBase\
|
|
DownloadPath) {
|
|
std::wstring parent_folder(L"\\test_parent_folder");
|
|
std::wstring file_name(L"");
|
|
|
|
std::wstringstream path(L"");
|
|
path << default_download_path_ << L"/" << "test_parent_folder";
|
|
|
|
std::wstring expected = path.str();
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPathWithParentFolder\
|
|
EndingWithSlashArgumentsShouldReturnParentFolderAppendedToBaseDownloadPath) {
|
|
std::wstring parent_folder(L"test_parent_folder/");
|
|
std::wstring file_name(L"");
|
|
|
|
std::wstringstream path(L"");
|
|
path << default_download_path_ << L"/" << "test_parent_folder";
|
|
|
|
std::wstring expected = path.str();
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPathWithParentFolder\
|
|
EndingWithBackslashArguments\
|
|
ShouldReturnParentFolderAppendedToBaseDownloadPath) {
|
|
std::wstring parent_folder(L"test_parent_folder\\");
|
|
std::wstring file_name(L"");
|
|
|
|
std::wstringstream path(L"");
|
|
path << default_download_path_ << L"/" << "test_parent_folder";
|
|
|
|
std::wstring expected = path.str();
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPathWithFileName\
|
|
BeginningWithSlashArgumentsShouldReturnFileNameAppendedToBaseDownloadPath) {
|
|
std::wstring parent_folder(L"");
|
|
std::wstring file_name(L"/test_file_name.name");
|
|
|
|
std::wstringstream path(L"");
|
|
path << default_download_path_ << L"/" << "test_file_name.name";
|
|
|
|
std::wstring expected = path.str();
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPathWithFileName\
|
|
BeginningWithBackslashArgumentsShouldReturnFileNameAppendedToBaseDownloadPath) {
|
|
std::wstring parent_folder(L"");
|
|
std::wstring file_name(L"\\test_file_name.name");
|
|
|
|
std::wstringstream path(L"");
|
|
path << default_download_path_ << L"/" << "test_file_name.name";
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(actual, path.str().c_str());
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPathWithFileNameEnding\
|
|
WithSlashArgumentsShouldReturnFileNameAppendedToBaseDownloadPath) {
|
|
std::wstring parent_folder(L"");
|
|
std::wstring file_name(L"test_file_name.name/");
|
|
|
|
std::wstringstream path(L"");
|
|
path << default_download_path_ << L"/" << "test_file_name.name";
|
|
|
|
std::wstring expected = path.str();
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPathWithFileNameEnding\
|
|
WithBackslashArgumentsShouldReturnFileNameAppendedToBaseDownloadPath) {
|
|
std::wstring parent_folder(L"");
|
|
std::wstring file_name(L"test_file_name.name\\");
|
|
|
|
std::wstringstream path(L"");
|
|
path << default_download_path_ << L"/" << "test_file_name.name";
|
|
|
|
std::wstring expected = path.str();
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPathWithParentFolderAnd\
|
|
FileNameArgumentsShould\
|
|
ReturnParentFolderAndFileNameAppendedToBaseDownloadPath) {
|
|
std::wstring parent_folder(L"test_parent_folder");
|
|
std::wstring file_name(L"test_file_name.name");
|
|
|
|
std::wstringstream path(L"");
|
|
path << default_download_path_ << L"/" << "test_parent_folder"
|
|
<< "/"
|
|
<< "test_file_name.name";
|
|
|
|
std::wstring expected = path.str();
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPathWithParentFolder\
|
|
EndingWithBackslashAndFileNameArgumentsShouldReturnParentFolderAndFileName\
|
|
AppendedToBaseDownloadPath) {
|
|
std::wstring parent_folder(L"test_parent_folder\\");
|
|
std::wstring file_name(L"test_file_name.name");
|
|
|
|
std::wstringstream path(L"");
|
|
path << default_download_path_ << L"/" << "test_parent_folder"
|
|
<< "/"
|
|
<< "test_file_name.name";
|
|
|
|
std::wstring expected = path.str();
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPathWithFileName\
|
|
StartingWithBackslashAndParentFolderArgumentsShouldReturnParentFolderAnd\
|
|
FileNameAppendedToBaseDownloadPath) {
|
|
std::wstring parent_folder(L"test_parent_folder");
|
|
std::wstring file_name(L"\\test_file_name.name");
|
|
|
|
std::wstringstream path(L"");
|
|
path << default_download_path_ << L"/" << "test_parent_folder"
|
|
<< "/"
|
|
<< "test_file_name.name";
|
|
|
|
std::wstring expected = path.str();
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPath_IllegalPathComponent\
|
|
ReturnsComponentWithUnderbarPrepended) {
|
|
for (auto illegal_file_name : kIllegalPathNames) {
|
|
std::wstring parent_folder(L"");
|
|
|
|
std::wstring expected(default_download_path_);
|
|
expected.append(L"/_" + std::wstring(illegal_file_name) +
|
|
std::wstring(L"/") + std::wstring(kFileName));
|
|
|
|
auto actual(FilePath::GetDownloadPath(illegal_file_name, kFileName));
|
|
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPath_IllegalPathComponentInLowerCase\
|
|
ReturnsComponentWithUnderbarPrepended) {
|
|
for (auto illegal_file_name : kIllegalPathNames) {
|
|
std::wstring parent_folder(L"");
|
|
|
|
std::wstring expected(default_download_path_);
|
|
|
|
expected.append(L"/_" + std::wstring(_wcslwr(
|
|
(std::wstring(illegal_file_name)).data())));
|
|
expected.append(L"/" + std::wstring(kFileName));
|
|
|
|
auto actual(FilePath::GetDownloadPath(
|
|
std::wstring(_wcslwr((std::wstring(illegal_file_name)).data())),
|
|
kFileName));
|
|
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPath_IllegalFileNameInLowerCase\
|
|
ReturnsFileNameWithUnderbarPrepended) {
|
|
for (auto illegal_file_name : kIllegalPathNames) {
|
|
std::wstring parent_folder(L"");
|
|
|
|
std::wstring expected(default_download_path_);
|
|
expected.append(L"/_" + std::wstring(_wcslwr(
|
|
(std::wstring(illegal_file_name)).data())));
|
|
|
|
auto actual(FilePath::GetDownloadPath(
|
|
parent_folder,
|
|
std::wstring(_wcslwr((std::wstring(illegal_file_name)).data()))));
|
|
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPath_IllegalFileNameCharacters\
|
|
ReturnsFileNameWithUnderbarSubstituted) {
|
|
// char illegal_character_sequence[]{ 0x48, 0x65, 0x6c, 0x6c, 0x6f, 0x05,
|
|
// 0x77, 0x6f, 0x72, 0x6c, 0x64, 0x21, 0 };
|
|
auto illegal_character_sequence(L"Test\x5Test");
|
|
std::wstring parent_folder(L"");
|
|
|
|
std::wstring expected(default_download_path_);
|
|
expected.append(L"/Test_Test");
|
|
|
|
auto actual(FilePath::GetDownloadPath(
|
|
parent_folder, std::wstring(illegal_character_sequence)));
|
|
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPath_LowestIllegalFileNameCharacter\
|
|
ReturnsFileNameWithUnderbarSubstituted) {
|
|
// char illegal_character_sequence[]{ 0x48, 0x65, 0x6c, 0x6c, 0x6f, 0x01,
|
|
// 0x77, 0x6f, 0x72, 0x6c, 0x64, 0x21, 0 };
|
|
auto illegal_character_sequence(L"Test\x1Test");
|
|
|
|
std::wstring parent_folder(L"");
|
|
|
|
std::wstring expected(default_download_path_);
|
|
expected.append(L"/Test_Test");
|
|
|
|
auto actual(FilePath::GetDownloadPath(
|
|
parent_folder, std::wstring(illegal_character_sequence)));
|
|
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPath_HighestIllegalFileNameCharacter\
|
|
ReturnsFileNameWithUnderbarSubstituted) {
|
|
// char illegal_character_sequence[]{ 0x48, 0x65, 0x6c, 0x6c, 0x6f, 0x1f,
|
|
// 0x77, 0x6f, 0x72, 0x6c, 0x64, 0x21, 0 };
|
|
auto illegal_character_sequence(L"Test\x1fTest");
|
|
|
|
std::wstring parent_folder(L"");
|
|
|
|
std::wstring expected(default_download_path_);
|
|
expected.append(L"/Test_Test");
|
|
|
|
auto actual(FilePath::GetDownloadPath(
|
|
parent_folder, std::wstring(illegal_character_sequence)));
|
|
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPath_IllegalFileNameCharacterQuestionMark\
|
|
ReturnsFileNameWithUnderbarSubstituted) {
|
|
// char illegal_character_sequence[]{ 0x48, 0x65, 0x6c, 0x6c, 0x6f, 0x2f,
|
|
// 0x77, 0x6f, 0x72, 0x6c, 0x64, 0x21, 0 };
|
|
auto illegal_character_sequence(L"Test?Test");
|
|
|
|
std::wstring parent_folder(L"");
|
|
|
|
std::wstring expected(default_download_path_);
|
|
expected.append(L"/Test_Test");
|
|
|
|
auto actual(FilePath::GetDownloadPath(
|
|
parent_folder, std::wstring(illegal_character_sequence)));
|
|
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPath_IllegalFileNameCharacterAsterisk\
|
|
ReturnsFileNameWithUnderbarSubstituted) {
|
|
// char illegal_character_sequence[]{ 0x48, 0x65, 0x6c, 0x6c, 0x6f, 0x2f,
|
|
// 0x77, 0x6f, 0x72, 0x6c, 0x64, 0x21, 0 };
|
|
auto illegal_character_sequence(L"Test*Test");
|
|
|
|
std::wstring parent_folder(L"");
|
|
|
|
std::wstring expected(default_download_path_);
|
|
expected.append(L"/Test_Test");
|
|
|
|
auto actual(FilePath::GetDownloadPath(
|
|
parent_folder, std::wstring(illegal_character_sequence)));
|
|
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPath_IllegalFileNameCharacterLessThan\
|
|
ReturnsFileNameWithUnderbarSubstituted) {
|
|
// char illegal_character_sequence[]{ 0x48, 0x65, 0x6c, 0x6c, 0x6f, 0x2f,
|
|
// 0x77, 0x6f, 0x72, 0x6c, 0x64, 0x21, 0 };
|
|
auto illegal_character_sequence(L"Test<Test");
|
|
|
|
std::wstring parent_folder(L"");
|
|
|
|
std::wstring expected(default_download_path_);
|
|
expected.append(L"/Test_Test");
|
|
|
|
auto actual(FilePath::GetDownloadPath(
|
|
parent_folder, std::wstring(illegal_character_sequence)));
|
|
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPath_IllegalFileNameCharacterGreaterThan\
|
|
ReturnsFileNameWithUnderbarSubstituted) {
|
|
// char illegal_character_sequence[]{ 0x48, 0x65, 0x6c, 0x6c, 0x6f, 0x2f,
|
|
// 0x77, 0x6f, 0x72, 0x6c, 0x64, 0x21, 0 };
|
|
auto illegal_character_sequence(L"Test>Test");
|
|
|
|
std::wstring parent_folder(L"");
|
|
|
|
std::wstring expected(default_download_path_);
|
|
expected.append(L"/Test_Test");
|
|
|
|
auto actual(FilePath::GetDownloadPath(
|
|
parent_folder, std::wstring(illegal_character_sequence)));
|
|
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPath_IllegalFileNameCharacterVerticalBar\
|
|
ReturnsFileNameWithUnderbarSubstituted) {
|
|
// char illegal_character_sequence[]{ 0x48, 0x65, 0x6c, 0x6c, 0x6f, 0x2f,
|
|
// 0x77, 0x6f, 0x72, 0x6c, 0x64, 0x21, 0 };
|
|
auto illegal_character_sequence(L"Test|Test");
|
|
|
|
std::wstring parent_folder(L"");
|
|
|
|
std::wstring expected(default_download_path_);
|
|
expected.append(L"/Test_Test");
|
|
|
|
auto actual(FilePath::GetDownloadPath(
|
|
parent_folder, std::wstring(illegal_character_sequence)));
|
|
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPath_IllegalFileNameCharacterColon\
|
|
ReturnsFileNameWithUnderbarSubstituted) {
|
|
// char illegal_character_sequence[]{ 0x48, 0x65, 0x6c, 0x6c, 0x6f, 0x2f,
|
|
// 0x77, 0x6f, 0x72, 0x6c, 0x64, 0x21, 0 };
|
|
auto illegal_character_sequence(L"Test:Test");
|
|
|
|
std::wstring parent_folder(L"");
|
|
|
|
std::wstring expected(default_download_path_);
|
|
expected.append(L"/Test_Test");
|
|
|
|
auto actual(FilePath::GetDownloadPath(
|
|
parent_folder, std::wstring(illegal_character_sequence)));
|
|
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPath_FileDoesntExist\
|
|
ReturnsFileWithPassedName) {
|
|
std::wstring file_name(kFileName);
|
|
std::wstring parent_folder(L"");
|
|
|
|
std::wstring expected(default_download_path_);
|
|
expected.append(L"/");
|
|
expected.append(file_name);
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPath_FileExistsReturns\
|
|
FileWithIncrementedName) {
|
|
std::wstring file_name(kFileName);
|
|
std::wstring renamed_file_name(kFirstIterationFileName);
|
|
std::wstring parent_folder(L"");
|
|
|
|
std::wstring output_file_path(default_download_path_);
|
|
output_file_path.append(L"/");
|
|
output_file_path.append(file_name);
|
|
|
|
std::wstring expected(default_download_path_);
|
|
expected += renamed_file_name;
|
|
|
|
std::wifstream input_file;
|
|
std::wofstream output_file;
|
|
|
|
output_file.open(output_file_path,
|
|
std::ofstream::binary | std::ofstream::out);
|
|
|
|
ASSERT_TRUE(output_file.rdstate() == std::ofstream::goodbit);
|
|
|
|
output_file.close();
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(actual, expected);
|
|
|
|
// Remove the file and check that it is removed
|
|
// File 1
|
|
_wremove(output_file_path.c_str());
|
|
|
|
input_file.open(output_file_path, std::ifstream::binary | std::ifstream::in);
|
|
|
|
ASSERT_FALSE(input_file.rdstate() == std::ifstream::goodbit);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPath_MultipleFilesExist\
|
|
ReturnsNextIncrementedFileName) {
|
|
std::ofstream output_file;
|
|
std::ifstream input_file;
|
|
|
|
std::wstring file_name(kFileName);
|
|
std::wstring first_renamed_file_name(kFirstIterationFileName);
|
|
std::wstring second_renamed_file_name(kSecondIterationFileName);
|
|
|
|
std::wstring parent_folder(L"");
|
|
|
|
std::wstring expected(default_download_path_);
|
|
expected.append(second_renamed_file_name.c_str());
|
|
|
|
std::wstring output_file1_path(default_download_path_);
|
|
output_file1_path.append(L"/" + file_name);
|
|
|
|
std::wstring output_file2_path(default_download_path_);
|
|
output_file2_path.append(first_renamed_file_name);
|
|
|
|
// Create the test files
|
|
output_file.open(output_file1_path,
|
|
std::ofstream::binary | std::ofstream::out);
|
|
ASSERT_TRUE(output_file.rdstate() == std::ofstream::goodbit);
|
|
output_file.close();
|
|
output_file.clear();
|
|
|
|
output_file.open(output_file2_path,
|
|
std::ofstream::binary | std::ofstream::out);
|
|
ASSERT_TRUE(output_file.rdstate() == std::ofstream::goodbit);
|
|
output_file.close();
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(expected, actual);
|
|
|
|
// Remove the test files and check that it is removed
|
|
// File 1
|
|
_wremove(output_file1_path.c_str());
|
|
input_file.open(output_file1_path, std::ifstream::binary | std::ifstream::in);
|
|
|
|
ASSERT_FALSE(input_file.rdstate() == std::ifstream::goodbit);
|
|
|
|
// File 2
|
|
_wremove(output_file2_path.c_str());
|
|
|
|
input_file.clear();
|
|
input_file.open(output_file2_path, std::ifstream::binary | std::ifstream::in);
|
|
|
|
ASSERT_FALSE(input_file.rdstate() == std::ifstream::goodbit);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPath_FileNameContains\
|
|
MultipleDotsReturnsIncrementBeforeFirstDot) {
|
|
std::ifstream input_file;
|
|
std::ofstream output_file;
|
|
|
|
std::wstring file_name(kMultipleDotsFileName);
|
|
std::wstring renamed_file_name(kOneIterationMultipleDotsFileName);
|
|
|
|
std::wstring parent_folder(L"");
|
|
|
|
std::wstring output_file1_path(default_download_path_);
|
|
output_file1_path.append(L"/" + file_name);
|
|
|
|
std::wstring output_file2_path(default_download_path_);
|
|
output_file2_path.append(renamed_file_name);
|
|
|
|
std::wstring expected(default_download_path_);
|
|
expected.append(renamed_file_name);
|
|
|
|
output_file.open(output_file1_path,
|
|
std::ofstream::binary | std::ofstream::out);
|
|
ASSERT_TRUE(output_file.rdstate() == std::ofstream::goodbit);
|
|
output_file.close();
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(expected, actual);
|
|
|
|
_wremove(output_file1_path.c_str());
|
|
input_file.open(output_file1_path, std::ifstream::binary | std::ifstream::in);
|
|
|
|
ASSERT_FALSE(input_file.rdstate() == std::ifstream::goodbit);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPath_FileNameContainsNo\
|
|
DotsReturnsWithIncrementAtEnd) {
|
|
std::ifstream input_file;
|
|
std::ofstream output_file;
|
|
|
|
std::wstring file_name(kNoDotsFileName);
|
|
std::wstring renamed_file_name(kOneIterationNoDotsFileName);
|
|
|
|
std::wstring parent_folder(L"");
|
|
|
|
std::wstring output_file1_path(default_download_path_);
|
|
output_file1_path.append(L"/" + file_name);
|
|
|
|
std::wstring output_file2_path(default_download_path_);
|
|
output_file2_path.append(L"/" + renamed_file_name);
|
|
|
|
std::wstring expected(default_download_path_);
|
|
expected.append(renamed_file_name);
|
|
|
|
output_file.open(output_file1_path,
|
|
std::ofstream::binary | std::ofstream::out);
|
|
ASSERT_TRUE(output_file.rdstate() == std::ofstream::goodbit);
|
|
output_file.close();
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(expected, actual);
|
|
|
|
_wremove(output_file1_path.c_str());
|
|
input_file.open(output_file1_path, std::ifstream::binary | std::ifstream::in);
|
|
|
|
ASSERT_FALSE(input_file.rdstate() == std::ifstream::goodbit);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPath_FileNameExistsWith\
|
|
AHoleBetweenRenamedFiles) {
|
|
std::ifstream input_file;
|
|
std::ofstream output_file;
|
|
|
|
std::wstring file_name(kFileName);
|
|
std::wstring file_name1(kFirstIterationFileName);
|
|
std::wstring file_name2(kSecondIterationFileName);
|
|
std::wstring file_name3(kThirdIterationFileName);
|
|
|
|
std::wstring parent_folder(L"");
|
|
|
|
// Create the path for the original file name
|
|
std::wstring output_file_path(default_download_path_);
|
|
output_file_path.append(
|
|
L"/" +
|
|
file_name); // Original file name example: "increment_file_test.txt"
|
|
|
|
// Create the path for the first iteration of the original file name
|
|
std::wstring output_file1_path(default_download_path_);
|
|
output_file1_path.append(file_name1); // First iteration on original file
|
|
// name example:
|
|
// "increment_file_test (1).txt"
|
|
|
|
// Create the path for the third iteration of the original file name
|
|
std::wstring output_file3_path(default_download_path_);
|
|
output_file3_path.append(
|
|
file_name3); // Third iteration on original file
|
|
// name example: "increment_file_test (3).txt"
|
|
|
|
// Create the expected result which is the second iteration of the original
|
|
// file name
|
|
std::wstring expected(default_download_path_);
|
|
expected.append(file_name2); // Second iteration on original file name
|
|
// example: "increment_file_test (2).txt"
|
|
|
|
// Create the original file
|
|
output_file.open(output_file_path,
|
|
std::ofstream::binary | std::ofstream::out);
|
|
ASSERT_TRUE(output_file.rdstate() == std::ofstream::goodbit);
|
|
output_file.close();
|
|
|
|
// Create the first iteration of the original file
|
|
output_file.clear();
|
|
output_file.open(output_file1_path,
|
|
std::ofstream::binary | std::ofstream::out);
|
|
ASSERT_TRUE(output_file.rdstate() == std::ofstream::goodbit);
|
|
output_file.close();
|
|
|
|
// Create the third iteration of the original file
|
|
output_file.clear();
|
|
output_file.open(output_file3_path,
|
|
std::ofstream::binary | std::ofstream::out);
|
|
ASSERT_TRUE(output_file.rdstate() == std::ofstream::goodbit);
|
|
output_file.close();
|
|
|
|
// This should return the second iteration of the original file
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(expected, actual);
|
|
|
|
// Delete the original file
|
|
_wremove(output_file_path.c_str());
|
|
input_file.open(output_file_path, std::ifstream::binary | std::ifstream::in);
|
|
ASSERT_FALSE(input_file.rdstate() == std::ifstream::goodbit);
|
|
|
|
// Delete the first iteration of the original file
|
|
input_file.clear(); // Reset the input_file state
|
|
_wremove(output_file1_path.c_str());
|
|
input_file.open(output_file1_path, std::ifstream::binary | std::ifstream::in);
|
|
ASSERT_FALSE(input_file.rdstate() == std::ifstream::goodbit);
|
|
|
|
// Delete the third iteration of the original file
|
|
input_file.clear(); // Reset the input_file state
|
|
_wremove(output_file3_path.c_str());
|
|
input_file.open(output_file3_path, std::ifstream::binary | std::ifstream::in);
|
|
ASSERT_FALSE(input_file.rdstate() == std::ifstream::goodbit);
|
|
}
|
|
|
|
void SetCheckIllegalCharactersFlag(bool value) {
|
|
NearbyFlags::GetInstance().OverrideBoolFlagValue(
|
|
nearby::connections::config_package_nearby::nearby_connections_feature::
|
|
kCheckIllegalCharacters,
|
|
value);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPathWithFileName\
|
|
FileNameTwoDotsFrontShouldReturnBaseDownloadPathWithFileNameTwoDotsFront) {
|
|
std::wstring parent_folder(L"");
|
|
std::wstring file_name(kFileNameWithFrontTwoDots);
|
|
|
|
std::wstringstream path(L"");
|
|
path << default_download_path_ << L"/" << file_name;
|
|
|
|
std::wstring expected = path.str();
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_NE(actual, expected);
|
|
|
|
SetCheckIllegalCharactersFlag(true);
|
|
actual = FilePath::GetDownloadPath(parent_folder, file_name);
|
|
SetCheckIllegalCharactersFlag(false);
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPathWithFileName\
|
|
FileNameThreeDotsShouldReturnBaseDownloadPathWithUnderscore) {
|
|
SetCheckIllegalCharactersFlag(true);
|
|
std::wstring parent_folder(L"");
|
|
std::wstring file_name(kFileNameWithThreeDots);
|
|
|
|
std::wstringstream path(L"");
|
|
path << default_download_path_ << L"/.._";
|
|
|
|
std::wstring expected = path.str();
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
SetCheckIllegalCharactersFlag(false);
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPath_FileExistsReturns\
|
|
FileWithIncrementedNameWithNull) {
|
|
SetCheckIllegalCharactersFlag(true);
|
|
|
|
std::wstring file_name(kFileName);
|
|
int size = file_name.size();
|
|
file_name.append(L"1.txt");
|
|
file_name[size] = L'\x00';
|
|
std::wstring renamed_file_name(kFileNameWithNullReplaced);
|
|
std::wstring parent_folder(L"");
|
|
|
|
std::wstring output_file_path(default_download_path_);
|
|
output_file_path.append(L"/");
|
|
output_file_path.append(file_name);
|
|
|
|
std::wstring expected(default_download_path_);
|
|
expected += renamed_file_name;
|
|
|
|
std::wifstream input_file;
|
|
std::wofstream output_file;
|
|
|
|
output_file.open(output_file_path,
|
|
std::ofstream::binary | std::ofstream::out);
|
|
|
|
ASSERT_TRUE(output_file.rdstate() == std::ofstream::goodbit);
|
|
|
|
output_file.close();
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
EXPECT_EQ(actual, expected);
|
|
|
|
// Remove the file and check that it is removed
|
|
// File 1
|
|
_wremove(output_file_path.c_str());
|
|
|
|
input_file.open(output_file_path, std::ifstream::binary | std::ifstream::in);
|
|
|
|
SetCheckIllegalCharactersFlag(false);
|
|
ASSERT_FALSE(input_file.rdstate() == std::ifstream::goodbit);
|
|
}
|
|
|
|
TEST_F(FilePathTests, GetDownloadPathWithFileName\
|
|
ParentFolderWithADotShouldBeReplaceedWithUnderscore) {
|
|
SetCheckIllegalCharactersFlag(true);
|
|
std::wstring parent_folder(L"test/./folder/");
|
|
std::wstring file_name(kFileName);
|
|
|
|
std::wstringstream path(L"");
|
|
path << default_download_path_ << L"/test/_/folder" << L"/" << kFileName;
|
|
|
|
std::wstring expected = path.str();
|
|
|
|
auto actual(FilePath::GetDownloadPath(parent_folder, file_name));
|
|
|
|
SetCheckIllegalCharactersFlag(false);
|
|
EXPECT_EQ(actual, expected);
|
|
}
|
|
} // namespace windows
|
|
} // namespace nearby
|