import hashlib
import libarchive
import pathlib
import unittest

_defaulthash = 'sha512'

def _readfp(fp):
	while True:
		r = fp.read(64*1024)
		# libarchive returns None on EOF
		if r == b'' or r is None:
			return

		yield r

def _hashfp(fp):
	hash = getattr(hashlib, _defaulthash)()
	for r in _readfp(fp):
		hash.update(r)

	return '%s:%s' % (_defaulthash, hash.hexdigest())


class TestArchive(unittest.TestCase):
	def setUp(self):
		self.fixtures = pathlib.Path(__file__).parent.parent / 'fixtures'

	def test_closed(self):
		fname = self.fixtures / 'testfile.tar.gz'

		with libarchive.Archive(fname) as arch:
			origfp = arch.f

			hashes = []

			for i in arch:
				if not i.isfile():
					continue

				with arch.readstream(i.size) as fp:
					hashes.append(_hashfp(fp))

				self.assertTrue(fp.closed)
				self.assertIsNone(arch._stream)

			self.assertEqual(hashes, [ 'sha512:90f8342520f0ac57fb5a779f5d331c2fa87aa40f8799940257f9ba619940951e67143a8d746535ed0284924b2b7bc1478f095198800ba96d01847d7b56ca465c', 'sha512:7d5768d47b6bc27dc4fa7e9732cfa2de506ca262a2749cb108923e5dddffde842bbfee6cb8d692fb43aca0f12946c521cce2633887914ca1f96898478d10ad3f' ])

		self.assertTrue(arch.f.closed)