import os
import sys
import shutil
import hashlib
import multiprocessing
from logging import FileHandler
from unittest import TestCase

import config
from hashtree import HashTree
from client import Client, Connection as ClientConnection
from server import Miniserver
from . import RedirectedOutput


config.logger.removeHandler(config.handler)
handler=FileHandler("/tmp/morevna.log")
handler.setFormatter(config.formatter)
config.logger.addHandler(handler)

config.batchSize=8

dataDir=os.path.join(config.directory,"src/tests/data")
filename=os.path.join(dataDir,"test.img")


def compareFiles(f1,f2):
	with open(f1,mode="rb") as f:
		h2=hashlib.sha256(f.read()).hexdigest()
	with open(f2,mode="rb") as f:
		h=hashlib.sha256(f.read()).hexdigest()
	return (h,h2)


class TestMorevna(RedirectedOutput,TestCase):
	_stdout=None

	def setUp(self):
		src=os.path.join(dataDir,"test1.img")
		shutil.copyfile(src,filename)

	@classmethod
	def tearDownClass(cls):
		super().tearDownClass()
		os.remove(filename)

	def test_build(self):
		treeFile=os.path.join(dataDir,"test.bin")
		refFile=os.path.join(dataDir,"test1.bin")

		tree=HashTree.fromFile(os.path.join(dataDir,"test1.img"))
		tree.save(treeFile)

		self.assertEqual(*compareFiles(refFile,treeFile))

		os.remove(treeFile)

	def test_push(self):
		ms=Miniserver(filename)
		p=multiprocessing.Process(target=ms.serve)
		p.start()

		for clientFile in ("test2.img","test3.img","test4.img"):
			clientFile=os.path.join(dataDir,clientFile)
			c=Client(clientFile)
			with ClientConnection("127.0.0.1",config.port) as con:
				c.setConnection(con)
				blocksToTransfer=c.negotiate()
				c.sendData(blocksToTransfer)

			self.assertEqual(*compareFiles(clientFile,filename))

		p.terminate()
		p.join()

	def test_pull(self):
		serverFile=os.path.join(dataDir,"test3.img")
		ms=Miniserver(serverFile)
		p=multiprocessing.Process(target=ms.serve)
		p.start()

		c=Client(filename)
		with ClientConnection("127.0.0.1",config.port) as con:
			c.setConnection(con)
			blocksToTransfer=c.negotiate()
			c.pullData(blocksToTransfer)

		self.assertEqual(*compareFiles(serverFile,filename))

		p.terminate()
		p.join()