"""Test the template helpers."""

import flexmock
import pytest

from auth import api
from auth import config
from auth import templates
from auth.connectors import manifest


def test_get_asset():
    """Test getting an asset by name."""
    with api.app.test_request_context('/'):
        assert templates.asset('assets/name.js') == '/static/assets/name.js'


@pytest.mark.parametrize(
    'environment', [config.PROD_ENVIRONMENT, config.QA_ENVIRONMENT])
def test_get_asset_in_prod_or_qa(environment):
    """Test getting an asset in production or qa."""
    asset_name = 'asset/name'
    cdn_url = 'http://cdn.theorchard.io/frontend-auth/18477344/asset/name'
    flexmock(config, ENVIRONMENT=environment)
    (flexmock(manifest.manifest)
        .should_receive('get_url')
        .and_return(cdn_url)
        .once())
    assert templates.asset(asset_name) is cdn_url