def test_get_iso_sha256(self): drp = models.DriverRepoPackage(self.directory) iso = drp.get_iso() fh = open(iso.get_loc()) truth_sha256 = utils.checksum_for_file(fh, 'sha256') fh.close() assert_equal(truth_sha256, iso.get_sha256())
def _get_checksum(self, checksum): fh = open(self.fileloc) try: csum = utils.checksum_for_file(fh, checksum) fh.close() # Verify csum against value in csum file if exists cfile_loc = "%s.%s" % (self.get_loc(), checksum) if os.path.isfile(cfile_loc): fh = open(cfile_loc) data = fh.read() fh.close() assert csum in data return csum except Exception, e: fh.close() raise e