def test__success(self, module_patch, time_patch, import_patch):
mock_module = self.setup_module(module_patch)
mock_root = self.setup_session_root(import_patch)
lic_1 = self.get_mock_license(SECONDS_PER_DAY * 366, 1, "lic1")
lic_2 = self.get_mock_license(SECONDS_PER_DAY * 500, 1, "lic2")
self.setup_licenses(mock_root, [lic_1, lic_2])
time_patch.return_value = SECONDS_PER_DAY
main()
test_days_left_dict = {
lic_1.unique_license_identifier: [365, lic_1.licensed_feature],
lic_2.unique_license_identifier: [499, lic_2.licensed_feature]
}
test_valid_dict = {
lic_1.unique_license_identifier: True,
lic_2.unique_license_identifier: True
}
test_result_dict = {
"validity": test_valid_dict,
"days_left": test_days_left_dict
}
self.validate_session(import_patch)
mock_module.fail_json.assert_not_called()
mock_module.exit_json.assert_called_once_with(changed=False,
result=test_result_dict)
def test__cannot_import(self, module_patch, import_patch):
mock_module = self.setup_module(module_patch)
import_patch.side_effect = ImportError("cannot import")
main()
import_patch.assert_called_once_with("vspk.v5_0")
mock_module.fail_json.assert_called_once_with(
msg="vspk is required for this module, or "
"API version specified does not exist.")
def test__no_licenses(self, module_patch, time_patch, import_patch):
mock_module = self.setup_module(module_patch)
mock_root = self.setup_session_root(import_patch)
self.setup_licenses(mock_root, [])
time_patch.return_value = SECONDS_PER_DAY
main()
self.validate_session(import_patch)
mock_module.exit_json.assert_called_once_with(changed=False,
result="False")
def test__has_expired(self, module_patch, time_patch, import_patch):
mock_module = self.setup_module(module_patch)
mock_root = self.setup_session_root(import_patch)
lic_1 = self.get_mock_license(SECONDS_PER_DAY - 1, 1)
lic_2 = self.get_mock_license(SECONDS_PER_DAY * 500, 0)
self.setup_licenses(mock_root, [lic_1, lic_2])
time_patch.return_value = SECONDS_PER_DAY
main()
self.validate_session(import_patch)
mock_module.fail_json.assert_called_once_with(
msg="VSD License has expired")
def test__bad_mode(self, module_patch, time_patch, import_patch):
mock_module = self.setup_module(module_patch)
mock_root = self.setup_session_root(import_patch)
lic_1 = self.get_mock_license(SECONDS_PER_DAY * 366, 1)
lic_2 = self.get_mock_license(SECONDS_PER_DAY * 500, 0)
self.setup_licenses(mock_root, [lic_1, lic_2])
time_patch.return_value = SECONDS_PER_DAY
main()
self.validate_session(import_patch)
mock_module.exit_json.assert_called_once_with(changed=False,
result="False")
def test__will_expire(self, module_patch, time_patch, import_patch):
mock_module = self.setup_module(module_patch)
mock_root = self.setup_session_root(import_patch)
lic_1 = self.get_mock_license(SECONDS_PER_DAY * 365, 1, "lic1")
lic_2 = self.get_mock_license(SECONDS_PER_DAY * 500, 0, "lic2")
self.setup_licenses(mock_root, [lic_1, lic_2])
time_patch.return_value = SECONDS_PER_DAY
main()
self.validate_session(import_patch)
mock_module.fail_json.assert_called_once_with(
msg="VSD License will expire in 364 days")
def test__has_expired(self, module_patch, time_patch, import_patch):
mock_module = self.setup_module(module_patch)
mock_root = self.setup_session_root(import_patch)
lic_1 = self.get_mock_license(SECONDS_PER_DAY - 1, 1, "lic1")
lic_2 = self.get_mock_license(SECONDS_PER_DAY * 500, 0, "lic2")
self.setup_licenses(mock_root, [lic_1, lic_2])
time_patch.return_value = SECONDS_PER_DAY
main()
self.validate_session(import_patch)
mock_module.fail_json.assert_called_once_with(
msg="The VSD License has expired, please renew the License before proceeding to install the VSD image")
def test__no_licenses(self, module_patch, time_patch, import_patch):
mock_module = self.setup_module(module_patch)
mock_root = self.setup_session_root(import_patch)
self.setup_licenses(mock_root, [])
time_patch.return_value = SECONDS_PER_DAY
main()
test_days_left_dict = {}
test_valid_dict = {}
test_result_dict = {"validity": test_valid_dict, "days_left": test_days_left_dict}
self.validate_session(import_patch)
mock_module.exit_json.assert_called_once_with(changed=False,
result=test_result_dict)
def test__cannot_connect(self, module_patch, import_patch):
mock_module = self.setup_module(module_patch)
mock_vspk = MagicMock()
import_patch.return_value = mock_vspk
mock_session = MagicMock()
mock_root = MagicMock()
mock_session.user = mock_root
mock_vspk.NUVSDSession.return_value = mock_session
mock_session.start.side_effect = Exception("cannot connect")
main()
import_patch.assert_called_once_with("vspk.v5_0")
mock_vspk.NUVSDSession.assert_called_with(**TEST_PARAMS["vsd_auth"])
mock_session.start.assert_called_with()
mock_module.fail_json.assert_called_once_with(
msg="Could not establish connection to VSD cannot connect")
def test__cannot_connect(self, module_patch, import_patch):
mock_module = self.setup_module(module_patch)
mock_vspk = MagicMock()
import_patch.return_value = mock_vspk
mock_session = MagicMock()
mock_root = MagicMock()
mock_session.user = mock_root
mock_vspk.NUVSDSession.return_value = mock_session
mock_session.start.side_effect = Exception("cannot connect")
main()
import_patch.assert_called_once_with("vspk.{0:s}".format(
TEST_PARAMS["vsd_auth"]["api_version"]))
mock_vspk.NUVSDSession.assert_called_with(
username=TEST_PARAMS["vsd_auth"]["username"],
password=TEST_PARAMS["vsd_auth"]["password"],
enterprise=TEST_PARAMS["vsd_auth"]["enterprise"],
api_url=TEST_PARAMS["vsd_auth"]["api_url"])
mock_session.start.assert_called_with()
mock_module.fail_json.assert_called_once_with(
msg="Could not establish connection to VSD cannot connect")