xvfb_test_script.py

#!/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.

"""Simple script for xvfb_unittest to launch.

This script outputs formatted data to stdout for the xvfb unit tests
to read and compare with expected output.
"""

import os
import signal
import sys
import time


def print_signal(sig, *_):
  print 'Signal :{}'.format(sig)


if __name__ == '__main__':
  signal.signal(signal.SIGTERM, print_signal)
  signal.signal(signal.SIGINT, print_signal)

  # test if inside xvfb flag is set.
  print 'Inside_xvfb :{}'.format(
      os.environ.get('_CHROMIUM_INSIDE_XVFB', 'None'))
  # test the subprocess display number.
  print 'Display :{}'.format(os.environ.get('DISPLAY', 'None'))

  if len(sys.argv) > 1 and sys.argv[1] == '--sleep':
    time.sleep(2)  # gives process time to receive signal.