import os, sys, mmap, random, string from regression import Regression loader = os.environ['PAL_LOADER'] def prepare_files(args): global file_exist file_exist = open('File', 'rb').read() if os.path.exists("file_nonexist.tmp"): os.remove("file_nonexist.tmp") with open("file_delete.tmp", "wb") as f: f.write(file_exist) # Running File regression = Regression(loader, "File", prepare_files) regression.add_check(name="Basic File Opening", check=lambda res: "File Open Test 1 OK" in res[0].log and "File Open Test 2 OK" in res[0].log and "File Open Test 3 OK" in res[0].log) regression.add_check(name="Basic File Creation", check=lambda res: "File Creation Test 1 OK" in res[0].log and "File Creation Test 2 OK" in res[0].log and "File Creation Test 3 OK" in res[0].log) regression.add_check(name="File Reading", check=lambda res: ("Read Test 1 (0th - 40th): " + file_exist[0:40].hex()) in res[0].log and ("Read Test 2 (0th - 40th): " + file_exist[0:40].hex()) in res[0].log and ("Read Test 3 (200th - 240th): " + file_exist[200:240].hex()) in res[0].log) def check_write(res): global file_exist with open("file_nonexist.tmp", "rb") as f: file_nonexist = f.read() return file_exist[0:40] == file_nonexist[200:240] and \ file_exist[200:240] == file_nonexist[0:40] regression.add_check(name="File Writing", check=check_write) regression.add_check(name="File Attribute Query", check=lambda res: ("Query: type = 1, size = %d" % (len(file_exist))) in res[0].log) regression.add_check(name="File Attribute Query by Handle", check=lambda res: ("Query by Handle: type = 1, size = %d" % (len(file_exist))) in res[0].log) regression.add_check(name="File Mapping", check=lambda res: ("Map Test 1 (0th - 40th): " + file_exist[0:40].hex()) in res[0].log and ("Map Test 2 (200th - 240th): " + file_exist[200:240].hex()) in res[0].log and ("Map Test 3 (4096th - 4136th): " + file_exist[4096:4136].hex()) in res[0].log and ("Map Test 4 (4296th - 4336th): " + file_exist[4296:4336].hex()) in res[0].log) regression.add_check(name="Set File Length", check=lambda res: os.stat("file_nonexist.tmp").st_size == mmap.ALLOCATIONGRANULARITY) regression.add_check(name="File Deletion", check=lambda res: not os.path.exists("file_delete.tmp")) rv = regression.run_checks() if rv: sys.exit(rv)