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