From d7d693b51692aa7aae90f3ca8caf5a8e9c581ed9 Mon Sep 17 00:00:00 2001 From: Bert Vermeulen Date: Thu, 12 Dec 2013 13:29:37 +0100 Subject: pdtest/runtc: Add support for binary output types. --- tests/pdtest | 19 ++++++++++++++++++- 1 file changed, 18 insertions(+), 1 deletion(-) (limited to 'tests/pdtest') diff --git a/tests/pdtest b/tests/pdtest index 608cd8a..2a90f2c 100755 --- a/tests/pdtest +++ b/tests/pdtest @@ -6,6 +6,8 @@ from getopt import getopt from tempfile import mkstemp from subprocess import Popen, PIPE from difflib import Differ +from hashlib import md5 +from shutil import copy DEBUG = 0 VERBOSE = False @@ -203,7 +205,7 @@ def get_tests(testnames): return tests -def diff_files(f1, f2): +def diff_textfiles(f1, f2): t1 = open(f1).readlines() t2 = open(f2).readlines() diff = [] @@ -215,6 +217,19 @@ def diff_files(f1, f2): return diff +def compare_binfiles(f1, f2): + h1 = md5() + h1.update(open(f1, 'rb').read()) + h2 = md5() + h2.update(open(f2, 'rb').read()) + if h1.digest() == h2.digest(): + result = None + else: + result = ["Binary output does not match."] + + return result + + def run_tests(tests, fix=False): errors = 0 results = [] @@ -265,6 +280,8 @@ def run_tests(tests, fix=False): diff = diff_error = None if op['type'] == 'annotation': diff = diff_textfiles(match, outfile) + elif op['type'] == 'binary': + diff = compare_binfiles(match, outfile) else: diff = ["Unsupported output type '%s'." % op['type']] except Exception as e: -- cgit v1.2.3-70-g09d2