import pytest
from pyinfra.config import get_config, Config
import os

@pytest.fixture(params=["aws", "azure"])
def storage_config(request) -> Config:
    if request.param == "aws":
        os.environ["STORAGE_BACKEND"] = "s3"
        os.environ["STORAGE_BUCKET_NAME"] = "pyinfra-test-bucket"
        os.environ["STORAGE_ENDPOINT"] = "https://s3.amazonaws.com"
        os.environ["STORAGE_KEY"] = "AKIA4QVP6D4LCDAGYGN2"
        os.environ["STORAGE_SECRET"] = "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"
        os.environ["STORAGE_REGION"] = "eu-west-1"
    else:
        os.environ["STORAGE_BACKEND"] = "azure"
        os.environ["STORAGE_AZURECONTAINERNAME"] = "pyinfra-test-bucket"
        os.environ["STORAGE_AZURECONNECTIONSTRING"] = "DefaultEndpointsProtocol=https;AccountName=iqserdevelopment;AccountKey=4imAbV9PYXaztSOMpIyAClg88bAZCXuXMGJG0GA1eIBpdh2PlnFGoRBnKqLy2YZUSTmZ3wJfC7tzfHtuC6FEhQ==;EndpointSuffix=core.windows.net"
    
    return get_config()