test/cmdline_test/cmdline_test.py

   1 #!/usr/bin/env python
   2
   3 #   BSD LICENSE
   4 #
   5 #   Copyright(c) 2010-2014 Intel Corporation. All rights reserved.
   6 #   All rights reserved.
   7 #
   8 #   Redistribution and use in source and binary forms, with or without
   9 #   modification, are permitted provided that the following conditions
  10 #   are met:
  11 #
  12 #     * Redistributions of source code must retain the above copyright
  13 #       notice, this list of conditions and the following disclaimer.
  14 #     * Redistributions in binary form must reproduce the above copyright
  15 #       notice, this list of conditions and the following disclaimer in
  16 #       the documentation and/or other materials provided with the
  17 #       distribution.
  18 #     * Neither the name of Intel Corporation nor the names of its
  19 #       contributors may be used to endorse or promote products derived
  20 #       from this software without specific prior written permission.
  21 #
  22 #   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  23 #   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  24 #   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  25 #   A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  26 #   OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  27 #   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  28 #   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  29 #   DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  30 #   THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  31 #   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  32 #   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  33
  34 # Script that runs cmdline_test app and feeds keystrokes into it.
  35 from __future__ import print_function
  36 import cmdline_test_data
  37 import os
  38 import pexpect
  39 import sys
  40
  41
  42 #
  43 # function to run test
  44 #
  45 def runTest(child, test):
  46     child.send(test["Sequence"])
  47     if test["Result"] is None:
  48         return 0
  49     child.expect(test["Result"], 1)
  50
  51
  52 #
  53 # history test is a special case
  54 #
  55 # This test does the following:
  56 # 1) fills the history with garbage up to its full capacity
  57 #    (just enough to remove last entry)
  58 # 2) scrolls back history to the very beginning
  59 # 3) checks if the output is as expected, that is, the first
  60 #    number in the sequence (not the last entry before it)
  61 #
  62 # This is a self-contained test, it needs only a pexpect child
  63 #
  64 def runHistoryTest(child):
  65     # find out history size
  66     child.sendline(cmdline_test_data.CMD_GET_BUFSIZE)
  67     child.expect("History buffer size: \\d+", timeout=1)
  68     history_size = int(child.after[len(cmdline_test_data.BUFSIZE_TEMPLATE):])
  69     i = 0
  70
  71     # fill the history with numbers
  72     while i < history_size / 10:
  73         # add 1 to prevent from parsing as octals
  74         child.send("1" + str(i).zfill(8) + cmdline_test_data.ENTER)
  75         # the app will simply print out the number
  76         child.expect(str(i + 100000000), timeout=1)
  77         i += 1
  78     # scroll back history
  79     child.send(cmdline_test_data.UP * (i + 2) + cmdline_test_data.ENTER)
  80     child.expect("100000000", timeout=1)
  81
  82 # the path to cmdline_test executable is supplied via command-line.
  83 if len(sys.argv) < 2:
  84     print("Error: please supply cmdline_test app path")
  85     sys.exit(1)
  86
  87 test_app_path = sys.argv[1]
  88
  89 if not os.path.exists(test_app_path):
  90     print("Error: please supply cmdline_test app path")
  91     sys.exit(1)
  92
  93 child = pexpect.spawn(test_app_path)
  94
  95 print("Running command-line tests...")
  96 for test in cmdline_test_data.tests:
  97     testname = (test["Name"] + ":").ljust(30)
  98     try:
  99         runTest(child, test)
 100         print(testname, "PASS")
 101     except:
 102         print(testname, "FAIL")
 103         print(child)
 104         sys.exit(1)
 105
 106 # since last test quits the app, run new instance
 107 child = pexpect.spawn(test_app_path)
 108
 109 testname = ("History fill test:").ljust(30)
 110 try:
 111     runHistoryTest(child)
 112     print(testname, "PASS")
 113 except:
 114     print(testname, "FAIL")
 115     print(child)
 116     sys.exit(1)
 117 child.close()
 118 sys.exit(0)