test/redirection.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
  25 __revision__ = "__FILE__ __REVISION__ __DATE__ __DEVELOPER__"
  26
  27 import TestSCons
  28
  29 _python_ = TestSCons._python_
  30
  31 test = TestSCons.TestSCons()
  32
  33 test.write('cat.py', r"""
  34 import sys
  35
  36 # write binary to stdout
  37 if sys.platform == "win32":
  38     import os, msvcrt
  39     msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY)
  40     msvcrt.setmode(sys.stdin.fileno(), os.O_BINARY)
  41
  42 try:
  43     with open(sys.argv[1], 'rb') as f:
  44         indata = f.read()
  45 except IndexError:
  46     indata = sys.stdin.buffer.read()
  47
  48 sys.stdout.buffer.write(indata)
  49 sys.exit(0)
  50 """)
  51
  52 test.write('SConstruct', r"""
  53 env = Environment()
  54 env.Command(target='foo1', source='bar1',
  55             action= r'%(_python_)s cat.py $SOURCES > $TARGET')
  56 env.Command(target='foo2', source='bar2',
  57             action= r'%(_python_)s cat.py < $SOURCES > $TARGET')
  58 env.Command(target='foo3', source='bar3',
  59             action=r'%(_python_)s cat.py $SOURCES | %(_python_)s cat.py > $TARGET')
  60 env.Command(target='foo4', source='bar4',
  61             action=r'%(_python_)s cat.py <$SOURCES |%(_python_)s cat.py >$TARGET')
  62 """ % locals())
  63
  64 test.write('bar1', 'bar1\r\n')
  65 test.write('bar2', 'bar2\r\n')
  66 test.write('bar3', 'bar3\r\n')
  67 test.write('bar4', 'bar4\r\n')
  68
  69 test.run(arguments='.')
  70
  71 test.must_match('foo1', 'bar1\r\n')
  72 test.must_match('foo2', 'bar2\r\n')
  73 test.must_match('foo3', 'bar3\r\n')
  74 test.must_match('foo4', 'bar4\r\n')
  75
  76 test.pass_test()
  77
  78 # Local Variables:
  79 # tab-width:4
  80 # indent-tabs-mode:nil
  81 # End:
  82 # vim: set expandtab tabstop=4 shiftwidth=4: