import pytest

from funkwhale_api.federation import keys


@pytest.mark.parametrize('raw, expected', [
    ('algorithm="test",keyId="https://test.com"', 'https://test.com'),
    ('keyId="https://test.com",algorithm="test"', 'https://test.com'),
])
def test_get_key_from_header(raw, expected):
    r = keys.get_key_id_from_signature_header(raw)
    assert r == expected


@pytest.mark.parametrize('raw', [
    'algorithm="test",keyid="badCase"',
    'algorithm="test",wrong="wrong"',
    'keyId = "wrong"',
    'keyId=\'wrong\'',
    'keyId="notanurl"',
    'keyId="wrong://test.com"',
])
def test_get_key_from_header_invalid(raw):
    with pytest.raises(ValueError):
        keys.get_key_id_from_signature_header(raw)