def test_run_main(mock_ansible_module, mock_license, mock_time): "Test correct cl_licence file execution when restart_switchd is no" instance = mock_ansible_module.return_value instance.mock_license.return_value = True values = mod_args.copy() instance.params.get.side_effect = mod_args_generator(values) instance.run_command.return_value = ['0', '', ''] main() instance.run_command.assert_called_with( '/usr/cumulus/bin/cl-license -i http://10.1.1.1/license.txt', check_rc=True) _msg = 'license updated/installed. remember to restart switchd' instance.exit_json.assert_called_with(msg=_msg, changed=True)
def test_run_main_restart_switchd_no(mock_ansible_module, mock_license, mock_time): "Test correct cl_licence file execution when restart_switchd is no" instance = mock_ansible_module.return_value instance.mock_license.return_value = True values = mod_args.copy() values['restart_switchd'] = False instance.params.get.side_effect = mod_args_generator(values) instance.run_command.return_value = ['0', '', ''] main() instance.run_command.assert_called_with( '/usr/cumulus/bin/cl-license -i http://10.1.1.1/license.txt', check_rc=True) _msg = 'license updated/installed. no request to restart switchd' instance.exit_json.assert_called_with(msg=_msg, changed=True)
def test_run_main_restart_switchd_yes(mock_restart_switchd, mock_ansible_module, mock_license): """ Test correct cl_licence file execution when restart_switchd_is_yes\ and switchd fails to start. Module should run fail_json """ mock_license.return_value = True mock_restart_switchd.return_value = False instance = mock_ansible_module.return_value instance.params.get.side_effect = mod_args_generator(mod_args) instance.run_command.return_value = ['0', '', ''] main() instance.run_command.assert_called_with( '/usr/cumulus/bin/cl-license -i http://10.1.1.1/license.txt', check_rc=True) instance.fail_json.assert_called_with( msg='license updated/installed. switchd failed to restart')
def test_run_main_restart_switchd_yes_doesnt_fail(mock_restart_switchd, mock_ansible_module, mock_license): """ Test correct cl_licence file execution when restart_switchd_is_yes\ and switchd starts properly """ mock_license.return_value = True mock_restart_switchd.return_value = True instance = mock_ansible_module.return_value instance.params.get.side_effect = mod_args_generator(mod_args) instance.run_command.return_value = ['0', '', ''] main() instance.run_command.assert_called_with( '/usr/cumulus/bin/cl-license -i http://10.1.1.1/license.txt', check_rc=True) assert_equals(instance.fail_json.call_count, 0) instance.exit_json.assert_called_with( msg='license updated/installed. switchd restarted', changed=True)