"""Test image asset util."""

import pytest

from ows_product_physical import config
from ows_product_physical.constant import assets
from ows_product_physical.utils import image_util


@pytest.mark.parametrize(('environment', 'expected_asset_path'), [
    ('dev', assets.QA_IMAGE_URL), ('prod', assets.PROD_IMAGE_URL),
    ('qa', assets.QA_IMAGE_URL)])
def test_image_path(environment, expected_asset_path, monkeypatch):
    """Test cloudfront image path util."""
    monkeypatch.setattr(config, 'ENVIRONMENT', environment)
    asset_path = image_util.get_asset_path()
    assert asset_path == expected_asset_path