#!/usr/bin/env python3 import shutil, os.path, json, subprocess, hashlib, glob import unittest, logging, sys successLogo = """ +----------------------------------+ | All test cases have PASSED | +----------------------------------+ """ resDir = "src/integrationTest/resources" log = logging.getLogger('TEST') log.setLevel(logging.DEBUG) consoleHandler = logging.StreamHandler(sys.stdout) consoleHandler.setFormatter(logging.Formatter(fmt='%(asctime)s %(levelname)-8s %(name)s - %(message)s', datefmt='%Y-%m-%d %H:%M:%S')) log.addHandler(consoleHandler) def hashFile(fileName): hasher = hashlib.md5() with open(fileName, 'rb') as afile: buf = afile.read() hasher.update(buf) return hasher.hexdigest() def deleteIfExists(inFile): if os.path.isfile(inFile): os.remove(inFile) def cleanUp(): log.info("clean up ...") shutil.rmtree("build", ignore_errors = True) deleteIfExists("boot.img") deleteIfExists("boot.img.clear") deleteIfExists("boot.img.google") deleteIfExists("boot.img.signed") deleteIfExists("recovery.img") deleteIfExists("recovery.img.clear") deleteIfExists("recovery.img.google") deleteIfExists("recovery.img.signed") deleteIfExists("vbmeta.img") deleteIfExists("vbmeta.img.signed") def verifySingleJson(inResourceDir, inImageDir, jsonFile): log.info(jsonFile) resDir = inResourceDir imgDir = inImageDir verifyItems = json.load(open(jsonFile)) for k, v in verifyItems["copy"].items(): shutil.copyfile(os.path.join(resDir, imgDir, k), v) subprocess.check_call("gradle unpack", shell = True) subprocess.check_call("gradle pack", shell = True) for k, v in verifyItems["hash"].items(): log.info("%s : %s" % (k, v)) unittest.TestCase().assertEqual(hashFile(k), v) def verifySingleDir(inResourceDir, inImageDir): resDir = inResourceDir imgDir = inImageDir log.info("Enter %s ..." % os.path.join(resDir, imgDir)) jsonFiles = glob.glob(os.path.join(resDir, imgDir) + "/*.json") for jsonFile in jsonFiles: cleanUp() verifySingleJson(inResourceDir, inImageDir, jsonFile) cleanUp() def main(): # 5.0 verifySingleDir(resDir, "5.0_fugu_lrx21m") # 6.0 verifySingleDir(resDir, "6.0.0_bullhead_mda89e") # 7.0 special boot cleanUp() subprocess.check_call("dd if=%s/7.1.1_volantis_n9f27m/boot.img of=boot.img bs=256 skip=1" % resDir, shell = True) verifySingleJson(resDir, "7.1.1_volantis_n9f27m", "%s/7.1.1_volantis_n9f27m/boot.json" % resDir) # 7.0 special recovery cleanUp() subprocess.check_call("dd if=%s/7.1.1_volantis_n9f27m/recovery.img of=recovery.img bs=256 skip=1" % resDir, shell = True) verifySingleJson(resDir, "7.1.1_volantis_n9f27m", "%s/7.1.1_volantis_n9f27m/recovery.json" % resDir) # 8.0 verifySingleDir(resDir, "8.0.0_fugu_opr2.170623.027") # 9.0 + avb cleanUp() subprocess.check_call("tar xf %s/9.0.0_blueline_pq1a.181105.017.a1/boot.img.tar.gz" % resDir, shell = True) verifySingleJson(resDir, "9.0.0_blueline_pq1a.181105.017.a1", "%s/9.0.0_blueline_pq1a.181105.017.a1/boot.json" % resDir) cleanUp() verifySingleJson(resDir, "9.0.0_blueline_pq1a.181105.017.a1", "%s/9.0.0_blueline_pq1a.181105.017.a1/vbmeta.json" % resDir) # Q preview verifySingleDir(resDir, "Q_preview_blueline_qpp2.190228.023") # from volunteers verifySingleDir(resDir, "recovery_image_from_s-trace") log.info(successLogo) if __name__ == "__main__": # execute only if run as a script main()