Brian Herrera
48 lines
1.8 KiB
Python
Executable File
48 lines
1.8 KiB
Python
Executable File
"""
|
|
All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
|
|
its licensors.
|
|
|
|
For complete copyright and license terms please see the LICENSE at the root of this
|
|
distribution (the "License"). All use of this software is governed by the License,
|
|
or, if provided, by the license below or the license accompanying this file. Do not
|
|
remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
"""
|
|
import os
|
|
import subprocess
|
|
|
|
import pytest
|
|
|
|
import ly_test_tools.environment.process_utils as process_utils
|
|
import ly_test_tools.environment.waiter as waiter
|
|
from ly_test_tools import WINDOWS
|
|
|
|
if WINDOWS:
|
|
pytestmark = pytest.mark.SUITE_smoke
|
|
else:
|
|
pytestmark = pytest.mark.skipif(not WINDOWS, reason="Only runs on Windows")
|
|
|
|
|
|
class TestSubprocessCheckOutputWrapper(object):
|
|
|
|
def test_KillWindowsProgram_WindowsProgramStarted_KilledSuccessfully(self):
|
|
windows_program = 'timeout.exe'
|
|
windows_directory = os.environ.get('windir')
|
|
command = [
|
|
os.path.join(f'{windows_directory}', 'System32', f'{windows_program}'),
|
|
'/T', # Timeout flag
|
|
'4', # 4 seconds
|
|
]
|
|
|
|
def process_killed():
|
|
return not process_utils.process_exists(windows_program, ignore_extensions=True)
|
|
|
|
assert os.path.exists(command[0]), (
|
|
f'The {windows_program} executable does not exist at: {command[0]}'
|
|
)
|
|
|
|
with subprocess.Popen(command, creationflags=subprocess.CREATE_NEW_CONSOLE) as process:
|
|
if process_utils.process_exists(windows_program, ignore_extensions=True):
|
|
process_utils.kill_process_with_pid(process.pid)
|
|
waiter.wait_for(process_killed, timeout=2) # Raises exception if the process is alive.
|