def test_freeze_command(self): eggs = [ 'M2Crypto==0.21.1', '-e [email protected]:s0undt3ch/salt-testing.git@9ed81aa2f918d59d3706e56b18f0782d1ea43bf8#egg=SaltTesting-dev', 'bbfreeze==1.1.0', 'bbfreeze-loader==1.1.0', 'pycrypto==2.6' ] mock = MagicMock( return_value={ 'retcode': 0, 'stdout': '\n'.join(eggs) } ) with patch.dict(pip.__salt__, {'cmd.run_all': mock}): with patch('salt.modules.pip.version', MagicMock(return_value='6.1.1')): ret = pip.freeze() mock.assert_called_once_with( ['pip', 'freeze'], cwd=None, runas=None, use_vt=False, python_shell=False, ) self.assertEqual(ret, eggs) mock = MagicMock( return_value={ 'retcode': 0, 'stdout': '\n'.join(eggs) } ) # Passing env_vars passes them to underlying command? with patch.dict(pip.__salt__, {'cmd.run_all': mock}): with patch('salt.modules.pip.version', MagicMock(return_value='6.1.1')): ret = pip.freeze(env_vars={"foo": "bar"}) mock.assert_called_once_with( ['pip', 'freeze'], cwd=None, runas=None, use_vt=False, python_shell=False, env={"foo": "bar"} ) self.assertEqual(ret, eggs) # Non zero returncode raises exception? mock = MagicMock(return_value={'retcode': 1, 'stderr': 'CABOOOOMMM!'}) with patch.dict(pip.__salt__, {'cmd.run_all': mock}): with patch('salt.modules.pip.version', MagicMock(return_value='6.1.1')): self.assertRaises( CommandExecutionError, pip.freeze, )
def test_freeze_deprecated_runas_triggers_warning(self): # We *always* want *all* warnings thrown on this module warnings.resetwarnings() warnings.filterwarnings('always', '', DeprecationWarning, __name__) mock = MagicMock(return_value={'retcode': 0, 'stdout': ''}) with patch.dict(pip.__salt__, {'cmd.run_all': mock}): with warnings.catch_warnings(record=True) as w: pip.freeze('/tmp/pip-env', runas='me!') self.assertEqual( 'The \'runas\' argument to pip.install is deprecated, and ' 'will be removed in Salt Lithium (Unreleased). Please ' 'use \'user\' instead.', str(w[-1].message))
def test_freeze_deprecated_runas_triggers_warning(self): # We *always* want *all* warnings thrown on this module warnings.resetwarnings() warnings.filterwarnings('always', '', DeprecationWarning, __name__) mock = MagicMock(return_value={'retcode': 0, 'stdout': ''}) with patch.dict(pip.__salt__, {'cmd.run_all': mock}): with warnings.catch_warnings(record=True) as w: pip.freeze('/tmp/pip-env', runas='me!') self.assertEqual( 'The \'runas\' argument to pip.install is deprecated, and ' 'will be removed in 0.18.0. Please use \'user\' instead.', str(w[-1].message) )
def test_freeze_deprecated_runas_triggers_warning(self): # We *always* want *all* warnings thrown on this module warnings.resetwarnings() warnings.filterwarnings("always", "", DeprecationWarning, __name__) mock = MagicMock(return_value={"retcode": 0, "stdout": ""}) with patch.dict(pip.__salt__, {"cmd.run_all": mock}): with warnings.catch_warnings(record=True) as w: pip.freeze("/tmp/pip-env", runas="me!") self.assertEqual( "The 'runas' argument to pip.install is deprecated, and " "will be removed in 0.18.0. Please use 'user' instead.", str(w[-1].message), )
def test_freeze_command(self): eggs = [ 'M2Crypto==0.21.1', '-e [email protected]:s0undt3ch/salt-testing.git@9ed81aa2f918d59d3706e56b18f0782d1ea43bf8#egg=SaltTesting-dev', 'bbfreeze==1.1.0', 'bbfreeze-loader==1.1.0', 'pycrypto==2.6' ] mock = MagicMock( return_value={ 'retcode': 0, 'stdout': '\n'.join(eggs) } ) with patch.dict(pip.__salt__, {'cmd.run_all': mock}): ret = pip.freeze() mock.assert_called_once_with( 'pip freeze', runas=None, cwd=None ) self.assertEqual(ret, eggs) # Non zero returncode raises exception? mock = MagicMock(return_value={'retcode': 1, 'stderr': 'CABOOOOMMM!'}) with patch.dict(pip.__salt__, {'cmd.run_all': mock}): self.assertRaises( CommandExecutionError, pip.freeze, )
def test_freeze_command_with_all(self): eggs = [ 'M2Crypto==0.21.1', '-e [email protected]:s0undt3ch/salt-testing.git@9ed81aa2f918d59d3706e56b18f0782d1ea43bf8#egg=SaltTesting-dev', 'bbfreeze==1.1.0', 'bbfreeze-loader==1.1.0', 'pip==0.9.1', 'pycrypto==2.6', 'setuptools==20.10.1' ] mock = MagicMock(return_value={ 'retcode': 0, 'stdout': '\n'.join(eggs) }) with patch.dict(pip.__salt__, {'cmd.run_all': mock}): with patch('salt.modules.pip.version', MagicMock(return_value='9.0.1')): ret = pip.freeze() expected = [sys.executable, '-m', 'pip', 'freeze', '--all'] mock.assert_called_with( expected, cwd=None, runas=None, use_vt=False, python_shell=False, ) self.assertEqual(ret, eggs) # Non zero returncode raises exception? mock = MagicMock(return_value={'retcode': 1, 'stderr': 'CABOOOOMMM!'}) with patch.dict(pip.__salt__, {'cmd.run_all': mock}): with patch('salt.modules.pip.version', MagicMock(return_value='9.0.1')): self.assertRaises( CommandExecutionError, pip.freeze, )
def test_freeze_command(self): eggs = [ 'M2Crypto==0.21.1', '-e [email protected]:s0undt3ch/salt-testing.git@9ed81aa2f918d59d3706e56b18f0782d1ea43bf8#egg=SaltTesting-dev', 'bbfreeze==1.1.0', 'bbfreeze-loader==1.1.0', 'pycrypto==2.6' ] mock = MagicMock(return_value={ 'retcode': 0, 'stdout': '\n'.join(eggs) }) with patch.dict(pip.__salt__, {'cmd.run_all': mock}): ret = pip.freeze() mock.assert_called_once_with( 'pip freeze', runas=None, cwd=None, python_shell=False, ) self.assertEqual(ret, eggs) # Non zero returncode raises exception? mock = MagicMock(return_value={'retcode': 1, 'stderr': 'CABOOOOMMM!'}) with patch.dict(pip.__salt__, {'cmd.run_all': mock}): self.assertRaises( CommandExecutionError, pip.freeze, )
def test_freeze_command(self): eggs = [ "M2Crypto==0.21.1", "-e [email protected]:s0undt3ch/salt-testing.git@9ed81aa2f918d59d3706e56b18f0782d1ea43bf8#egg=SaltTesting-dev", "bbfreeze==1.1.0", "bbfreeze-loader==1.1.0", "pycrypto==2.6", ] mock = MagicMock(return_value={"retcode": 0, "stdout": "\n".join(eggs)}) with patch.dict(pip.__salt__, {"cmd.run_all": mock}): ret = pip.freeze() mock.assert_called_once_with("pip freeze", runas=None, cwd=None) self.assertEqual(ret, eggs) # Non zero returncode raises exception? mock = MagicMock(return_value={"retcode": 1, "stderr": "CABOOOOMMM!"}) with patch.dict(pip.__salt__, {"cmd.run_all": mock}): self.assertRaises(CommandExecutionError, pip.freeze)