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
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
# -*- coding: utf-8 -*-
"""
PC-BASIC test.session
unit tests for session API
(c) 2020--2023 Rob Hagemans
This file is released under the GNU GPL version 3 or later.
"""
import os
import io
import sys
import unittest
from pcbasic import Session
from pcbasic.basic import BASICError
from tests.unit.utils import TestCase, run_tests
from pcbasic.data import read_codepage
from pcbasic.compat import PY2, WIN32
def pythoncall(script):
return '"{0}" "{1}"'.format(sys.executable, script)
class DosTest(TestCase):
"""Unit tests for DOS module."""
tag = u'dos'
@unittest.skipIf(PY2, 'shell codepage agreement known not to work in Python 2.')
@unittest.skipIf(WIN32, 'shell codepage agreement known not to work on Windows.')
def test_shell(self):
"""Test SHELL statement with commands."""
helper = os.path.join(os.path.dirname(__file__), 'simple_shell_helper.py')
with Session(shell=pythoncall(helper), codepage=read_codepage('850')) as s:
# outputs come through stdout
s.execute(u'SHELL "echo 1"')
# test non-ascii char
s.execute(u'SHELL "echo £"')
# outputs come through stderr
s.execute(u'SHELL "x"')
# test non-ascii char
s.execute(u'SHELL "£"')
outstr = self.get_text_stripped(s)[:4]
# this fails on Windows, we're getting \xa3 (latin-1 for £)
# instead of \xc9 (per cp850, our local codepage))
assert outstr == [b'1', b'\x9c', b"'x' is not recognised.", b"'\x9c' is not recognised."], outstr
@unittest.skipIf(PY2, 'shell codepage agreement known not to work in Python 2.')
def test_shell_utf16(self):
"""Test SHELL statement to utf-16 script with commands."""
helper = os.path.join(os.path.dirname(__file__), 'simple_shell_helper.py')
with Session(shell=pythoncall(helper) + ' -u', codepage=read_codepage('850')) as s:
# outputs come through stdout
s.execute(u'SHELL "echo 1"')
# test non-ascii char
s.execute(u'SHELL "echo £"')
# outputs come through stderr
s.execute(u'SHELL "x"')
# test non-ascii char
s.execute(u'SHELL "£"')
outstr = self.get_text_stripped(s)[:4]
assert outstr == [b'1', b'\x9c', b"'x' is not recognised.", b"'\x9c' is not recognised."], outstr
def test_no_shell(self):
"""Test SHELL statement with no shell specified."""
with Session() as s:
# assertRaises doesn't work as the error is absorbed by the session
#with self.assertRaises(BASICError):
s.execute(u'SHELL "echo 1"')
assert self.get_text_stripped(s)[0] == b'Illegal function call\xff'
def test_bad_shell(self):
"""Test SHELL statement with nonexistant shell specified."""
with Session(shell='_this_does_not_exist_') as s:
s.execute(u'SHELL "echo 1"')
assert self.get_text_stripped(s)[0] == b'Illegal function call\xff'
@unittest.skipIf(PY2, 'shell codepage agreement known not to work in Python 2.')
def test_interactive_shell(self):
"""Test SHELL statement with interaction."""
helper = os.path.join(os.path.dirname(__file__), 'simple_shell_helper.py')
with Session(shell=pythoncall(helper), codepage=read_codepage('850')) as s:
s.press_keys(u'echo _check_for_this_')
# test backspace
s.press_keys(u'\rexix\bt\r')
s.execute(u'SHELL')
# output is messy due to race between press_keys and shell thread, but this should work
assert b'_check_for_this' in self.get_text_stripped(s)[1]
@unittest.skipIf(PY2, 'shell codepage agreement known not to work in Python 2.')
def test_interactive_shell_no_lf_at_end(self):
"""Test SHELL statement with interaction, helper script ends without LF."""
helper = os.path.join(os.path.dirname(__file__), 'simple_shell_helper.py')
with Session(shell=pythoncall(helper)+ ' -b') as s:
s.press_keys(u'exit\r')
s.execute(u'SHELL')
assert self.get_text_stripped(s)[1] == b'Bye!'
def test_environ(self):
"""Test ENVIRON statement."""
with Session() as s:
s.execute(u'ENVIRON "test=ok"')
assert s.evaluate(u'ENVIRON$("test")') == b'ok'
assert s.evaluate(u'ENVIRON$("TEST")') == b'ok'
assert s.evaluate(u'ENVIRON$("Test")') == b'ok'
s.execute(u'ENVIRON "TEST=OK"')
assert s.evaluate(u'ENVIRON$("test")') == b'OK'
assert s.evaluate(u'ENVIRON$("TEST")') == b'OK'
assert s.evaluate(u'ENVIRON$("Test")') == b'OK'
def test_environ_noascii_key(self):
"""Test ENVIRON statement with non-ascii key."""
with Session() as s:
s.execute(u'ENVIRON "t£st=ok"')
assert self.get_text_stripped(s)[0] == b'Illegal function call\xff'
def test_environ_fn_noascii_key(self):
"""Test ENVIRON$ function with non-ascii key."""
with Session() as s:
s.evaluate(u'ENVIRON$("t£st")')
assert self.get_text_stripped(s)[0] == b'Illegal function call\xff'
def test_environ_noascii_value(self):
"""Test ENVIRON statement with non-ascii values."""
with Session() as s:
s.execute(u'ENVIRON "TEST=£"')
assert self.get_text_stripped(s)[0] == b''
if __name__ == '__main__':
run_tests()
|
