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
|
import subprocess
import time
from unittest.mock import MagicMock
import pytest
from briefcase.integrations.subprocess import Subprocess
# hardcoded here since subprocess will only include these constants if Python is literally on Windows
CREATE_NO_WINDOW = 0x8000000
CREATE_NEW_PROCESS_GROUP = 0x200
@pytest.fixture
def streaming_process():
"""Mock process returned for Popen context manager in
Subprocess._run_and_stream_output."""
process = MagicMock()
# Mock the readline values of an actual process. The final return value is "",
# indicating that the process has exited; however, we insert a short sleep
# to ensure that any other threads will have a chance to run before this
# thread actually terminates.
def mock_readline():
yield from [
"output line 1\n",
"\n",
"output line 3\n",
]
time.sleep(0.1)
yield ""
process.stdout.readline.side_effect = mock_readline()
process.poll.return_value = -3
return process
@pytest.fixture
def mock_sub(mock_tools, streaming_process):
mock_tools.os.environ = {
"VAR1": "Value 1",
"PS1": "\nLine 2\n\nLine 4",
"PWD": "/home/user/",
}
sub = Subprocess(mock_tools)
sub._subprocess = MagicMock(spec=subprocess)
run_result = MagicMock(spec=subprocess.CompletedProcess)
run_result.returncode = 0
sub._subprocess.run.return_value = run_result
sub._subprocess.check_output.return_value = "some output line 1\nmore output line 2"
sub._subprocess.CREATE_NO_WINDOW = CREATE_NO_WINDOW
sub._subprocess.CREATE_NEW_PROCESS_GROUP = CREATE_NEW_PROCESS_GROUP
# Return the streaming process for output streaming
sub._subprocess.Popen.return_value.__enter__.return_value = streaming_process
return sub
|
