test_invenio_s3.py

# -*- coding: utf-8 -*-
#
# Copyright (C) 2018, 2019 Esteban J. G. Gabancho.
# Copyright (C) 2024 KTH Royal Institute of Technology.
#
# Invenio-S3 is free software; you can redistribute it and/or modify it
# under the terms of the MIT License; see LICENSE file for more details.
"""Module tests."""

from invenio_s3 import InvenioS3


def test_version():
    """Test version import."""
    from invenio_s3 import __version__

    assert __version__


def test_init(appctx):
    """Test extension initialization."""
    assert "invenio-s3" in appctx.extensions

    appctx.config["S3_ENDPOINT_URL"] = "https://example.com:1234"
    appctx.config["S3_REGION_NAME"] = "eu-west-1"
    s3_connection_info = appctx.extensions["invenio-s3"].init_s3fs_info
    assert (
        s3_connection_info["client_kwargs"]["endpoint_url"]
        == "https://example.com:1234"
    )
    assert s3_connection_info["client_kwargs"]["region_name"] == "eu-west-1"


def test_access_key(appctx):
    """Test correct access key works together with flawed one."""
    appctx.config["S3_ACCCESS_KEY_ID"] = "secret"
    try:
        # Delete the cached value in case it's there already
        del appctx.extensions["invenio-s3"].__dict__["init_s3fs_info"]
    except KeyError:
        pass
    s3_connection_info = appctx.extensions["invenio-s3"].init_s3fs_info
    assert s3_connection_info["key"] == "secret"


def test_secret_key(appctx):
    """Test correct secret key works together with flawed one."""
    appctx.config["S3_SECRECT_ACCESS_KEY"] = "secret"
    try:
        # Delete the cached value in case it's there already
        del appctx.extensions["invenio-s3"].__dict__["init_s3fs_info"]
    except KeyError:
        pass
    s3_connection_info = appctx.extensions["invenio-s3"].init_s3fs_info
    assert s3_connection_info["secret"] == "secret"