import json
from lxml import etree
import pytest


@pytest.mark.django_db
def test_host_meta_json(client):
    res = client.get('/.well-known/host-meta.json')
    assert res.status_code == 200
    assert res['Content-Type'] == 'application/json'
    meta = json.loads(res.content)
    assert meta.keys() == {'links', 'subject'}
    assert meta['subject'] == 'https://example.com'
    assert len(meta['links']) == 13


@pytest.mark.django_db
def test_host_meta_xml(client):
    res = client.get('/.well-known/host-meta')
    assert res.status_code == 200
    assert res['Content-Type'] == 'application/xrd+xml'
    root = etree.XML(res.content)
    ns = '{http://docs.oasis-open.org/ns/xri/xrd-1.0}'
    assert root.tag == (ns + 'XRD')
    assert root.findtext(ns + 'Subject') == 'https://example.com'
    assert len(root.findall(ns + 'Link')) == 13