def test_get_session(self, mock_get_session, monkeypatch):
"""Test get_session."""
mock_env = MagicMock()
mock_env.aws_region = "us-east-1"
monkeypatch.setattr(Context, "boto3_credentials", {})
obj = Context(deploy_environment=mock_env)
assert obj.get_session()
mock_get_session.assert_called_once_with(region=mock_env.aws_region)
def test_get_session_env_creds(self, mock_get_session, monkeypatch):
"""Test get_session with env creds."""
creds = {
"aws_access_key_id": "test-key",
"aws_secret_access_key": "test-secret",
"aws_session_token": "test-token",
}
mock_env = MagicMock()
mock_env.aws_region = "us-east-1"
monkeypatch.setattr(Context, "boto3_credentials", creds)
obj = Context(deploy_environment=mock_env)
assert obj.get_session()
mock_get_session.assert_called_once()
call_kwargs = mock_get_session.call_args.kwargs
assert call_kwargs.pop("access_key") == creds["aws_access_key_id"]
assert call_kwargs.pop("region") == mock_env.aws_region
assert call_kwargs.pop("secret_key") == creds["aws_secret_access_key"]
assert call_kwargs.pop("session_token") == creds["aws_session_token"]
assert not call_kwargs