def test_on_exit(self): daemon = MCRunner( config_file=self.config_file.name, pid_file=self.pid_file.name ) daemon.load_config() daemon.logger = mock.MagicMock() daemon.servers['survival'].stop = mock.MagicMock() daemon.servers['creative'].stop = mock.MagicMock() daemon.on_exit() assert daemon.servers['survival'].stop.call_count == 1 assert daemon.servers['creative'].stop.call_count == 1