mirror of
https://github.com/samsonjs/vdirsyncer.git
synced 2026-03-25 08:55:50 +00:00
121 lines
3.4 KiB
Python
121 lines
3.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
'''
|
|
tests.storage.test_http
|
|
~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
:copyright: (c) 2014 Markus Unterwaditzer & contributors
|
|
:license: MIT, see LICENSE for more details.
|
|
'''
|
|
|
|
import pytest
|
|
|
|
from requests import Response
|
|
|
|
from tests import normalize_item
|
|
|
|
from vdirsyncer.storage.http import HttpStorage, prepare_auth
|
|
|
|
|
|
def test_list(monkeypatch):
|
|
collection_url = 'http://127.0.0.1/calendar/collection.ics'
|
|
|
|
items = [
|
|
(u'BEGIN:VEVENT\n'
|
|
u'SUMMARY:Eine Kurzinfo\n'
|
|
u'DESCRIPTION:Beschreibung des Termines\n'
|
|
u'END:VEVENT'),
|
|
(u'BEGIN:VEVENT\n'
|
|
u'SUMMARY:Eine zweite Küèrzinfo\n'
|
|
u'DESCRIPTION:Beschreibung des anderen Termines\n'
|
|
u'BEGIN:VALARM\n'
|
|
u'ACTION:AUDIO\n'
|
|
u'TRIGGER:19980403T120000\n'
|
|
u'ATTACH;FMTTYPE=audio/basic:http://host.com/pub/ssbanner.aud\n'
|
|
u'REPEAT:4\n'
|
|
u'DURATION:PT1H\n'
|
|
u'END:VALARM\n'
|
|
u'END:VEVENT')
|
|
]
|
|
|
|
responses = [
|
|
u'\n'.join([u'BEGIN:VCALENDAR'] + items + [u'END:VCALENDAR'])
|
|
] * 2
|
|
|
|
def get(self, method, url, *a, **kw):
|
|
assert method == 'GET'
|
|
assert url == collection_url
|
|
r = Response()
|
|
r.status_code = 200
|
|
assert responses
|
|
r._content = responses.pop().encode('utf-8')
|
|
r.headers['Content-Type'] = 'text/icalendar'
|
|
r.encoding = 'ISO-8859-1'
|
|
return r
|
|
|
|
monkeypatch.setattr('requests.sessions.Session.request', get)
|
|
|
|
s = HttpStorage(url=collection_url)
|
|
|
|
found_items = {}
|
|
|
|
for href, etag in s.list():
|
|
item, etag2 = s.get(href)
|
|
assert item.uid is None
|
|
assert etag2 == etag
|
|
found_items[normalize_item(item)] = href
|
|
|
|
expected = set(normalize_item(u'BEGIN:VCALENDAR\n' + x + '\nEND:VCALENDAR')
|
|
for x in items)
|
|
|
|
assert set(found_items) == expected
|
|
|
|
for href, etag in s.list():
|
|
item, etag2 = s.get(href)
|
|
assert item.uid is None
|
|
assert etag2 == etag
|
|
assert found_items[normalize_item(item)] == href
|
|
|
|
|
|
def test_readonly_param():
|
|
url = u'http://example.com/'
|
|
with pytest.raises(ValueError):
|
|
HttpStorage(url=url, read_only=False)
|
|
|
|
a = HttpStorage(url=url, read_only=True).read_only
|
|
b = HttpStorage(url=url, read_only=None).read_only
|
|
assert a is b is True
|
|
|
|
|
|
def test_prepare_auth():
|
|
assert prepare_auth(None, '', '') is None
|
|
|
|
assert prepare_auth('basic', 'user', 'pwd') == ('user', 'pwd')
|
|
with pytest.raises(ValueError) as excinfo:
|
|
assert prepare_auth('basic', '', 'pwd')
|
|
assert 'you need to specify username and password' in \
|
|
str(excinfo.value).lower()
|
|
|
|
from requests.auth import HTTPDigestAuth
|
|
assert isinstance(prepare_auth('digest', 'user', 'pwd'),
|
|
HTTPDigestAuth)
|
|
|
|
with pytest.raises(ValueError) as excinfo:
|
|
prepare_auth('ladida', 'user', 'pwd')
|
|
|
|
assert 'unknown authentication method' in str(excinfo.value).lower()
|
|
|
|
|
|
@pytest.mark.parametrize('auth', (None, 'guess'))
|
|
def test_prepare_auth_guess(monkeypatch, auth):
|
|
import requests_toolbelt
|
|
|
|
assert isinstance(prepare_auth(auth, 'user', 'pwd'),
|
|
requests_toolbelt.GuessAuth)
|
|
|
|
if hasattr(requests_toolbelt, 'GuessAuth'):
|
|
monkeypatch.delattr(requests_toolbelt, 'GuessAuth')
|
|
|
|
with pytest.raises(RuntimeError) as excinfo:
|
|
prepare_auth(auth, 'user', 'pwd')
|
|
|
|
assert 'requests_toolbelt is too old' in str(excinfo.value).lower()
|