本文整理汇总了Python中clc_ansible_module.clc_alert_policy.ClcAlertPolicy.do_work方法的典型用法代码示例。如果您正苦于以下问题：Python ClcAlertPolicy.do_work方法的具体用法？Python ClcAlertPolicy.do_work怎么用？Python ClcAlertPolicy.do_work使用的例子？那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类clc_ansible_module.clc_alert_policy.ClcAlertPolicy的用法示例。
在下文中一共展示了ClcAlertPolicy.do_work方法的2个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: TestClcAlertPolicy
# 需要导入模块: from clc_ansible_module.clc_alert_policy import ClcAlertPolicy [as 别名]
# 或者: from clc_ansible_module.clc_alert_policy.ClcAlertPolicy import do_work [as 别名]
class TestClcAlertPolicy(unittest.TestCase):
def setUp(self):
self.clc = mock.MagicMock()
self.module = FakeAnsibleModule()
self.policy = ClcAlertPolicy(self.module)
self.policy.module.exit_json = mock.MagicMock()
def notestNoCreds(self):
self.policy.module.fail_json = mock.MagicMock(side_effect=Exception('nocreds'))
try:
result = self.policy.do_work()
except:
pass
self.assertEqual(self.policy.module.fail_json.called, True)
def testLoginMagic(self):
self.policy.clc.v2.SetCredentials = mock.MagicMock()
with patch.dict('os.environ', {'CLC_V2_API_USERNAME':'passWORD', 'CLC_V2_API_PASSWD':'UsErnaME'}):
try:
self.policy.process_request()
except:
# It'll die, and we don't care
pass
self.policy.clc.v2.SetCredentials.assert_called_once_with(api_username='passWORD',api_passwd='UsErnaME')
@patch.object(clc_alert_policy, 'clc_sdk')
def test_set_user_agent(self, mock_clc_sdk):
clc_alert_policy.__version__ = "1"
ClcAlertPolicy._set_user_agent(mock_clc_sdk)
self.assertTrue(mock_clc_sdk.SetRequestsSession.called)
@patch.object(ClcAlertPolicy, 'clc')
def test_set_clc_credentials_from_env(self, mock_clc_sdk):
with patch.dict('os.environ', {'CLC_V2_API_TOKEN': 'dummyToken',
'CLC_ACCT_ALIAS': 'TEST'}):
under_test = ClcAlertPolicy(self.module)
under_test._set_clc_credentials_from_env()
self.assertEqual(under_test.clc._LOGIN_TOKEN_V2, 'dummyToken')
self.assertFalse(mock_clc_sdk.v2.SetCredentials.called)
self.assertEqual(self.module.fail_json.called, False)
@patch.object(ClcAlertPolicy, 'clc')
def test_set_clc_credentials_w_creds(self, mock_clc_sdk):
with patch.dict('os.environ', {'CLC_V2_API_USERNAME': 'dummyuser', 'CLC_V2_API_PASSWD': 'dummypwd'}):
under_test = ClcAlertPolicy(self.module)
under_test._set_clc_credentials_from_env()
mock_clc_sdk.v2.SetCredentials.assert_called_once_with(api_username='dummyuser', api_passwd='dummypwd')
@patch.object(ClcAlertPolicy, 'clc')
def test_set_clc_credentials_w_api_url(self, mock_clc_sdk):
with patch.dict('os.environ', {'CLC_V2_API_URL': 'dummyapiurl'}):
under_test = ClcAlertPolicy(self.module)
under_test._set_clc_credentials_from_env()
self.assertEqual(under_test.clc.defaults.ENDPOINT_URL_V2, 'dummyapiurl')
def test_set_clc_credentials_w_no_creds(self):
with patch.dict('os.environ', {}, clear=True):
under_test = ClcAlertPolicy(self.module)
under_test._set_clc_credentials_from_env()
self.assertEqual(self.module.fail_json.called, True)
def test_clc_module_not_found(self):
# Setup Mock Import Function
import __builtin__ as builtins
real_import = builtins.__import__
def mock_import(name, *args):
if name == 'clc': raise ImportError
return real_import(name, *args)
# Under Test
with mock.patch('__builtin__.__import__', side_effect=mock_import):
reload(clc_alert_policy)
ClcAlertPolicy(self.module)
# Assert Expected Behavior
self.module.fail_json.assert_called_with(msg='clc-python-sdk required for this module')
reload(clc_alert_policy)
def test_requests_invalid_version(self):
# Setup Mock Import Function
import __builtin__ as builtins
real_import = builtins.__import__
def mock_import(name, *args):
if name == 'requests':
args[0]['requests'].__version__ = '2.4.0'
return real_import(name, *args)
# Under Test
with mock.patch('__builtin__.__import__', side_effect=mock_import):
reload(clc_alert_policy)
ClcAlertPolicy(self.module)
# Assert Expected Behavior
self.module.fail_json.assert_called_with(msg='requests library version should be >= 2.5.0')
reload(clc_alert_policy)
def test_requests_module_not_found(self):
# Setup Mock Import Function
import __builtin__ as builtins
real_import = builtins.__import__
def mock_import(name, *args):
#.........这里部分代码省略.........
示例2: TestClcAlertPolicy
# 需要导入模块: from clc_ansible_module.clc_alert_policy import ClcAlertPolicy [as 别名]
# 或者: from clc_ansible_module.clc_alert_policy.ClcAlertPolicy import do_work [as 别名]
class TestClcAlertPolicy(unittest.TestCase):
def setUp(self):
self.clc = mock.MagicMock()
self.module = FakeAnsibleModule()
self.policy = ClcAlertPolicy(self.module)
self.policy.module.exit_json = mock.MagicMock()
def notestNoCreds(self):
self.policy.module.fail_json = mock.MagicMock(side_effect=Exception("nocreds"))
try:
result = self.policy.do_work()
except:
pass
self.assertEqual(self.policy.module.fail_json.called, True)
def testLoginMagic(self):
self.policy.clc.v2.SetCredentials = mock.MagicMock()
with patch.dict("os.environ", {"CLC_V2_API_USERNAME": "passWORD", "CLC_V2_API_PASSWD": "UsErnaME"}):
try:
self.policy.process_request()
except:
# It'll die, and we don't care
pass
self.policy.clc.v2.SetCredentials.assert_called_once_with(api_username="passWORD", api_passwd="UsErnaME")
@patch.object(clc_alert_policy, "clc_sdk")
def test_set_user_agent(self, mock_clc_sdk):
clc_alert_policy.__version__ = "1"
ClcAlertPolicy._set_user_agent(mock_clc_sdk)
self.assertTrue(mock_clc_sdk.SetRequestsSession.called)
@patch.object(ClcAlertPolicy, "clc")
def test_set_clc_credentials_from_env(self, mock_clc_sdk):
with patch.dict("os.environ", {"CLC_V2_API_TOKEN": "dummyToken", "CLC_ACCT_ALIAS": "TEST"}):
under_test = ClcAlertPolicy(self.module)
under_test._set_clc_credentials_from_env()
self.assertEqual(under_test.clc._LOGIN_TOKEN_V2, "dummyToken")
self.assertFalse(mock_clc_sdk.v2.SetCredentials.called)
self.assertEqual(self.module.fail_json.called, False)
@patch.object(ClcAlertPolicy, "clc")
def test_set_clc_credentials_w_creds(self, mock_clc_sdk):
with patch.dict("os.environ", {"CLC_V2_API_USERNAME": "dummyuser", "CLC_V2_API_PASSWD": "dummypwd"}):
under_test = ClcAlertPolicy(self.module)
under_test._set_clc_credentials_from_env()
mock_clc_sdk.v2.SetCredentials.assert_called_once_with(api_username="dummyuser", api_passwd="dummypwd")
@patch.object(ClcAlertPolicy, "clc")
def test_set_clc_credentials_w_api_url(self, mock_clc_sdk):
with patch.dict("os.environ", {"CLC_V2_API_URL": "dummyapiurl"}):
under_test = ClcAlertPolicy(self.module)
under_test._set_clc_credentials_from_env()
self.assertEqual(under_test.clc.defaults.ENDPOINT_URL_V2, "dummyapiurl")
def test_set_clc_credentials_w_no_creds(self):
with patch.dict("os.environ", {}, clear=True):
under_test = ClcAlertPolicy(self.module)
under_test._set_clc_credentials_from_env()
self.assertEqual(self.module.fail_json.called, True)
def test_clc_module_not_found(self):
# Setup Mock Import Function
import __builtin__ as builtins
real_import = builtins.__import__
def mock_import(name, *args):
if name == "clc":
raise ImportError
return real_import(name, *args)
# Under Test
with mock.patch("__builtin__.__import__", side_effect=mock_import):
reload(clc_alert_policy)
ClcAlertPolicy(self.module)
# Assert Expected Behavior
self.module.fail_json.assert_called_with(msg="clc-python-sdk required for this module")
reload(clc_alert_policy)
def test_requests_invalid_version(self):
# Setup Mock Import Function
import __builtin__ as builtins
real_import = builtins.__import__
def mock_import(name, *args):
if name == "requests":
args[0]["requests"].__version__ = "2.4.0"
return real_import(name, *args)
# Under Test
with mock.patch("__builtin__.__import__", side_effect=mock_import):
reload(clc_alert_policy)
ClcAlertPolicy(self.module)
# Assert Expected Behavior
self.module.fail_json.assert_called_with(msg="requests library version should be >= 2.5.0")
reload(clc_alert_policy)
#.........这里部分代码省略.........