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

#	Copyright 2011-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
import sys

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

		self.type = "testRunner"
		self.priorities = {
			"test-suite": 0,
			"default": -1,
		}
		self.requires = (
			self._mm.mods(type="execute"),
		)
		self.uses = (
			self._mm.mods(type="test"),
			self._mm.mods(type="testSuite"),
		)

	def enable(self):
		self._modules = set(self._mm.mods(type="modules")).pop()
		self._execute = self._modules.default(type="execute")
		self._execute.startRunning.handle(self._run)

		self.active = True

	def disable(self):
		self.active = False

		self._execute.startRunning.unhandle(self._run)
		del self._execute
		del self._modules

	def _run(self):
		try:
			mode = sys.argv[1]
		except (IndexError, KeyError):
			print("This command needs one command line argument, 'mode', which determines which tests are run. Not sure what to use? Use 'all'.", file=sys.stderr)
			return

		testSuite = unittest.TestSuite()
		for module in self._mm.mods("active", type="test"):
			#set the test mode
			module.TestCase.mode = mode
			newTests = unittest.defaultTestLoader.loadTestsFromTestCase(module.TestCase)
			testSuite.addTests(newTests)
		for module in self._mm.mods("active", type="testSuite"):
			testSuite.addTests(module.testSuite)
		runner = unittest.TextTestRunner()
		exitCode = not runner.run(testSuite).wasSuccessful()
		sys.exit(exitCode)

def init(moduleManager):
	return TestRunnerModule(moduleManager)
