#!/usr/bin/env python
# -*- coding: utf-8 -*-

import logging

import boto3
from boto3.session import Session
from botocore.session import get_session

from client_grants import ClientGrantsCredentialProvider

boto3.set_stream_logger('boto3.resources', logging.DEBUG)

bc_session = get_session()
bc_session.get_component('credential_provider').insert_before(
    'env',
    ClientGrantsCredentialProvider('NZLOOFRSluw9RfIkuHGqfk1HFp4a',

        this.resourceLoc = resourceLoc;
        this.treeConnection = treeConnection.acquire();
    }

    @Override
    public SmbSessionImpl getSession() {
        return this.treeConnection.getSession();
    }

    @Override
    public void ensureDFSResolved() throws CIFSException {
        this.treeConnection.ensureDFSResolved(this.resourceLoc);
    }

    @Override

            }, validationErrorLambda);
            return null;
        });
    }

    private OptionalThing<HttpSession> getSession() {
        final HttpSession session = request.getSession(false);
        if (session != null) {
            return OptionalEntity.of(session);
        }
        return OptionalEntity.empty();
    }

    }

    @Test
    @DisplayName("getSession(): may return null and should be observed as such")
    void getSession_returnsNull() {
        // Arrange
        when(handle.getSession()).thenReturn(null);

        // Act
        SmbSession result = handle.getSession();

        // Assert
        assertNull(result);
        verify(handle).getSession();
    }

    @Test

     *
     * @return session that this file has been loaded through
     */
    @SuppressWarnings("resource")
    public SmbSessionImpl getSession() {
        final SmbTreeImpl t = getTreeInternal();
        if (t != null) {
            return t.getSession();
        }
        return null;
    }

    /**
     *
     * Only call this method while holding a tree handle
     *

        toolchainModels.add(ToolchainModel.newBuilder().type("unknown").build());
        when(session.getRequest()).thenReturn(executionRequest);
        Session sessionv4 = mock(Session.class);
        when(session.getSession()).thenReturn(sessionv4);
        when(sessionv4.getToolchains()).thenReturn(toolchainModels);

        List<Toolchain> toolchains = toolchainManager.getToolchains(session, "unknown", null);

    }

    @Test
    @DisplayName("getSession returns underlying session from tree connection")
    void getSessionReturnsUnderlying() {
        // Verifies getSession() exposes the session provided by the connection
        assertSame(session, handle.getSession());
        verify(treeConnection).getSession();
    }

    @Test

        // Setup request session
        lenient().when(request.getSession()).thenReturn(session);
        lenient().when(request.getSession(false)).thenReturn(session);
        lenient().when(request.getSession(true)).thenReturn(session);

        // Setup default session attribute behavior

        @Override
        public Toolchain getToolchainFromBuildContext(String type, MavenSession session) {
            return getDelegate()
                    .getToolchainFromBuildContext(session.getSession(), type)
                    .map(ToolchainManagerFactory.this::getToolchainV3)
                    .orElse(null);
        }

        @Override

    @Inject
    private ArtifactRepositoryFactory artifactRepositoryFactory;

    @Override
    public AuthenticationInfo getAuthenticationInfo(String id) {
        MavenSession session = legacySupport.getSession();

        if (session != null && id != null) {
            MavenExecutionRequest request = session.getRequest();

            if (request != null) {
                List<Server> servers = request.getServers();