"""Tests of session module"""

from unittest import mock

import pytest

from fiona.session import (
    DummySession,
    AWSSession,
    Session,
    OSSSession,
    GSSession,
    SwiftSession,
    AzureSession,
)


def test_base_session_hascreds_notimpl():
    """Session.hascreds must be overridden"""
    assert Session.hascreds({}) is NotImplemented


def test_base_session_get_credential_options_notimpl():
    """Session.get_credential_options must be overridden"""
    assert Session().get_credential_options() is NotImplemented


def test_dummy_session():
    """DummySession works"""
    sesh = DummySession()
    assert sesh._session is None
    assert sesh.get_credential_options() == {}


def test_aws_session_class():
    """AWSSession works"""
    sesh = AWSSession(aws_access_key_id="foo", aws_secret_access_key="bar")
    assert sesh._session
    assert sesh.get_credential_options()["AWS_ACCESS_KEY_ID"] == "foo"
    assert sesh.get_credential_options()["AWS_SECRET_ACCESS_KEY"] == "bar"


def test_aws_session_class_session():
    """AWSSession works"""
    boto3 = pytest.importorskip("boto3")
    sesh = AWSSession(
        session=boto3.session.Session(
            aws_access_key_id="foo", aws_secret_access_key="bar"
        )
    )
    assert sesh._session
    assert sesh.get_credential_options()["AWS_ACCESS_KEY_ID"] == "foo"
    assert sesh.get_credential_options()["AWS_SECRET_ACCESS_KEY"] == "bar"


def test_aws_session_class_unsigned():
    """AWSSession works"""
    pytest.importorskip("boto3")
    sesh = AWSSession(aws_unsigned=True, region_name="us-mountain-1")
    assert sesh.get_credential_options()["AWS_NO_SIGN_REQUEST"] == "YES"
    assert sesh.get_credential_options()["AWS_REGION"] == "us-mountain-1"


def test_aws_session_class_profile(tmpdir, monkeypatch):
    """Confirm that profile_name kwarg works."""
    pytest.importorskip("boto3")
    credentials_file = tmpdir.join("credentials")
    credentials_file.write(
        "[testing]\n"
        "aws_access_key_id = foo\n"
        "aws_secret_access_key = bar\n"
        "aws_session_token = baz"
    )
    monkeypatch.setenv("AWS_SHARED_CREDENTIALS_FILE", str(credentials_file))
    monkeypatch.setenv("AWS_SESSION_TOKEN", "ignore_me")
    sesh = AWSSession(profile_name="testing")
    assert sesh._session
    assert sesh.get_credential_options()["AWS_ACCESS_KEY_ID"] == "foo"
    assert sesh.get_credential_options()["AWS_SECRET_ACCESS_KEY"] == "bar"
    assert sesh.get_credential_options()["AWS_SESSION_TOKEN"] == "baz"
    monkeypatch.undo()


def test_aws_session_class_endpoint():
    """Confirm that endpoint_url kwarg works."""
    pytest.importorskip("boto3")
    sesh = AWSSession(endpoint_url="example.com")
    assert sesh.get_credential_options()["AWS_S3_ENDPOINT"] == "example.com"


def test_session_factory_unparsed():
    """Get a DummySession for unparsed paths"""
    sesh = Session.from_path("/vsicurl/lolwut")
    assert isinstance(sesh, DummySession)


def test_session_factory_empty():
    """Get a DummySession for no path"""
    sesh = Session.from_path("")
    assert isinstance(sesh, DummySession)


def test_session_factory_local():
    """Get a DummySession for local paths"""
    sesh = Session.from_path("file:///lolwut")
    assert isinstance(sesh, DummySession)


def test_session_factory_unknown():
    """Get a DummySession for unknown paths"""
    sesh = Session.from_path("https://fancy-cloud.com/lolwut")
    assert isinstance(sesh, DummySession)


def test_session_factory_s3():
    """Get an AWSSession for s3:// paths"""
    pytest.importorskip("boto3")
    sesh = Session.from_path("s3://lol/wut")
    assert isinstance(sesh, AWSSession)


def test_session_factory_s3_presigned_url():
    """Get a DummySession for presigned URLs"""
    sesh = Session.from_path("https://fancy-cloud.com/lolwut?X-Amz-Signature=foo")
    assert isinstance(sesh, DummySession)


def test_session_factory_s3_no_boto3(monkeypatch):
    """Get an AWSSession for s3:// paths"""
    pytest.importorskip("boto3")
    with monkeypatch.context() as mpctx:
        mpctx.setattr("fiona.session.boto3", None)
        sesh = Session.from_path("s3://lol/wut")
        assert isinstance(sesh, DummySession)


def test_session_factory_s3_kwargs():
    """Get an AWSSession for s3:// paths with keywords"""
    pytest.importorskip("boto3")
    sesh = Session.from_path(
        "s3://lol/wut", aws_access_key_id="foo", aws_secret_access_key="bar"
    )
    assert isinstance(sesh, AWSSession)
    assert sesh._session.get_credentials().access_key == "foo"
    assert sesh._session.get_credentials().secret_key == "bar"


def test_foreign_session_factory_dummy():
    sesh = Session.from_foreign_session(None)
    assert isinstance(sesh, DummySession)


def test_foreign_session_factory_s3():
    boto3 = pytest.importorskip("boto3")
    aws_session = boto3.Session(aws_access_key_id="foo", aws_secret_access_key="bar")
    sesh = Session.from_foreign_session(aws_session, cls=AWSSession)
    assert isinstance(sesh, AWSSession)
    assert sesh._session.get_credentials().access_key == "foo"
    assert sesh._session.get_credentials().secret_key == "bar"


def test_requester_pays():
    """GDAL is configured with requester pays"""
    sesh = AWSSession(
        aws_access_key_id="foo", aws_secret_access_key="bar", requester_pays=True
    )
    assert sesh._session
    assert sesh.get_credential_options()["AWS_REQUEST_PAYER"] == "requester"


def test_oss_session_class():
    """OSSSession works"""
    oss_session = OSSSession(
        oss_access_key_id="foo",
        oss_secret_access_key="bar",
        oss_endpoint="null-island-1",
    )
    assert oss_session._creds
    assert oss_session.get_credential_options()["OSS_ACCESS_KEY_ID"] == "foo"
    assert oss_session.get_credential_options()["OSS_SECRET_ACCESS_KEY"] == "bar"


def test_session_factory_oss_kwargs():
    """Get an OSSSession for oss:// paths with keywords"""
    sesh = Session.from_path(
        "oss://lol/wut", oss_access_key_id="foo", oss_secret_access_key="bar"
    )
    assert isinstance(sesh, OSSSession)
    assert sesh.get_credential_options()["OSS_ACCESS_KEY_ID"] == "foo"
    assert sesh.get_credential_options()["OSS_SECRET_ACCESS_KEY"] == "bar"


def test_google_session_ctor_no_arg():
    session = GSSession()
    assert not session._creds


def test_gs_session_class():
    """GSSession works"""
    gs_session = GSSession(google_application_credentials="foo")
    assert gs_session._creds
    assert (
        gs_session.get_credential_options()["GOOGLE_APPLICATION_CREDENTIALS"] == "foo"
    )
    assert gs_session.hascreds({"GOOGLE_APPLICATION_CREDENTIALS": "foo"})


def test_swift_session_class():
    """SwiftSession works"""
    swift_session = SwiftSession(
        swift_storage_url="foo",
        swift_auth_token="bar",
    )
    assert swift_session._creds
    assert swift_session.get_credential_options()["SWIFT_STORAGE_URL"] == "foo"
    assert swift_session.get_credential_options()["SWIFT_AUTH_TOKEN"] == "bar"


def test_swift_session_by_user_key():
    def mock_init(
        self,
        session=None,
        swift_storage_url=None,
        swift_auth_token=None,
        swift_auth_v1_url=None,
        swift_user=None,
        swift_key=None,
    ):
        self._creds = {"SWIFT_STORAGE_URL": "foo", "SWIFT_AUTH_TOKEN": "bar"}

    with mock.patch("fiona.session.SwiftSession.__init__", new=mock_init):
        swift_session = SwiftSession(
            swift_auth_v1_url="foo", swift_user="bar", swift_key="key"
        )
        assert swift_session._creds
        assert swift_session.get_credential_options()["SWIFT_STORAGE_URL"] == "foo"
        assert swift_session.get_credential_options()["SWIFT_AUTH_TOKEN"] == "bar"


def test_session_factory_swift_kwargs():
    """Get an SwiftSession for /vsiswift/bucket/key with keywords"""
    sesh = Session.from_path(
        "/vsiswift/lol/wut", swift_storage_url="foo", swift_auth_token="bar"
    )
    assert isinstance(sesh, DummySession)


def test_session_aws_or_dummy_aws():
    """Get an AWSSession when boto3 is available"""
    boto3 = pytest.importorskip("boto3")
    assert isinstance(Session.aws_or_dummy(), AWSSession)


def test_session_aws_or_dummy_dummy(monkeypatch):
    """Get a DummySession when boto3 is not available"""
    boto3 = pytest.importorskip("boto3")
    with monkeypatch.context() as mpctx:
        mpctx.setattr("fiona.session.boto3", None)
        assert isinstance(Session.aws_or_dummy(), DummySession)


def test_no_sign_request(monkeypatch):
    """If AWS_NO_SIGN_REQUEST is set do not default to aws_unsigned=False"""
    monkeypatch.setenv("AWS_NO_SIGN_REQUEST", "YES")
    assert AWSSession().unsigned


def test_no_credentialization_if_unsigned(monkeypatch):
    """Don't get credentials if we're not signing, see #1984"""
    sesh = AWSSession(aws_unsigned=True)
    assert sesh._creds is None


def test_azure_session_class():
    """AzureSession works"""
    azure_session = AzureSession(
        azure_storage_account="foo", azure_storage_access_key="bar"
    )
    assert azure_session._creds
    assert azure_session.get_credential_options()["AZURE_STORAGE_ACCOUNT"] == "foo"
    assert azure_session.get_credential_options()["AZURE_STORAGE_ACCESS_KEY"] == "bar"


def test_azure_session_class_connection_string():
    """AzureSession works"""
    azure_session = AzureSession(
        azure_storage_connection_string="AccountName=myaccount;AccountKey=MY_ACCOUNT_KEY"
    )
    assert azure_session._creds
    assert (
        azure_session.get_credential_options()["AZURE_STORAGE_CONNECTION_STRING"]
        == "AccountName=myaccount;AccountKey=MY_ACCOUNT_KEY"
    )


def test_session_factory_az_kwargs():
    """Get an AzureSession for az:// paths with keywords"""
    sesh = Session.from_path(
        "az://lol/wut", azure_storage_account="foo", azure_storage_access_key="bar"
    )
    assert isinstance(sesh, AzureSession)
    assert sesh.get_credential_options()["AZURE_STORAGE_ACCOUNT"] == "foo"
    assert sesh.get_credential_options()["AZURE_STORAGE_ACCESS_KEY"] == "bar"


def test_session_factory_az_kwargs_connection_string():
    """Get an AzureSession for az:// paths with keywords"""
    sesh = Session.from_path(
        "az://lol/wut",
        azure_storage_connection_string="AccountName=myaccount;AccountKey=MY_ACCOUNT_KEY",
    )
    assert isinstance(sesh, AzureSession)
    assert (
        sesh.get_credential_options()["AZURE_STORAGE_CONNECTION_STRING"]
        == "AccountName=myaccount;AccountKey=MY_ACCOUNT_KEY"
    )


def test_azure_no_sign_request(monkeypatch):
    """If AZURE_NO_SIGN_REQUEST is set do not default to azure_unsigned=False"""
    monkeypatch.setenv("AZURE_NO_SIGN_REQUEST", "YES")
    assert AzureSession().unsigned


def test_azure_session_class_unsigned():
    """AzureSession works"""
    sesh = AzureSession(azure_unsigned=True, azure_storage_account="naipblobs")
    assert sesh.get_credential_options()["AZURE_NO_SIGN_REQUEST"] == "YES"
    assert sesh.get_credential_options()["AZURE_STORAGE_ACCOUNT"] == "naipblobs"