3fade1f647
syntax"
This is a reland of commit 8defc6c6da
Uses `sys.path.append(...)` to import the `common` module in
testing/scripts/run_variations_smoke_tests.py (as is done elsewhere in
the original CL) and resolve errors thrown for attempting a relative
import with no known parent package.
Original change's description:
>and Pylint2.7, and fix resulting errors.
>
>Bug: 1262363
>Change-Id: I09628427dd4caf9d873f3b489b897054430f8567
>Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/3557490
>Reviewed-by: Ben Pastene <bpastene@chromium.org>
>Commit-Queue: Joshua Hood <jdh@chromium.org>
>Cr-Commit-Position: refs/heads/main@{#996200}
Bug: 1262363
Change-Id: I60fe062f62962d0b3dc4e93d9e29626ed181f688
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/3610783
Reviewed-by: Ben Pastene <bpastene@chromium.org>
Commit-Queue: Joshua Hood <jdh@chromium.org>
Cr-Commit-Position: refs/heads/main@{#999432}
99 lines
2.9 KiB
Python
Executable File
99 lines
2.9 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# Copyright (c) 2019 The Chromium Authors. All rights reserved.
|
|
# Use of this source code is governed by a BSD-style license that can be
|
|
# found in the LICENSE file.
|
|
|
|
"""Unit tests for test_env.py functionality.
|
|
|
|
Each unit test is launches python process that uses test_env.py
|
|
to launch another python process. Then signal handling and
|
|
propagation is tested. This similates how Swarming uses test_env.py.
|
|
"""
|
|
|
|
import os
|
|
import signal
|
|
import subprocess
|
|
import sys
|
|
import time
|
|
import unittest
|
|
|
|
HERE = os.path.dirname(os.path.abspath(__file__))
|
|
TEST_SCRIPT = os.path.join(HERE, 'test_env_user_script.py')
|
|
|
|
# pylint: disable=super-with-arguments
|
|
|
|
|
|
def launch_process_windows(args):
|
|
# The `universal_newlines` option is equivalent to `text` in Python 3.
|
|
return subprocess.Popen(
|
|
[sys.executable, TEST_SCRIPT] + args,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.STDOUT,
|
|
env=os.environ.copy(),
|
|
creationflags=subprocess.CREATE_NEW_PROCESS_GROUP,
|
|
universal_newlines=True)
|
|
|
|
|
|
def launch_process_nonwindows(args):
|
|
# The `universal_newlines` option is equivalent to `text` in Python 3.
|
|
return subprocess.Popen(
|
|
[sys.executable, TEST_SCRIPT] + args,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.STDOUT,
|
|
env=os.environ.copy(),
|
|
universal_newlines=True)
|
|
|
|
|
|
# pylint: disable=inconsistent-return-statements
|
|
def read_subprocess_message(proc, starts_with):
|
|
"""Finds the value after first line prefix condition."""
|
|
for line in proc.stdout:
|
|
if line.startswith(starts_with):
|
|
return line.rstrip().replace(starts_with, '')
|
|
# pylint: enable=inconsistent-return-statements
|
|
|
|
|
|
def send_and_wait(proc, sig, sleep_time=0.3):
|
|
"""Sends a signal to subprocess."""
|
|
time.sleep(sleep_time) # gives process time to launch.
|
|
os.kill(proc.pid, sig)
|
|
proc.wait()
|
|
|
|
|
|
class SignalingWindowsTest(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
super(SignalingWindowsTest, self).setUp()
|
|
if sys.platform != 'win32':
|
|
self.skipTest('test only runs on Windows')
|
|
|
|
def test_send_ctrl_break_event(self):
|
|
proc = launch_process_windows([])
|
|
send_and_wait(proc, signal.CTRL_BREAK_EVENT) # pylint: disable=no-member
|
|
sig = read_subprocess_message(proc, 'Signal :')
|
|
self.assertEqual(sig, str(int(signal.SIGBREAK))) # pylint: disable=no-member
|
|
|
|
|
|
class SignalingNonWindowsTest(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
super(SignalingNonWindowsTest, self).setUp()
|
|
if sys.platform == 'win32':
|
|
self.skipTest('test does not run on Windows')
|
|
|
|
def test_send_sigterm(self):
|
|
proc = launch_process_nonwindows([])
|
|
send_and_wait(proc, signal.SIGTERM)
|
|
sig = read_subprocess_message(proc, 'Signal :')
|
|
self.assertEqual(sig, str(int(signal.SIGTERM)))
|
|
|
|
def test_send_sigint(self):
|
|
proc = launch_process_nonwindows([])
|
|
send_and_wait(proc, signal.SIGINT)
|
|
sig = read_subprocess_message(proc, 'Signal :')
|
|
self.assertEqual(sig, str(int(signal.SIGINT)))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|