node test scripts

diff --git a/qaapi/qa/tests/node/busyloop.py b/qaapi/qa/tests/node/busyloop.py
new file mode 100755 (executable)
index 0000000..8cff8a5
--- /dev/null
+++ b/qaapi/qa/tests/node/busyloop.py
@@ -0,0 +1,14 @@
+#!/usr/bin/python
+
+import sys
+import time
+
+def main(sec):                         
+    now = time.time()
+    stop = now + float(sec)
+    print 'about to busy-wait for', sec, 'seconds'
+    while now < stop:
+        now = time.time()
+
+if __name__ == "__main__":
+    main(sys.argv[1])

diff --git a/qaapi/qa/tests/node/node_cpu_sched.py b/qaapi/qa/tests/node/node_cpu_sched.py
new file mode 100755 (executable)
index 0000000..2dc16ef
--- /dev/null
+++ b/qaapi/qa/tests/node/node_cpu_sched.py
@@ -0,0 +1,157 @@
+#!/usr/bin/python
+
+"""
+This file contains a stand-alone CPU scheduler test to be run on a node.
+It still needs to be integrated into the testing framework.
+"""
+
+import sys
+import commands
+import re
+from threading import Thread
+
+class CpuTest:
+    def __init__(self, ctx, cpu, resv, share, min, max, desc):
+        self.ctx = ctx
+        self.cpu = cpu
+        self.resv = resv
+        self.share = share
+        self.min = min
+        self.max = max
+        self.desc = desc
+
+
+class Spinner(Thread):
+    def __init__(self, test):
+        Thread.__init__(self)
+
+        cmd = '/usr/bin/time ./busyloop.py 15'
+        if test.share:
+            flags = '--idle-time'
+        else:
+            flags = ''
+
+        self.cmd = 'taskset -c ' + str(test.cpu) \
+            + ' vcontext --create --xid ' + str(test.ctx) + ' --disconnect ' \
+            + '-- vsched --fill-rate ' + str(test.resv) + ' --interval 100 ' \
+            + '--fill-rate2 ' + str(test.share) + ' --interval2 1000 ' \
+            + '--tokens 100 --tokens-min 50 --tokens-max 100 --force '\
+            + flags + ' -- vattribute --flag sched_hard -- ' + cmd
+
+        self.test = test
+
+    def run(self):
+        self.output = commands.getoutput( self.cmd )
+
+    def passed(self):
+        match = re.search('elapsed (\d+)%CPU', self.output)
+        if match:
+            self.pct = int(match.group(1))
+        else:
+            print "Error parsing output: cannot get CPU%"
+            self.pct = 0
+
+        return ( self.pct >= self.test.min and self.pct <= self.test.max )
+
+
+def run_test(testlist):
+    failures = 0
+
+    for test in testlist:
+        test.thread = Spinner(test)
+        test.thread.start()
+
+    for test in testlist:
+        test.thread.join()
+
+    for test in testlist:
+        if test.thread.passed():
+            print "[PASSED] (" + str(test.thread.pct) + "%)\t", test.desc
+        else:
+            print "[FAILED] (" + str(test.thread.pct) + "%)\t", test.desc
+            failures += 1
+
+    return failures
+
+### Test 1: test hard rate-limiting to 25% (should get <= 25%)
+def test_1():
+    test = CpuTest(ctx=600, cpu=0, resv=25, share=0, min=15, max=25, 
+                   desc = "Test 1: single ctx, 25% resv")
+
+    return run_test([ test ])
+
+### Test 2: test share scheduling of a single task (should get 100%)
+def test_2():
+    test = CpuTest(ctx=600, cpu=0, resv=0, share=1, min=95, max=100, 
+                   desc = "Test 2: single ctx, one share")
+
+    return run_test([ test ])
+
+### Test 3: test hard & share scheduling of a single task (should get 100%)
+def test_3():
+    test = CpuTest(ctx=600, cpu=0, resv=25, share=1, min=95, max=100, 
+                   desc = "Test 3: single ctx, 25% resv, one share")
+
+    return run_test([ test ])
+
+### Test 4: test relative share scheduling of several tasks
+def test_4():
+    test1 = CpuTest(ctx=600, cpu=0, resv=0, share=1, min=15, max=17, 
+                   desc = "Test 4: ctx 1, one share")
+    test2 = CpuTest(ctx=601, cpu=0, resv=0, share=2, min=32, max=34, 
+                   desc = "        ctx 2, two shares")
+    test3 = CpuTest(ctx=602, cpu=0, resv=0, share=3, min=49, max=51, 
+                   desc = "        ctx 3, three shares")
+
+    return run_test([ test1, test2, test3 ])
+
+### Test 5: test hard rate limit and shares
+def test_5():
+    test1 = CpuTest(ctx=600, cpu=0, resv=0, share=1, min=14, max=16, 
+                   desc = "Test 5: ctx 1, one share")
+    test2 = CpuTest(ctx=601, cpu=0, resv=0, share=2, min=29, max=33, 
+                   desc = "        ctx 2, two shares")
+    test3 = CpuTest(ctx=602, cpu=0, resv=0, share=3, min=44, max=47, 
+                   desc = "        ctx 3, three shares")
+    test4 = CpuTest(ctx=603, cpu=0, resv=10, share=0, min=7, max=10, 
+                   desc = "        ctx 4, 10% resv")
+
+    return run_test([ test1, test2, test3, test4 ])
+
+### Test 6: test guarantee and shares
+def test_6():
+    test1 = CpuTest(ctx=600, cpu=0, resv=0, share=1, min=9, max=11, 
+                   desc = "Test 6: ctx 1, one share")
+    test2 = CpuTest(ctx=601, cpu=0, resv=0, share=2, min=19, max=21, 
+                   desc = "        ctx 2, two shares")
+    test3 = CpuTest(ctx=602, cpu=0, resv=0, share=3, min=29, max=31, 
+                   desc = "        ctx 3, three shares")
+    test4 = CpuTest(ctx=603, cpu=0, resv=30, share=1, min=35, max=40, 
+                   desc = "        ctx 4, 30% resv, one share")
+
+    return run_test([ test1, test2, test3, test4 ])
+
+### Test 7: SMP active (both tasks should get 100% on an SMP)
+def test_7():
+    test1 = CpuTest(ctx=600, cpu=0, resv=0, share=1, min=95, max=100, 
+                   desc = "Test 7: ctx 1, processor 1")
+    test2 = CpuTest(ctx=601, cpu=1, resv=0, share=1, min=95, max=100, 
+                   desc = "        ctx 2, processor 2")
+
+    return run_test([ test1, test2 ])
+
+def main():
+    failures = 0
+
+    failures += test_1()
+    failures += test_2()
+    failures += test_3()
+    failures += test_4()
+    failures += test_5()
+    failures += test_6()
+    failures += test_7()
+
+    return failures
+
+if __name__ == "__main__":
+    sys.exit(main())