#pylint: disable = C0103
""" 
basic test of KLDigitizer to see if digits come out on real spills
and if errors come out on empty
""" 
import json
import unittest
import os
import maus_cpp.converter
import MAUS

from Configuration import Configuration

class MapCppKLMCDigitizer(unittest.TestCase):  #pylint: disable = R0904
    """ basic MapCppKLMCDigitizer test to check
    if we get digits on good evts and errors on bad
    """
    @classmethod
    def setUpClass(cls):  # pylint: disable-msg=C0103
        """ Class Initializer.
            The set up is called before each test function
            is called.
        """
        cls.mapper = MAUS.MapCppKLMCDigitizer()
        conf_json = json.loads(Configuration().getConfigJSON())
        conf_json["simulation_geometry_filename"] = "Stage6.dat"
        # Test whether the configuration files were loaded correctly at birth
        cls.mapper.birth(json.dumps(conf_json))


    def test_death(self):
        """ Test to make sure death occurs """
        self.mapper.death()

    def test_process(self):
        """ Test of the process function """
        root_dir = os.environ.get("MAUS_ROOT_DIR")
        assert root_dir != None
        assert os.path.isdir(root_dir)
        _filename = \
        '%s/src/map/MapCppKLMCDigitizer/mc_test.dat' % root_dir
        assert os.path.isfile(_filename)
        _file = open(_filename, 'r')
        # File is open.
        # Spill 1 is corrupted.
        spill = "{}"
        output = self.mapper.process(spill)
        self.assertTrue("errors" in maus_cpp.converter.json_repr(output))
        # a real spill
        spill = _file.readline().rstrip()
        output = self.mapper.process(spill)
        doc = maus_cpp.converter.json_repr(output)
        self.assertTrue("kl_digits" in \
                             doc["recon_events"][0]["kl_event"])
        _file.close()

    @classmethod
    def tear_down_class(cls):
        """___"""
        cls.mapper.death()
        cls.mapper = None

if __name__ == '__main__':
    unittest.main()