Performance tweaking and benchmarks.

1 files changed, 75 insertions, 0 deletions

diff --git a/tests/performance.py b/tests/performance.py
new file mode 100644
index 00000000..15b9bad8
--- /dev/null
+++ b/tests/performance.py
@@ -0,0 +1,75 @@
+# test script to evaluate Cobbler API performance
+#
+# Michael DeHaan <mdehaan@redhat.com>
+
+import os
+import cobbler.api as capi
+import time
+import sys
+import random
+
+N = 200
+print "sample size is %s" % N
+
+api = capi.BootAPI()
+
+# part one ... create our test systems for benchmarking purposes if
+# they do not seem to exist.
+
+if not api.profiles().find("foo"):
+    print "CREATE A PROFILE NAMED 'foo' to be able to run this test"
+    sys.exit(0)
+
+def random_mac():
+    mac = [ 0x00, 0x16, 0x3e,
+      random.randint(0x00, 0x7f),
+      random.randint(0x00, 0xff),
+      random.randint(0x00, 0xff) ]
+    return ':'.join(map(lambda x: "%02x" % x, mac))
+
+print "Deleting autotest entries from a previous run"
+time1 = time.time()
+for x in xrange(0,N):
+   try:
+       sys = api.systems().remove("autotest-%s" % x,with_delete=True)
+   except:
+       pass
+time2 = time.time()
+print "ELAPSED: %s seconds" % (time2 - time1)
+
+print "Creating test systems from scratch"
+time1 = time.time()
+for x in xrange(0,N):
+   sys = api.new_system()
+   sys.set_name("autotest-%s" % x)
+   sys.set_mac_address(random_mac())
+   sys.set_profile("foo") # assumes there is already a foo
+   # print "... adding: %s" % sys.name
+   api.systems().add(sys,save=True,with_sync=False,with_triggers=False)
+time2 = time.time()
+print "ELAPSED %s seconds" % (time2 - time1)
+
+for mode2 in [ "fast", "normal", "full" ]:
+   for mode in [ "on", "off" ]:
+
+       print "Running netboot edit benchmarks (turn %s, %s)" % (mode, mode2)
+       time1 = time.time()
+       for x in xrange(0,N):
+           sys = api.systems().find("autotest-%s" % x)
+           if mode == "off":
+               sys.set_netboot_enabled(0)
+           else:
+               sys.set_netboot_enabled(1)
+           # print "... editing: %s" % sys.name
+           if mode2 == "fast":
+               api.systems().add(sys, save=True, with_sync=False, with_triggers=False, quick_pxe_update=True)
+           if mode2 == "normal":
+               api.systems().add(sys, save=True, with_sync=False, with_triggers=False)
+           if mode2 == "full":
+               api.systems().add(sys, save=True, with_sync=True, with_triggers=True)
+
+       time2 = time.time()
+       print "ELAPSED: %s seconds" % (time2 - time1)
+
+
+