tests/testrun.py

   1 #!/usr/bin/env python
   2 from basetest import BaseTest
   3 import os, sys, subprocess
   4 import unittest
   5 from StringIO import StringIO
   6
   7 sys.path.insert(0, '..')
   8
   9 from zeroinstall.injector import policy, run, handler, model
  10 from zeroinstall import SafeException
  11
  12 mydir = os.path.abspath(os.path.dirname(__file__))
  13 local_0launch = os.path.join(os.path.dirname(mydir), '0launch')
  14 runnable = os.path.join(mydir, 'runnable', 'Runnable.xml')
  15 recursive_runner = os.path.join(mydir, 'runnable', 'RecursiveRunner.xml')
  16 command_feed = os.path.join(mydir, 'Command.xml')
  17
  18 class TestRun(BaseTest):
  19         def testRunnable(self):
  20                 child = subprocess.Popen([local_0launch, '--', runnable, 'user-arg'], stdout = subprocess.PIPE)
  21                 stdout, _ = child.communicate()
  22                 assert 'Runner: script=A test script: args=command-arg -- user-arg' in stdout, stdout
  23
  24         def testCommandBindings(self):
  25                 h = handler.Handler()
  26                 p = policy.Policy(command_feed, handler = h)
  27                 h.wait_for_blocker(p.solve_with_downloads())
  28                 old_stdout = sys.stdout
  29                 try:
  30                         sys.stdout = StringIO()
  31                         run.execute_selections(p.solver.selections, [], main = 'runner', dry_run = True)
  32                 finally:
  33                         sys.stdout = old_stdout
  34                 assert 'local' in os.environ['LOCAL'], os.environ['LOCAL']
  35
  36         def testAbsMain(self):
  37                 h = handler.Handler()
  38                 p = policy.Policy(command_feed, handler = h)
  39                 h.wait_for_blocker(p.solve_with_downloads())
  40
  41                 old_stdout = sys.stdout
  42                 try:
  43                         sys.stdout = StringIO()
  44                         run.execute_selections(p.solver.selections, [], main = '/runnable/runner', dry_run = True)
  45                 finally:
  46                         sys.stdout = old_stdout
  47
  48                 try:
  49                         old_stdout = sys.stdout
  50                         try:
  51                                 sys.stdout = StringIO()
  52                                 run.execute_selections(p.solver.selections, [], main = '/runnable/not-there', dry_run = True)
  53                         finally:
  54                                 sys.stdout = old_stdout
  55                 except SafeException, ex:
  56                         assert 'not-there' in unicode(ex)
  57
  58         def testArgs(self):
  59                 h = handler.Handler()
  60                 p = policy.Policy(runnable, handler = h)
  61                 h.wait_for_blocker(p.solve_with_downloads())
  62                 old_stdout = sys.stdout
  63                 try:
  64                         sys.stdout = StringIO()
  65                         run.execute_selections(p.solver.selections, [], dry_run = True)
  66                         out = sys.stdout.getvalue()
  67                 finally:
  68                         sys.stdout = old_stdout
  69                 assert 'runner-arg' in out, out
  70
  71         def testWrapper(self):
  72                 h = handler.Handler()
  73                 p = policy.Policy(runnable, handler = h)
  74                 h.wait_for_blocker(p.solve_with_downloads())
  75                 old_stdout = sys.stdout
  76                 try:
  77                         sys.stdout = StringIO()
  78                         run.execute_selections(p.solver.selections, [], wrapper = 'echo', dry_run = True)
  79                         out = sys.stdout.getvalue()
  80                 finally:
  81                         sys.stdout = old_stdout
  82                 assert '/bin/sh -c echo "$@"' in out, out
  83                 assert 'runner-arg' in out, out
  84                 assert 'script' in out, out
  85
  86         def testRecursive(self):
  87                 child = subprocess.Popen([local_0launch, '--', recursive_runner, 'user-arg'], stdout = subprocess.PIPE)
  88                 stdout, _ = child.communicate()
  89                 assert 'Runner: script=A test script: args=command-arg -- arg-for-runnable recursive-arg -- user-arg' in stdout, stdout
  90
  91 if __name__ == '__main__':
  92         unittest.main()