def setUpClass(cls): super().setUpClass() if not "script" in cls.kwargs: raise ValueError("Proszę podać skrypt stawiający bazę danych jako argument --script") try: load_script(StringIO(cls.kwargs['script']), cls.db_name, cls.db_name) except SubprocessError as e: cls.additional_output_list.append(e.output)
def setUpClass(cls): super().setUpClass() cls.additional_output_list = [] if not "script" in cls.kwargs: raise ValueError("Proszę podać skrypt stawiający bazę danych jako argument --script") try: load_result = load_script(StringIO(cls.kwargs['script']), cls.db_name, cls.db_name) pattern = '='*30 + '\nPSQL output\n' + '='*30 + '\n' cls.additional_output_list.append(pattern + load_result.decode('utf-8') + pattern) except SubprocessError as e: pattern = '='*30 + '\nPSQL error\n' + '='*30 + '\n' cls.additional_output_list.append(pattern + e.output.decode('utf-8') + pattern)