Changeset 225 for test

Timestamp:

04/05/11 11:17:41 (14 years ago)

Author:

tim

Message:

added test case for python callback read/seek
began adding windows support in pyregfi
improved win32 build target

File:

• test/pyregfi-smoketest.py (modified) (10 diffs)

test/pyregfi-smoketest.py

@@ -3 +3 @@
 import sys
 import gc
+import io
 import time
 import pyregfi
+
 
 def usage():
@@ -15 +17 @@
 # Gathers various (meaningless) statistics to exercise simple attribute access
 # and to hopefully smoke out any bugs that can be identified by changing stats
-def iterTallyNames(hive):
+def iterTallyNames(hive, fh):
     key_count = 0
     key_lens = 0
@@ -60 +62 @@
 # to find the same key again, verifying it is the same.  This test is currently
 # very slow because no key caching is used.
-def iterParentWalk(hive):
+def iterParentWalk(hive, fh):
     i = 1
     for k in hive:
@@ -77 +79 @@
 # Uses the HiveIterator to walk all keys
 # Gathers various (meaningless) statistics about data/data_raw attributes
-def iterTallyData(hive):
+def iterTallyData(hive, fh):
     data_stat = 0.0
     dataraw_stat = 0.0
@@ -127 +129 @@
 # list dictionary access.  Also builds nonsensical statistics as an excuse
 # to access various base structure attributes.
-def recurseKeyTally(hive):
+def recurseKeyTally(hive, fh):
     checkValues(hive.root)
     recurseTree(hive.root, checkValues)
@@ -135 +137 @@
 
 # Iterates hive gathering stats about security and classname records
-def iterFetchRelated(hive):
+def iterFetchRelated(hive, fh):
     security_stat = 0.0
     classname_stat = 0.0
@@ -157 +159 @@
 
 
-def iterIterWalk(hive):
+def iterIterWalk(hive, fh):
     sk_stat = 0.0
     v_stat = 0.0
@@ -183 +185 @@
 
 
-
+def iterCallbackIO(hive, fh):
+    fh.seek(0)
+    new_fh = io.BytesIO(fh.read())
+    new_hive = pyregfi.Hive(new_fh)
+    for k in new_hive:
+        pass
+
+    
 if len(sys.argv) < 2:
     usage()
@@ -196 +205 @@
          ("iterIterWalk",iterIterWalk),]
 
-tests = [("iterIterWalk",iterIterWalk),]
+tests = [("iterCallbackIO",iterCallbackIO),]
 
 
@@ -211 +220 @@
         tstr = "'%s' on '%s'" % (tname,hname)
         print("##BEGIN %s:" % tstr)
-        t(hive)
+        t(hive, fh)
         print("##END %s; runtime=%f; messages:" % (tstr, time.time() - teststart))
         print(pyregfi.GetLogMessages())