#! /usr/bin/env python3
# -*- coding: utf-8 -*-

#	Copyright 2012-2013, Marten de Vries
#
#	This file is part of OpenTeacher.
#
#	OpenTeacher is free software: you can redistribute it and/or modify
#	it under the terms of the GNU General Public License as published by
#	the Free Software Foundation, either version 3 of the License, or
#	(at your option) any later version.
#
#	OpenTeacher is distributed in the hope that it will be useful,
#	but WITHOUT ANY WARRANTY; without even the implied warranty of
#	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#	GNU General Public License for more details.
#
#	You should have received a copy of the GNU General Public License
#	along with OpenTeacher.  If not, see <http://www.gnu.org/licenses/>.

import unittest

MODES = ("all", "buttonRegister",)

class TestCase(unittest.TestCase):
	"""Tests for the buttonRegister module"""

	def testAddButton(self):
		"""Test if the addButton event works"""

		def func(myB):
			data["myB"] = myB
		for mod in self._mm.mods("active", type="buttonRegister"):
			data = {}
			mod.addButton.handle(func)
			b = mod.registerButton("category")
			self.assertEqual(b, data["myB"])

	def testRemoveButton(self):
		"""Test if the removeButton event works"""

		def func(myB):
			data["myB"] = myB
		for mod in self._mm.mods("active", type="buttonRegister"):
			data = {}
			mod.removeButton.handle(func)
			b = mod.registerButton("category")
			mod.unregisterButton(b)
			self.assertEqual(b, data["myB"])

	def testRegisterButton(self):
		"""Test if the button interface is correct"""

		def checkEvent(item):
			self.assertIsNotNone(item.handle)
			self.assertIsNotNone(item.unhandle)
			self.assertIsNotNone(item.send)

		for mod in self._mm.mods("active", type="buttonRegister"):
			b = mod.registerButton("myCategory")
			self.assertEqual(b.category, "myCategory")
			checkEvent(b.clicked)
			checkEvent(b.changeText)
			checkEvent(b.changePriority)
			checkEvent(b.changeIcon)
			checkEvent(b.changeSize)

	def testClickingRegisteredButtons(self):
		if self.mode not in MODES:
			return
		for button in self._buttons:
			button.clicked.send()

class TestModule:
	def __init__(self, moduleManager, *args, **kwargs):
		super().__init__(*args, **kwargs)
		self._mm = moduleManager

		self.type = "test"
		self.requires = (
			self._mm.mods(type="buttonRegister"),
		)

	def _addButton(self, button):
		self.TestCase._buttons.add(button)

	def _removeButton(self, button):
		self.TestCase._buttons.remove(button)

	def enable(self):
		self.TestCase = TestCase
		self.TestCase._mm = self._mm
		self.TestCase._buttons = set()

		for mod in self._mm.mods(type="buttonRegister"):
			mod.addButton.handle(self._addButton)
			mod.removeButton.handle(self._removeButton)

		self.active = True

	def disable(self):
		self.active = False

		del self.TestCase
		for mod in self._mm.mods(type="buttonRegister"):
			mod.addButton.unhandle(self._addButton)
			mod.removeButton.unhandle(self._removeButton)

def init(moduleManager):
	return TestModule(moduleManager)
