Move files to AzFramework/Test and AzToolsFramework/Test accordingly

Signed-off-by: Esteban Papp <81431996+amznestebanpapp@users.noreply.github.com>
2021-07-09 08:07:47 -07:00
parent f278bbcd65
commit afd97abdcf
62 changed files with 0 additions and 0 deletions
@@ -0,0 +1,78 @@
+/*
+ * Copyright (c) Contributors to the Open 3D Engine Project. For complete copyright and license terms please see the LICENSE at the root of this distribution.
+ * 
+ * SPDX-License-Identifier: Apache-2.0 OR MIT
+ *
+ */
+
+#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