Stone_Red/dvc
1
0
Fork 0
mirror of https://github.com/Stone-Red-Code/dvc.git synced 2026-11-02 07:18:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

replace exception catch with signal

Browse Source
This commit is contained in:
Kirill Vasin
2019-06-20 13:39:22 +03:00
parent ccb8e338e1
commit 5457c16e76
2 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/func/test_run.py
+6 -1
View File
@@ -324,10 +324,15 @@ class TestCmdRun(TestDvc):
self.assertEqual(stage.cmd, 'echo "foo bar"')
@mock.patch.object(subprocess, "Popen", side_effect=KeyboardInterrupt)
def test_keyboard_interrupt(self, _):
def test_keyboard_interrupt_before_communicate(self, _):
ret = main(["run", "mycmd"])
self.assertEqual(ret, 252)
@mock.patch.object(subprocess.Popen, "wait", new=KeyboardInterrupt)
def test_keyboard_interrupt_during_communicate(self):
ret = main(["run", "python", "code.py"])
self.assertEqual(ret, 1)
class TestRunRemoveOuts(TestDvc):
def test(self):