def test_check_pip_is_installed_failure(self, mock_sys_exit):
if six.PY3:
module_name = 'builtins.__import__'
else:
module_name = '__builtin__.__import__'
with mock.patch(module_name) as mock_import:
mock_import.side_effect = ImportError('not found')
self.assertEqual(mock_sys_exit.call_count, 0)
check_pip_is_installed()
self.assertEqual(mock_sys_exit.call_count, 1)
self.assertEqual(mock_sys_exit.call_args_list[0][0], (1, ))
def test_check_pip_is_installed_success(self):
self.assertTrue(check_pip_is_installed())