1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
from pypy.module.cpyext.test.test_cpyext import AppTestCpythonExtensionBase
from pypy.module.cpyext.test.test_api import BaseApiTest
from rpython.rtyper.lltypesystem import rffi
class AppTestSysModule(AppTestCpythonExtensionBase):
def test_sysmodule(self):
module = self.import_extension('foo', [
("get", "METH_VARARGS",
"""
char *name = _PyUnicode_AsString(PyTuple_GetItem(args, 0));
PyObject *retval = PySys_GetObject(name);
return PyBool_FromLong(retval != NULL);
""")])
assert module.get("excepthook")
assert not module.get("spam_spam_spam")
def test_writestdout(self):
module = self.import_extension('foo', [
("writestdout", "METH_NOARGS",
"""
PySys_WriteStdout("format: %d\\n", 42);
Py_RETURN_NONE;
""")])
import sys, io
prev = sys.stdout
sys.stdout = io.StringIO()
try:
module.writestdout()
assert sys.stdout.getvalue() == "format: 42\n"
finally:
sys.stdout = prev
def test_sysgetset(self):
module = self.import_extension('foo', [
("setobject", "METH_VARARGS",
"""
const char *name;
Py_ssize_t size;
PyObject *value = NULL;
if (!PyArg_ParseTuple(args, "z#|O", &name, &size, &value)) {
return NULL;
}
if (value == Py_None) {
value = NULL;
}
int ret = PySys_SetObject(name, value);
if (ret < 0)
return NULL;
return PyLong_FromLong(ret);
""")], PY_SSIZE_T_CLEAN=True)
import sys
module.setobject("newattr", 1)
assert sys.newattr == 1
module.setobject("newattr")
assert not hasattr(sys, 'newattr')
# Cpython lets you call this, even if the attribute does not exist
module.setobject("newattr")
with raises(UnicodeDecodeError):
module.setobject(b'\xff', 1)
|
