test/ninja/shutdown_scons_daemon.py

   1 #!/usr/bin/env python
   2 #
   3 # Copyright The SCons Foundation
   4 #
   5 # Permission is hereby granted, free of charge, to any person obtaining
   6 # a copy of this software and associated documentation files (the
   7 # "Software"), to deal in the Software without restriction, including
   8 # without limitation the rights to use, copy, modify, merge, publish,
   9 # distribute, sublicense, and/or sell copies of the Software, and to
  10 # permit persons to whom the Software is furnished to do so, subject to
  11 # the following conditions:
  12 #
  13 # The above copyright notice and this permission notice shall be included
  14 # in all copies or substantial portions of the Software.
  15 #
  16 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY
  17 # KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
  18 # WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  19 # NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  20 # LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  21 # OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  22 # WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  23 #
  24
  25 import os
  26 from timeit import default_timer as timer
  27
  28 import TestSCons
  29 from TestCmd import IS_WINDOWS
  30
  31 test = TestSCons.TestSCons()
  32
  33 try:
  34     import ninja
  35 except ImportError:
  36     test.skip_test("Could not find ninja module in python")
  37
  38 try:
  39     import psutil
  40 except ImportError:
  41     test.skip_test("Could not find psutil module in python")
  42
  43
  44 _python_ = TestSCons._python_
  45 _exe = TestSCons._exe
  46
  47 ninja_bin = os.path.abspath(
  48     os.path.join(ninja.__file__, os.pardir, "data", "bin", "ninja" + _exe)
  49 )
  50
  51 test.dir_fixture("ninja-fixture")
  52
  53 test.file_fixture(
  54     "ninja_test_sconscripts/sconstruct_force_scons_callback", "SConstruct"
  55 )
  56
  57 test.run(stdout=None)
  58 pid = None
  59 test.must_exist(test.workpath('.ninja/scons_daemon_dirty'))
  60 with open(test.workpath('.ninja/scons_daemon_dirty')) as f:
  61     pid = int(f.read())
  62     if pid not in [proc.pid for proc in psutil.process_iter()]:
  63         test.fail_test(message="daemon not running!")
  64
  65 program = test.workpath("run_ninja_env.bat") if IS_WINDOWS else ninja_bin
  66 test.run(program=program, arguments='shutdown-ninja-scons-daemon', stdout=None)
  67
  68 wait_time = 10
  69 start_time = timer()
  70 while True:
  71     if wait_time > (timer() - start_time):
  72         if pid not in [proc.pid for proc in psutil.process_iter()]:
  73             break
  74     else:
  75         test.fail_test(message=f"daemon still not shutdown after {wait_time} seconds.")
  76
  77 test.must_not_exist(test.workpath('.ninja/scons_daemon_dirty'))
  78 test.pass_test()
  79
  80 # Local Variables:
  81 # tab-width:4
  82 # indent-tabs-mode:nil
  83 # End:
  84 # vim: set expandtab tabstop=4 shiftwidth=4: