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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
import pytest
from pypy.module.cpyext.test.test_api import BaseApiTest
from pypy.module.cpyext.object import Py_PRINT_RAW
from pypy.interpreter.error import OperationError
from rpython.rtyper.lltypesystem import rffi
from rpython.tool.udir import udir
class TestFile(BaseApiTest):
def test_file_fromstring(self, space, api):
filename = rffi.str2charp(str(udir / "_test_file"))
mode = rffi.str2charp("wb")
w_file = api.PyFile_FromString(filename, mode)
rffi.free_charp(filename)
rffi.free_charp(mode)
space.call_method(w_file, "write", space.newbytes("text"))
space.call_method(w_file, "close")
assert (udir / "_test_file").read() == "text"
def test_file_getline(self, space, api):
filename = rffi.str2charp(str(udir / "_test_file"))
mode = rffi.str2charp("w")
w_file = api.PyFile_FromString(filename, mode)
space.call_method(w_file, "write",
space.wrap("line1\nline2\nline3\nline4"))
space.call_method(w_file, "close")
rffi.free_charp(mode)
mode = rffi.str2charp("r")
w_file = api.PyFile_FromString(filename, mode)
rffi.free_charp(filename)
rffi.free_charp(mode)
w_line = api.PyFile_GetLine(w_file, 0)
assert space.text_w(w_line) == "line1\n"
w_line = api.PyFile_GetLine(w_file, 4)
assert space.text_w(w_line) == "line"
w_line = api.PyFile_GetLine(w_file, 0)
assert space.text_w(w_line) == "2\n"
# XXX We ought to raise an EOFError here, but don't
w_line = api.PyFile_GetLine(w_file, -1)
# assert api.PyErr_Occurred() is space.w_EOFError
assert space.text_w(w_line) == "line3\n"
space.call_method(w_file, "close")
def test_file_fromfd(self, space, api):
name = str(udir / "_test_file")
with rffi.scoped_str2charp(name) as filename:
with rffi.scoped_str2charp("wb") as mode:
w_file = api.PyFile_FromString(filename, mode)
fp = space.int_w(w_file.fileno_w(space))
assert fp is not None
w_file2 = api.PyFile_FromFd(fp, filename, mode, -1, None, None, None, 1)
assert w_file2 is not None
@pytest.mark.xfail
def test_file_setbufsize(self, space, api):
api.PyFile_SetBufSize()
def test_file_writestring(self, space, api, capfd):
w_stdout = space.sys.get("stdout")
with rffi.scoped_str2charp("test\n") as s:
api.PyFile_WriteString(s, w_stdout)
space.call_method(w_stdout, "flush")
out, err = capfd.readouterr()
out = out.replace('\r\n', '\n')
assert out == "test\n"
def test_file_writeobject(self, space, api, capfd):
w_obj = space.wrap("test\n")
w_stdout = space.sys.get("stdout")
api.PyFile_WriteObject(w_obj, w_stdout, Py_PRINT_RAW)
api.PyFile_WriteObject(w_obj, w_stdout, 0)
space.call_method(w_stdout, "flush")
out, err = capfd.readouterr()
out = out.replace('\r\n', '\n')
assert out == "test\n'test\\n'"
def test_fspath(self, space, api):
w_obj = space.newtext("test")
w_ret = api.PyOS_FSPath(w_obj)
assert space.eq_w(w_ret, w_obj)
w_obj = space.newint(3)
with pytest.raises(OperationError):
w_ret = api.PyOS_FSPath(w_obj)
w_p1 = space.appexec([], '''():
class Pathlike():
def __fspath__(self):
return 'test'
return Pathlike()''')
w_p2 = space.appexec([], '''():
class UnPathlike():
def __fspath__(self):
return 42
return UnPathlike()''')
w_ret = api.PyOS_FSPath(w_p1)
assert space.eq_w(w_ret, space.newtext('test'))
with pytest.raises(OperationError):
w_ret = api.PyOS_FSPath(w_p2)
|
