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
112
113
|
import os
import sys
import unittest
raise unittest.SkipTest("FIXME! broken test see: https://github.com/python/cpython/pull/109004")
from test.support import os_helper
from test.support.os_helper import TESTFN, TESTFN_ASCII
if sys.platform != "win32":
raise unittest.SkipTest("windows related tests")
import _winapi
import msvcrt;
from _testconsole import write_input
class TestFileOperations(unittest.TestCase):
def test_locking(self):
with open(TESTFN, "w") as f:
self.addCleanup(os_helper.unlink, TESTFN)
msvcrt.locking(f.fileno(), msvcrt.LK_LOCK, 1)
self.assertRaises(OSError, msvcrt.locking, f.fileno(), msvcrt.LK_NBLCK, 1)
def test_unlockfile(self):
with open(TESTFN, "w") as f:
self.addCleanup(os_helper.unlink, TESTFN)
msvcrt.locking(f.fileno(), msvcrt.LK_LOCK, 1)
msvcrt.locking(f.fileno(), msvcrt.LK_UNLCK, 1)
msvcrt.locking(f.fileno(), msvcrt.LK_LOCK, 1)
def test_setmode(self):
with open(TESTFN, "w") as f:
self.addCleanup(os_helper.unlink, TESTFN)
msvcrt.setmode(f.fileno(), os.O_BINARY)
msvcrt.setmode(f.fileno(), os.O_TEXT)
def test_open_osfhandle(self):
h = _winapi.CreateFile(TESTFN_ASCII, _winapi.GENERIC_WRITE, 0, 0, 1, 128, 0)
self.addCleanup(os_helper.unlink, TESTFN_ASCII)
try:
fd = msvcrt.open_osfhandle(h, os.O_RDONLY)
h = None
os.close(fd)
finally:
if h:
_winapi.CloseHandle(h)
def test_get_osfhandle(self):
with open(TESTFN, "w") as f:
self.addCleanup(os_helper.unlink, TESTFN)
msvcrt.get_osfhandle(f.fileno())
c = '\u5b57' # unicode CJK char (meaning 'character') for 'wide-char' tests
c_encoded = b'\x57\x5b' # utf-16-le (which windows internally used) encoded char for this CJK char
class TestConsoleIO(unittest.TestCase):
def test_kbhit(self):
self.assertEqual(msvcrt.kbhit(), 0)
def test_getch(self):
msvcrt.ungetch(b'c')
self.assertEqual(msvcrt.getch(), b'c')
def test_getwch(self):
stdin = open('CONIN$', 'r')
old_stdin = sys.stdin
try:
sys.stdin = stdin
write_input(stdin.buffer.raw, c_encoded)
self.assertEqual(msvcrt.getwch(), c)
finally:
sys.stdin = old_stdin
def test_getche(self):
msvcrt.ungetch(b'c')
self.assertEqual(msvcrt.getche(), b'c')
def test_getwche(self):
stdin = open('CONIN$', 'r')
old_stdin = sys.stdin
try:
sys.stdin = stdin
write_input(stdin.buffer.raw, c_encoded)
self.assertEqual(msvcrt.getwche(), c)
finally:
sys.stdin = old_stdin
def test_putch(self):
msvcrt.putch(b'c')
def test_putwch(self):
msvcrt.putwch(c)
class TestOther(unittest.TestCase):
def test_heap_min(self):
try:
msvcrt.heapmin()
except OSError:
pass
if __name__ == "__main__":
unittest.main()
|
