examples: fix whitespace
[dpdk.git] / app / cmdline_test / cmdline_test.py
1 #!/usr/bin/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
36 import sys, pexpect, string, os, cmdline_test_data
37
38 #
39 # function to run test
40 #
41 def runTest(child,test):
42         child.send(test["Sequence"])
43         if test["Result"] == None:
44                 return 0
45         child.expect(test["Result"],1)
46
47 #
48 # history test is a special case
49 #
50 # This test does the following:
51 # 1) fills the history with garbage up to its full capacity
52 #    (just enough to remove last entry)
53 # 2) scrolls back history to the very beginning
54 # 3) checks if the output is as expected, that is, the first
55 #    number in the sequence (not the last entry before it)
56 #
57 # This is a self-contained test, it needs only a pexpect child
58 #
59 def runHistoryTest(child):
60         # find out history size
61         child.sendline(cmdline_test_data.CMD_GET_BUFSIZE)
62         child.expect("History buffer size: \\d+", timeout=1)
63         history_size = int(child.after[len(cmdline_test_data.BUFSIZE_TEMPLATE):])
64         i = 0
65
66         # fill the history with numbers
67         while i < history_size / 10:
68                 # add 1 to prevent from parsing as octals
69                 child.send("1" + str(i).zfill(8) + cmdline_test_data.ENTER)
70                 # the app will simply print out the number
71                 child.expect(str(i + 100000000), timeout=1)
72                 i += 1
73         # scroll back history
74         child.send(cmdline_test_data.UP * (i + 2) + cmdline_test_data.ENTER)
75         child.expect("100000000", timeout=1)
76
77 # the path to cmdline_test executable is supplied via command-line.
78 if len(sys.argv) < 2:
79         print "Error: please supply cmdline_test app path"
80         sys.exit(1)
81
82 test_app_path = sys.argv[1]
83
84 if not os.path.exists(test_app_path):
85         print "Error: please supply cmdline_test app path"
86         sys.exit(1)
87
88 child = pexpect.spawn(test_app_path)
89
90 print "Running command-line tests..."
91 for test in cmdline_test_data.tests:
92         print (test["Name"] + ":").ljust(30),
93         try:
94                 runTest(child,test)
95                 print "PASS"
96         except:
97                 print "FAIL"
98                 print child
99                 sys.exit(1)
100
101 # since last test quits the app, run new instance
102 child = pexpect.spawn(test_app_path)
103
104 print ("History fill test:").ljust(30),
105 try:
106         runHistoryTest(child)
107         print "PASS"
108 except:
109         print "FAIL"
110         print child
111         sys.exit(1)
112 child.close()
113 sys.exit(0)