mirror of
https://github.com/OrcaSlicer/OrcaSlicer.git
synced 2026-10-07 15:51:08 +00:00
71 lines
2.1 KiB
C++
71 lines
2.1 KiB
C++
#include <catch2/catch_all.hpp>
|
|
#include <catch2/catch_test_macros.hpp>
|
|
|
|
#include <slic3r/plugin/PluginFsUtils.hpp>
|
|
#include <slic3r/plugin/PluginManager.hpp>
|
|
#include <slic3r/plugin/PythonInterpreter.hpp>
|
|
|
|
#include "plugin_test_utils.hpp"
|
|
|
|
#include <exception>
|
|
#include <nlohmann/json.hpp>
|
|
#include <pybind11/embed.h>
|
|
#include <pybind11/gil.h>
|
|
#include <pybind11/pytypes.h>
|
|
|
|
using namespace Slic3r;
|
|
|
|
namespace {
|
|
// Brings the embedded interpreter up for one test and tears it down before boost::log does,
|
|
// mirroring the ScopedPluginManager idiom in the other plugin tests.
|
|
struct ScopedPluginManager
|
|
{
|
|
ScopedDataDir python_data_dir{"plugin-json-depth"};
|
|
bool initialized = PluginManager::instance().initialize();
|
|
~ScopedPluginManager()
|
|
{
|
|
PluginManager::instance().shutdown();
|
|
PythonInterpreter::instance().shutdown();
|
|
}
|
|
};
|
|
} // namespace
|
|
|
|
TEST_CASE("py_to_json raises instead of overflowing on pathologically deep input", "[PluginHost][Python]")
|
|
{
|
|
ScopedPluginManager manager;
|
|
REQUIRE(manager.initialized);
|
|
namespace py = pybind11;
|
|
py::gil_scoped_acquire gil;
|
|
|
|
// [[[ ... 0 ... ]]] nested 300 deep: past the 200 conversion-depth cap, but shallow enough
|
|
// that the pre-fix code returns without crashing, so a regression fails cleanly rather than
|
|
// taking the process down. Built in C++ so the test does not depend on Python builtins.
|
|
py::object deep = py::int_(0);
|
|
for (int i = 0; i < 300; ++i) {
|
|
py::list wrapper;
|
|
wrapper.append(deep);
|
|
deep = std::move(wrapper);
|
|
}
|
|
CHECK_THROWS_AS(py_to_json(deep), std::exception);
|
|
}
|
|
|
|
TEST_CASE("py_to_json still converts reasonably nested input", "[PluginHost][Python]")
|
|
{
|
|
ScopedPluginManager manager;
|
|
REQUIRE(manager.initialized);
|
|
namespace py = pybind11;
|
|
py::gil_scoped_acquire gil;
|
|
|
|
py::dict d;
|
|
d["a"] = py::int_(1);
|
|
py::list inner;
|
|
inner.append(py::str("x"));
|
|
inner.append(py::int_(2));
|
|
d["b"] = inner;
|
|
|
|
const nlohmann::json j = py_to_json(d);
|
|
CHECK(j.at("a").get<std::int64_t>() == 1);
|
|
CHECK(j.at("b").at(0).get<std::string>() == "x");
|
|
CHECK(j.at("b").at(1).get<std::int64_t>() == 2);
|
|
}
|