def test_warn_denied_variable(mocker): lookup = aws_ssm.LookupModule() lookup._load_name = "aws_ssm" boto3_double = mocker.MagicMock() boto3_double.Session.return_value.client.return_value.get_parameters.side_effect = ClientError(error_response, operation_name) with pytest.raises(AnsibleError): with mocker.patch.object(boto3, 'session', boto3_double): lookup.run(["denied_variable"], {}, **dummy_credentials)
def test_warn_missing_variable(mocker): lookup = aws_ssm.LookupModule() lookup._load_name = "aws_ssm" boto3_double = mocker.MagicMock() boto3_double.Session.return_value.client.return_value.get_parameters.return_value = missing_variable_fail_response with pytest.raises(AnsibleError): with mocker.patch.object(boto3, 'session', boto3_double): lookup.run(["missing_variable"], {}, **dummy_credentials)
def test_lookup_variable(mocker): lookup = aws_ssm.LookupModule() lookup._load_name = "aws_ssm" boto3_double = mocker.MagicMock() boto3_double.Session.return_value.client.return_value.get_parameters.return_value = simple_variable_success_response boto3_client_double = boto3_double.Session.return_value.client with mocker.patch.object(boto3, 'session', boto3_double): retval = lookup.run(["simple_variable"], {}, **dummy_credentials) assert(retval[0] == "simplevalue") boto3_client_double.assert_called_with('ssm', 'eu-west-1', aws_access_key_id='notakey', aws_secret_access_key="notasecret", aws_session_token=None)
def test_return_none_for_missing_variable(mocker): """ during jinja2 templates, we can't shouldn't normally raise exceptions since this blocks the ability to use defaults. for this reason we return ```None``` for missing variables """ lookup = aws_ssm.LookupModule() lookup._load_name = "aws_ssm" boto3_double = mocker.MagicMock() boto3_double.Session.return_value.client.return_value.get_parameters.return_value = missing_variable_response with mocker.patch.object(boto3, 'session', boto3_double): retval = lookup.run(["missing_variable"], {}, **dummy_credentials) assert (retval[0] is None)
def test_match_retvals_to_call_params_even_with_some_missing_variables(mocker): """ If we get a complex list of variables with some missing and some not, we still have to return a list which matches with the original variable list. """ lookup = aws_ssm.LookupModule() lookup._load_name = "aws_ssm" boto3_double = mocker.MagicMock() boto3_double.Session.return_value.client.return_value.get_parameters.return_value = some_missing_variable_response with mocker.patch.object(boto3, 'session', boto3_double): retval = lookup.run( ["simple", "missing_variable", "/testpath/won", "simple"], {}, **dummy_credentials) assert (retval == [ "simple_value", None, "simple_value_won", "simple_value" ])
def test_path_lookup_variable(mocker): lookup = aws_ssm.LookupModule() lookup._load_name = "aws_ssm" boto3_double = mocker.MagicMock() get_path_fn = boto3_double.Session.return_value.client.return_value.get_parameters_by_path get_path_fn.return_value = path_success_response boto3_client_double = boto3_double.Session.return_value.client with mocker.patch.object(boto3, 'session', boto3_double): args = copy(dummy_credentials) args["bypath"] = 'true' retval = lookup.run(["/testpath"], {}, **args) assert(retval[0]["/testpath/won"] == "simple_value_won") assert(retval[0]["/testpath/too"] == "simple_value_too") boto3_client_double.assert_called_with('ssm', 'eu-west-1', aws_access_key_id='notakey', aws_secret_access_key="notasecret", aws_session_token=None) get_path_fn.assert_called_with(Path="/testpath", Recursive=False, WithDecryption=True)