test/Flatten.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 """
  28 Test that the Flatten() function is available and works.
  29 """
  30
  31 import TestSCons
  32
  33 test = TestSCons.TestSCons()
  34
  35 test.subdir('work')
  36
  37 test.write(['work', 'SConstruct'], """
  38 def cat(env, source, target):
  39     target = str(target[0])
  40     with open(target, "wb") as ofp:
  41         for src in source:
  42             with open(str(src), "rb") as ifp:
  43                 ofp.write(ifp.read())
  44 env = Environment(BUILDERS={'Cat':Builder(action=cat)})
  45 f1 = env.Cat('../file1.out', 'file1.in')
  46 f2 = env.Cat('../file2.out', ['file2a.in', 'file2b.in'])
  47 print(list(map(str, Flatten(['begin', f1, 'middle', f2, 'end']))))
  48 print(list(map(str, env.Flatten([f1, [['a', 'b'], 'c'], f2]))))
  49 SConscript('SConscript', "env")
  50 """)
  51
  52 test.write(['work', 'SConscript'], """
  53 Import("env")
  54 print(Flatten([1, [2, 3], 4]))
  55 print(env.Flatten([[[[1], 2], 3], 4]))
  56 """)
  57
  58 test.write('file1.in', "file1.in\n")
  59 test.write('file2a.in', "file2a.in\n")
  60 test.write('file2b.in', "file2b.in\n")
  61
  62 def double_backslash(f):
  63     p = test.workpath(f)
  64     return p.replace('\\', '\\\\')
  65
  66 expect = """\
  67 ['begin', '%s', 'middle', '%s', 'end']
  68 ['%s', 'a', 'b', 'c', '%s']
  69 [1, 2, 3, 4]
  70 [1, 2, 3, 4]
  71 """ % (double_backslash('file1.out'), double_backslash('file2.out'),
  72        double_backslash('file1.out'), double_backslash('file2.out'))
  73
  74 test.run(chdir = "work",
  75          arguments = ".",
  76          stdout = test.wrap_stdout(read_str = expect,
  77                                    build_str = "scons: `.' is up to date.\n"))
  78
  79 test.pass_test()
  80
  81 # Local Variables:
  82 # tab-width:4
  83 # indent-tabs-mode:nil
  84 # End:
  85 # vim: set expandtab tabstop=4 shiftwidth=4: