def run_dbt_and_capture(self, *args, **kwargs): try: initial_stdout = log_manager.stdout initial_stderr = log_manager.stderr stringbuf = io.StringIO() log_manager.set_output_stream(stringbuf) res = self.run_dbt(*args, **kwargs) stdout = stringbuf.getvalue() finally: log_manager.set_output_stream(initial_stdout, initial_stderr) return res, stdout
def tearDown(self): log_manager.set_output_stream(self.initial_stdout, self.initial_stderr) super().tearDown()
def setUp(self): super().setUp() self.initial_stdout = log_manager.stdout self.initial_stderr = log_manager.stderr self.stringbuf = io.StringIO() log_manager.set_output_stream(self.stringbuf)