Cleanup copyright.

Cleanup debian/rules.

Adding test.py to examples.


git-svn-id: svn://svn.autonomy.net.au/python-dmidecode@158 abc39116-655e-4be6-ad55-d661dc543056

1 files changed, 103 insertions, 0 deletions

diff --git a/examples/test.py b/examples/test.py
new file mode 100755
index 0000000..699a0bf
--- /dev/null
+++ b/examples/test.py
@@ -0,0 +1,103 @@
+#!/usr/bin/env python
+#.awk '$0 ~ /case [0-9]+: .. 3/ { sys.stdout.write($2 }' src/dmidecode.c|tr ':\n' ', '
+
+from pprint import pprint
+import os, sys, random, tempfile, time
+
+FH, DUMP = tempfile.mkstemp()
+os.unlink(DUMP)
+os.close(FH)
+
+total = 0
+success = 0
+
+def test(r):
+  global total
+  global success
+
+  total += 1
+  if r:
+    sys.stdout.write("Good\n")
+    success += 1
+    return True
+  else:
+    sys.stdout.write("FAILED\n")
+    return False
+
+total += 1
+print "-"*80
+sys.stdout.write("Importing module...")
+try:
+  import dmidecode
+  success += 1
+  sys.stdout.write("Done\n")
+  sys.stdout.write(" * Version: %s\n"%dmidecode.version)
+  sys.stdout.write(" * DMI Version String: %s\n"%dmidecode.dmi)
+
+  print "-"*80
+  sys.stdout.write("Testing that default device is /dev/mem...")
+  test(dmidecode.get_dev() == "/dev/mem")
+
+  sys.stdout.write("Testing that write-lock will not break on dump()...")
+  test(not dmidecode.dump())
+
+  sys.stdout.write("Testing ability to change device to %s..."%DUMP)
+  test(dmidecode.set_dev(DUMP))
+
+  sys.stdout.write("Testing that device has changed to %s..."%DUMP)
+  test(dmidecode.get_dev() == DUMP)
+
+  sys.stdout.write("Testing that write on new file is ok...")
+  test(dmidecode.dump())
+
+  sys.stdout.write("Testing that file was actually written...")
+  time.sleep(0.1)
+  test(os.path.exists(DUMP))
+  os.unlink(DUMP)
+
+  types = range(0, 42)+range(126, 128)
+  bad_types = [-1, -1000, 256]
+  sections = ["bios", "system", "baseboard", "chassis", "processor", "memory", "cache", "connector", "slot"]
+  devices = []
+  if os.path.exists("private"):
+    devices.extend([os.path.join("private", _) for _ in os.listdir("private")])
+    devices.remove('private/.svn')
+  else:
+    sys.stdout.write("If you have memory dumps to test, create a directory called `private' and drop them in there.\n")
+  devices.append("/dev/mem")
+  random.shuffle(types)
+  random.shuffle(devices)
+  random.shuffle(sections)
+
+  for dev in devices:
+    sys.stdout.write(" * Testing %s..."%dev); sys.stdout.flush()
+    if test(dmidecode.set_dev(dev) and dmidecode.get_dev() == dev):
+      print "-"*80
+      print sections
+      for section in sections:
+        sys.stdout.write("   * Testing %s..."%section); sys.stdout.flush()
+        output = getattr(dmidecode, section)()
+        test(output is not False)
+        if output: sys.stdout.write("     * %s\n"%output.keys())
+
+      print "-"*80
+      for i in bad_types:
+        sys.stdout.write("   * Testing bad type %i..."%i); sys.stdout.flush()
+        try:
+          output = dmidecode.type(i)
+          test(output is False)
+        except SystemError:
+          sys.stdout.write("FAILED\n")
+
+      print "-"*80
+      for i in types:
+        sys.stdout.write("   * Testing type %i..."%i); sys.stdout.flush()
+        output = dmidecode.type(i)
+        test(output is not False)
+        if output:
+          sys.stdout.write("     * %s\n"%output.keys())
+
+except ImportError:
+  sys.stdout.write("FAILED\n")
+
+sys.stdout.write("Score: %d/%d\n"%(success, total))