test/Rpcgen/RPCGEN.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 _exe=TestSCons._exe
  31
  32 test=TestSCons.TestSCons()
  33
  34
  35
  36 test.write('myrpcgen.py', """
  37 import getopt
  38 import sys
  39 cmd_opts, args = getopt.getopt(sys.argv[1:], 'chlmo:', [])
  40 for opt, arg in cmd_opts:
  41     if opt == '-o': out = arg
  42 with open(out, 'w') as ofp:
  43     ofp.write(" ".join(sys.argv) + "\\n")
  44     for a in args:
  45         with open(a, 'r') as ifp:
  46             contents = ifp.read()
  47         ofp.write(contents.replace('RPCGEN', 'myrpcgen.py'))
  48 sys.exit(0)
  49 """)
  50
  51 test.write('SConstruct', """\
  52 env = Environment(RPCGEN = r'%(_python_)s myrpcgen.py',
  53                   tools=['default', 'rpcgen'])
  54 env.RPCGenHeader('rpcif')
  55 env.RPCGenClient('rpcif')
  56 env.RPCGenService('rpcif')
  57 env.RPCGenXDR('rpcif')
  58 """ % locals())
  59
  60 test.write('rpcif.x', """\
  61 RPCGEN
  62 """)
  63
  64 test.run()
  65
  66 output = "myrpcgen.py %s -o %s rpcif.x\nmyrpcgen.py\n"
  67 expect_clnt = output % ('-l', test.workpath('rpcif_clnt.c'))
  68 expect_h = output % ('-h', test.workpath('rpcif.h'))
  69 expect_svc = output % ('-m', test.workpath('rpcif_svc.c'))
  70 expect_xdr = output % ('-c', test.workpath('rpcif_xdr.c'))
  71
  72 test.must_contain('rpcif_clnt.c', expect_clnt, mode='r')
  73 test.must_contain('rpcif.h', expect_h, mode='r')
  74 test.must_contain('rpcif_svc.c', expect_svc, mode='r')
  75 test.must_contain('rpcif_xdr.c', expect_xdr, mode='r')
  76
  77
  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: