vtk-m/vtkm/cont/testing/UnitTestError.cxx
2021-04-16 17:46:43 +00:00

84 lines
2.3 KiB
C++

//============================================================================
// Copyright (c) Kitware, Inc.
// All rights reserved.
// See LICENSE.txt for details.
//
// This software is distributed WITHOUT ANY WARRANTY; without even
// the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
// PURPOSE. See the above copyright notice for more information.
//============================================================================
#include <vtkm/cont/Error.h>
#include <vtkm/cont/ErrorBadValue.h>
#include <vtkm/cont/Logging.h>
#include <vtkm/cont/testing/Testing.h>
namespace
{
void RecursiveFunction(int recurse)
{
if (recurse < 5)
{
RecursiveFunction(++recurse);
}
else
{
throw vtkm::cont::ErrorBadValue("Too much recursion");
}
}
void ValidateError(const vtkm::cont::Error& error)
{
std::string message = "Too much recursion";
std::string stackTrace = error.GetStackTrace();
std::stringstream stackTraceStream(stackTrace);
std::string tmp;
size_t count = 0;
while (std::getline(stackTraceStream, tmp))
{
count++;
}
// StackTrace may be unavailable on certain Devices
if (stackTrace == "(Stack trace unavailable)")
{
VTKM_TEST_ASSERT(count == 1, "Logging disabled, stack trace shouldn't be available");
}
else
{
#if defined(NDEBUG)
// The compiler can optimize out the recursion and other function calls in release
// mode, but the backtrace should contain atleast one entry.
std::string assert_msg = "No entries in the stack frame\n" + stackTrace;
VTKM_TEST_ASSERT(count >= 1, assert_msg);
#else
std::string assert_msg = "Expected more entries in the stack frame\n" + stackTrace;
VTKM_TEST_ASSERT(count > 2, assert_msg);
#endif
}
VTKM_TEST_ASSERT(test_equal(message, error.GetMessage()), "Message was incorrect");
VTKM_TEST_ASSERT(test_equal(message + "\n" + stackTrace, std::string(error.what())),
"what() was incorrect");
}
void DoErrorTest()
{
VTKM_LOG_S(vtkm::cont::LogLevel::Info, "Check base error messages");
try
{
RecursiveFunction(0);
}
catch (const vtkm::cont::Error& e)
{
ValidateError(e);
}
}
} // anonymous namespace
int UnitTestError(int argc, char* argv[])
{
return vtkm::cont::testing::Testing::Run(DoErrorTest, argc, argv);
}