git.cworth.org Git - apitrace-tests/blob - cli_driver.py

   1 #!/usr/bin/env python
   2
   3 # Copyright 2012 Intel Corporation
   4 #
   5 # Permission is hereby granted, free of charge, to any person
   6 # obtaining a copy of this software and associated documentation files
   7 # (the "Software"), to deal in the Software without restriction,
   8 # including without limitation the rights to use, copy, modify, merge,
   9 # publish, distribute, sublicense, and/or sell copies of the Software,
  10 # and to permit persons to whom the Software is furnished to do so,
  11 # subject to the following conditions:
  12 #
  13 # The above copyright notice and this permission notice shall be
  14 # included in all copies or substantial portions of the Software.
  15 #
  16 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  17 # EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  18 # MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  19 # NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
  20 # BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
  21 # ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  22 # CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  23 # SOFTWARE.
  24
  25 '''Test driver for scripts in the cli directory.'''
  26
  27 import json, errno, shutil, subprocess
  28
  29 from base_driver import *
  30
  31 class CliDriver(Driver):
  32
  33     def do_apitrace(self, args):
  34         cmd = [self.options.apitrace] + args[1:]
  35
  36         print " ".join(cmd)
  37         proc = subprocess.Popen(cmd, stdout = subprocess.PIPE)
  38         self.output = proc.communicate()[0]
  39
  40         proc.wait()
  41
  42         if (self.expect_failure):
  43             if (proc.returncode == 0):
  44                 fail("Command unexpectedly passed when expecting failure:\n    " + " ".join(cmd))
  45         else:
  46             if (proc.returncode != 0):
  47                 fail("Command failed (returned non-zero):\n    " + " ".join(cmd))
  48
  49     def do_expect(self, args):
  50         expected = json.loads(" ".join(args[1:]))
  51         if (self.output != expected):
  52             fail("Unexpected output:\n    Expected: %s\n    Received: %s\n" % (expected, self.output))
  53
  54     def do_rm_and_mkdir(self, args):
  55
  56         # Operate only on local directories
  57         dir = './' + args[1]
  58
  59         # Failing to delete a directory that doesn't exist is no failure
  60         def rmtree_onerror(function, path, excinfo):
  61             if excinfo[0] == OSError and excinfo[1].errno != errno.ENOENT:
  62                 raise
  63
  64         shutil.rmtree(dir, onerror = rmtree_onerror)
  65         os.makedirs(dir)
  66
  67     def unknown_command(self, args):
  68         fail('Broken test script: Unknown command: %s' % (args[0]))
  69
  70     def run_script(self, cli_script):
  71         "Execute the commands in the given cli script."
  72
  73         commands = {
  74             'apitrace': self.do_apitrace,
  75             'expect': self.do_expect,
  76             'rm_and_mkdir': self.do_rm_and_mkdir
  77         }
  78
  79         script = open(cli_script, 'rt')
  80
  81         while True:
  82
  83             self.expect_failure = False
  84
  85             line = script.readline()
  86
  87             # Exit loop on EOF
  88             if (line == ''):
  89                 break
  90
  91             cmd = line.split()
  92
  93             # Ignore blank lines and comments
  94             if (len(cmd) == 0 or line[0] == '#'):
  95                 continue
  96
  97             if (cmd[0] == 'EXPECT_FAILURE:'):
  98                 self.expect_failure = True
  99                 cmd.pop(0)
 100
 101             commands.get(cmd[0], self.unknown_command)(cmd)
 102
 103     def run(self):
 104         self.parseOptions()
 105
 106         self.run_script(self.args[0])
 107
 108         pass_()
 109
 110 if __name__ == '__main__':
 111     CliDriver().run()