浏览代码

Tweak to test failing on some machines.

Dan Helfman 4 年之前
父节点
当前提交
ad7198ba66
共有 1 个文件被更改,包括 6 次插入1 次删除
  1. 6 1
      tests/integration/test_execute.py

+ 6 - 1
tests/integration/test_execute.py

@@ -126,10 +126,15 @@ def test_log_outputs_kills_other_processes_when_one_errors():
 def test_log_outputs_truncates_long_error_output():
     flexmock(module).ERROR_OUTPUT_MAX_LINE_COUNT = 0
     flexmock(module.logger).should_receive('log')
-    flexmock(module).should_receive('exit_code_indicates_error').and_return(True)
     flexmock(module).should_receive('command_for_process').and_return('grep')
 
     process = subprocess.Popen(['grep'], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
+    flexmock(module).should_receive('exit_code_indicates_error').with_args(
+        process, None, 'borg'
+    ).and_return(False)
+    flexmock(module).should_receive('exit_code_indicates_error').with_args(
+        process, 2, 'borg'
+    ).and_return(True)
     flexmock(module).should_receive('output_buffer_for_process').and_return(process.stdout)
 
     with pytest.raises(subprocess.CalledProcessError) as error: