def test_open_git_status__fail(self, pipe_open_cls_mock, sys_mock): popen_mock = Mock() popen_mock.returncode = 1 popen_mock.communicate.return_value = (" ", None) pipe_open_cls_mock.return_value = popen_mock open_git_status() self.assertTrue(sys_mock.exit.called)
def test_open_git_status(self, pipe_open_cls_mock): status = "## branch\n M zsh/test.py\n" popen_mock = Mock() popen_mock.returncode = 0 popen_mock.communicate.return_value = (status, None) pipe_open_cls_mock.return_value = popen_mock results = open_git_status() self.assertListEqual( [u"## branch", u" M zsh/test.py"], results.splitlines(), )