test/CC/CCFLAGS.py

   1 #!/usr/bin/env python
   2 #
   3 # MIT License
   4 #
   5 # Copyright The SCons Foundation
   6 #
   7 # Permission is hereby granted, free of charge, to any person obtaining
   8 # a copy of this software and associated documentation files (the
   9 # "Software"), to deal in the Software without restriction, including
  10 # without limitation the rights to use, copy, modify, merge, publish,
  11 # distribute, sublicense, and/or sell copies of the Software, and to
  12 # permit persons to whom the Software is furnished to do so, subject to
  13 # the following conditions:
  14 #
  15 # The above copyright notice and this permission notice shall be included
  16 # in all copies or substantial portions of the Software.
  17 #
  18 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY
  19 # KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
  20 # WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  21 # NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  22 # LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  23 # OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  24 # WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  25
  26 import sys
  27 import TestSCons
  28
  29 _obj = TestSCons._obj
  30
  31 if sys.platform == 'win32':
  32     import SCons.Tool.MSCommon as msc
  33
  34     if not msc.msvc_exists():
  35         fooflags = '-DFOO'
  36         barflags = '-DBAR'
  37     else:
  38         fooflags = '/nologo -DFOO'
  39         barflags = '/nologo -DBAR'
  40 else:
  41     fooflags = '-DFOO'
  42     barflags = '-DBAR'
  43
  44 test = TestSCons.TestSCons()
  45
  46 test.write('SConstruct', """
  47 DefaultEnvironment(tool=[])
  48 foo = Environment(CCFLAGS = '%s')
  49 bar = Environment(CCFLAGS = '%s')
  50 foo.Object(target = 'foo%s', source = 'prog.c')
  51 bar.Object(target = 'bar%s', source = 'prog.c')
  52 foo.Program(target = 'foo', source = 'foo%s')
  53 bar.Program(target = 'bar', source = 'bar%s')
  54 foo.Program(target = 'prog', source = 'prog.c',
  55             CCFLAGS = '$CCFLAGS -DBAR $BAZ', BAZ = '-DBAZ')
  56 """ % (fooflags, barflags, _obj, _obj, _obj, _obj))
  57
  58 test.write('prog.c', r"""
  59 #include <stdio.h>
  60 #include <stdlib.h>
  61
  62 int
  63 main(int argc, char *argv[])
  64 {
  65         argv[argc++] = "--";
  66 #ifdef FOO
  67         printf("prog.c:  FOO\n");
  68 #endif
  69 #ifdef BAR
  70         printf("prog.c:  BAR\n");
  71 #endif
  72 #ifdef BAZ
  73         printf("prog.c:  BAZ\n");
  74 #endif
  75         exit (0);
  76 }
  77 """)
  78
  79
  80 test.run(arguments = '.')
  81
  82 test.run(program = test.workpath('foo'), stdout = "prog.c:  FOO\n")
  83 test.run(program = test.workpath('bar'), stdout = "prog.c:  BAR\n")
  84 test.run(program = test.workpath('prog'), stdout = """\
  85 prog.c:  FOO
  86 prog.c:  BAR
  87 prog.c:  BAZ
  88 """)
  89
  90 test.write('SConstruct', """
  91 DefaultEnvironment(tool=[])
  92
  93 bar = Environment(CCFLAGS = '%s')
  94 bar.Object(target = 'foo%s', source = 'prog.c')
  95 bar.Object(target = 'bar%s', source = 'prog.c')
  96 bar.Program(target = 'foo', source = 'foo%s')
  97 bar.Program(target = 'bar', source = 'bar%s')
  98 """ % (barflags, _obj, _obj, _obj, _obj))
  99
 100 test.run(arguments = '.')
 101
 102 test.run(program = test.workpath('foo'), stdout = "prog.c:  BAR\n")
 103 test.run(program = test.workpath('bar'), stdout = "prog.c:  BAR\n")
 104
 105 test.pass_test()
 106
 107 # Local Variables:
 108 # tab-width:4
 109 # indent-tabs-mode:nil
 110 # End:
 111 # vim: set expandtab tabstop=4 shiftwidth=4: