cyphesis/tests/PythonArithmeticScriptTest.cpp
2018-05-07 11:15:26 +02:00

151 lines
4.1 KiB
C++

// Cyphesis Online RPG Server and AI Engine
// Copyright (C) 2009 Alistair Riddoch
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software Foundation,
// Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
#ifdef NDEBUG
#undef NDEBUG
#endif
#ifndef DEBUG
#define DEBUG
#endif
#include <Python.h>
#include "python_testers.h"
#include "rulesets/PythonArithmeticScript.h"
#include "common/compose.hpp"
#include "common/log.h"
#include <cassert>
#include <rulesets/Python_Script_Utils.h>
static PyMethodDef no_methods[] = {
{nullptr, nullptr} /* Sentinel */
};
PyObject * Get_PyClass(PyObject * module,
const std::string & package,
const std::string & type);
PyObject * Get_PyModule(const std::string & package)
{
PyObject * package_name = PyUnicode_FromString((char *)package.c_str());
PyObject * module = PyImport_Import(package_name);
Py_DECREF(package_name);
return module;
}
static PyObject* init_testmod() {
static struct PyModuleDef def = {
PyModuleDef_HEAD_INIT,
"testmod",
nullptr,
0,
no_methods,
nullptr,
nullptr,
nullptr,
nullptr
};
return PyModule_Create(&def);
}
int main()
{
{
PyImport_AppendInittab("testmod", &init_testmod);
Py_Initialize();
run_python_string("import testmod");
run_python_string("class TestArithmeticScript(object):\n"
" def __init__(self):\n"
" self.foo=1\n"
" self.bar=1.1\n"
" self.baz=None\n"
" self.qux='1'\n"
);
run_python_string("testmod.TestArithmeticScript=TestArithmeticScript");
auto testmod = Get_PyModule("testmod");
assert(testmod);
PyObject * clss = Get_PyClass(testmod, "testmod", "TestArithmeticScript");
assert(clss != 0);
PyObject * instance = PyEval_CallFunction(clss,"()");
assert(instance != 0);
PythonArithmeticScript pas(instance);
float val;
pas.attribute("foo", val);
pas.attribute("bar", val);
pas.attribute("baz", val);
pas.attribute("qux", val);
pas.attribute("nonexistent", val);
pas.set("foo", 1.0f);
pas.set("mim", 1.0f);
}
Py_Finalize();
return 0;
}
// stubs
PyObject * Get_PyClass(PyObject * module,
const std::string & package,
const std::string & type)
{
PyObject * py_class = PyObject_GetAttrString(module, (char *)type.c_str());
if (py_class == nullptr) {
log(ERROR, String::compose("Could not find python class \"%1.%2\"",
package, type));
PyErr_Print();
return nullptr;
}
if (PyCallable_Check(py_class) == 0) {
log(ERROR, String::compose("Could not instance python class \"%1.%2\"",
package, type));
Py_DECREF(py_class);
return nullptr;
}
if (PyType_Check(py_class) == 0) {
log(ERROR, String::compose("PyCallable_Check returned true, "
"but PyType_Check returned false \"%1.%2\"",
package, type));
Py_DECREF(py_class);
return nullptr;
}
return py_class;
}
ArithmeticScript::~ArithmeticScript()
{
}
void log(LogLevel lvl, const std::string & msg)
{
}