test/ENV.py

   1 #!/usr/bin/env python
   2 #
   3 # __COPYRIGHT__
   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 __revision__ = "__FILE__ __REVISION__ __DATE__ __DEVELOPER__"
  25
  26 import os
  27 import TestSCons
  28
  29 _python_ = TestSCons._python_
  30
  31 test = TestSCons.TestSCons()
  32
  33 bin1_build_py = test.workpath('bin1', 'build.py')
  34 bin2_build_py = test.workpath('bin2', 'build.py')
  35
  36 test.write('SConstruct', """
  37 import os
  38 Bld = Builder(action = r'%(_python_)s build.py $TARGET $SOURCES')
  39 env1 = Environment(BUILDERS = { 'Bld' : Bld })
  40 env2 = Environment(BUILDERS = { 'Bld' : Bld })
  41 env1['ENV']['X'] = 'env1'
  42 env2['ENV']['X'] = 'env2'
  43 env1.Bld(target = 'env1.out', source = 'input')
  44 env2.Bld(target = 'env2.out', source = 'input')
  45 """ % locals())
  46
  47 test.write('build.py',
  48 r"""\
  49 import os
  50 import sys
  51 with open(sys.argv[2], 'r') as f:
  52     contents = f.read()
  53 with open(sys.argv[1], 'w') as f:
  54     f.write("build.py %s\n%s" % (os.environ['X'], contents))
  55 """)
  56
  57 test.write('input', "input file\n")
  58
  59 test.run(arguments = '.')
  60
  61 test.must_match('env1.out', "build.py env1\ninput file\n", mode='r')
  62 test.must_match('env2.out', "build.py env2\ninput file\n", mode='r')
  63
  64
  65 test.write('SConstruct', """
  66 env = Environment()
  67 foo = env.Command('foo', [], r'%(_python_)s build.py $TARGET')
  68 env['ENV']['LIST'] = [foo, 'bar']
  69 env['ENV']['FOO'] = foo
  70 """ % locals())
  71
  72 test.write('build.py', r"""
  73 import os
  74 print('LIST:', os.environ['LIST'])
  75 print('FOO:', os.environ['FOO'])
  76 """)
  77
  78 test.run()
  79
  80 expect = [
  81     "LIST: foo%sbar" % os.pathsep,
  82     "FOO: foo",
  83 ]
  84
  85 test.must_contain_all_lines(test.stdout(), expect)
  86
  87 test.pass_test()
  88
  89 # Local Variables:
  90 # tab-width:4
  91 # indent-tabs-mode:nil
  92 # End:
  93 # vim: set expandtab tabstop=4 shiftwidth=4: