You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
84 lines
2.5 KiB
C++
84 lines
2.5 KiB
C++
/*
|
|
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
|
|
* its licensors.
|
|
*
|
|
* For complete copyright and license terms please see the LICENSE at the root of this
|
|
* distribution (the "License"). All use of this software is governed by the License,
|
|
* or, if provided, by the license below or the license accompanying this file. Do not
|
|
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
*
|
|
*/
|
|
|
|
#include "Utils.h"
|
|
#include <AzCore/IO/Path/Path.h>
|
|
#include <AzCore/StringFunc/StringFunc.h>
|
|
|
|
UnitTest::ScopedTemporaryDirectory::ScopedTemporaryDirectory()
|
|
{
|
|
constexpr int MaxAttempts = 255;
|
|
#if !AZ_TRAIT_USE_POSIX_TEMP_FOLDER
|
|
const auto userTempFolder = std::filesystem::temp_directory_path();
|
|
#else
|
|
AZ::IO::Path userTempFolder("/tmp");
|
|
#endif
|
|
|
|
for (int i = 0; i < MaxAttempts; ++i)
|
|
{
|
|
auto randomFolder = AZ::Uuid::CreateRandom().ToString<AZStd::fixed_string<512>>(false, false);
|
|
AZ::IO::FixedMaxPath testPath;
|
|
#if !AZ_TRAIT_USE_POSIX_TEMP_FOLDER
|
|
auto path = userTempFolder / ("UnitTest-" + randomFolder).c_str();
|
|
testPath = path.string().c_str();
|
|
#else
|
|
userTempFolder /= ("UnitTest-" + randomFolder).c_str();
|
|
testPath = userTempFolder.c_str();
|
|
#endif
|
|
if (!AZ::IO::SystemFile::Exists(testPath.c_str()))
|
|
{
|
|
#if !AZ_TRAIT_USE_POSIX_TEMP_FOLDER
|
|
m_path = path;
|
|
m_tempDirectory = m_path.string().c_str();
|
|
#else
|
|
m_tempDirectory = testPath;
|
|
#endif
|
|
m_directoryExists = AZ::IO::SystemFile::CreateDir(m_tempDirectory.c_str());
|
|
break;
|
|
}
|
|
}
|
|
|
|
AZ_Error("ScopedTemporaryDirectory", !m_tempDirectory.empty(), "Failed to create unique temporary directory after attempting %d random folder names", MaxAttempts);
|
|
}
|
|
|
|
UnitTest::ScopedTemporaryDirectory::~ScopedTemporaryDirectory()
|
|
{
|
|
if (m_directoryExists)
|
|
{
|
|
AZ::IO::SystemFile::DeleteDir(m_tempDirectory.c_str());
|
|
}
|
|
}
|
|
|
|
bool UnitTest::ScopedTemporaryDirectory::IsValid() const
|
|
{
|
|
return m_directoryExists;
|
|
}
|
|
|
|
const char* UnitTest::ScopedTemporaryDirectory::GetDirectory() const
|
|
{
|
|
return m_tempDirectory.c_str();
|
|
}
|
|
|
|
|
|
#if !AZ_TRAIT_USE_POSIX_TEMP_FOLDER
|
|
const std::filesystem::path& UnitTest::ScopedTemporaryDirectory::GetPath() const
|
|
{
|
|
return m_path;
|
|
}
|
|
|
|
std::filesystem::path UnitTest::ScopedTemporaryDirectory::operator/(const std::filesystem::path& rhs) const
|
|
{
|
|
return m_path / rhs;
|
|
}
|
|
|
|
#endif // !AZ_TRAIT_USE_POSIX_TEMP_FOLDER
|