148 lines
5.5 KiB
Python
148 lines
5.5 KiB
Python
|
|
import unittest
|
|
|
|
import json
|
|
from datetime import timedelta, tzinfo
|
|
|
|
from dateutil import tz
|
|
from isodate.duration import Duration
|
|
|
|
from icalendar_timeseries_server.config import _keycheck, _parse_timedelta, _parse_timezone, Config
|
|
|
|
|
|
_CONFIG_VALID = """
|
|
{
|
|
"addr": "127.0.0.1",
|
|
"port": 8090,
|
|
"start_delta": "-PT3H",
|
|
"end_delta": "P30D",
|
|
"tz": "Europe/Zurich",
|
|
"calendars": {
|
|
"private": {
|
|
"url": "https://example.cloud/dav/me/private.ics",
|
|
"auth": {
|
|
"type": "basic",
|
|
"username": "me",
|
|
"password": "mysupersecurepassword"
|
|
}
|
|
},
|
|
"public": {
|
|
"url": "https://example.cloud/dav/me/public.ics",
|
|
"interval": "P1D"
|
|
},
|
|
"confidential": {
|
|
"url": "https://example.cloud/dav/me/confidential.ics",
|
|
"interval": "PT5M",
|
|
"ca": "/etc/ssl/ca.pem",
|
|
"auth": {
|
|
"type": "tls",
|
|
"keyfile": "/etc/ssl/client.pem",
|
|
"passphrase": "mysupersecurepassword"
|
|
}
|
|
}
|
|
},
|
|
"key_replace": {
|
|
"summary": "01_summary",
|
|
"description": "02_description"
|
|
},
|
|
"value_replace": {
|
|
"summary": "{{ summary|truncate(100) }}",
|
|
"description": "{{ description|truncate(100) }}"
|
|
}
|
|
}
|
|
"""
|
|
|
|
|
|
class ConfigTest(unittest.TestCase):
|
|
|
|
def test_keycheck_valid(self) -> None:
|
|
config = {
|
|
'foo': 'bar',
|
|
'bar': 42,
|
|
'baz': [],
|
|
'qux': {}
|
|
}
|
|
strvalue = _keycheck('foo', config, str, '', valid_values=['foo', 'bar', 'baz'])
|
|
self.assertEqual(strvalue, 'bar')
|
|
strdefvalue = _keycheck('fooo', config, str, '', default_value='baar')
|
|
self.assertEqual(strdefvalue, 'baar')
|
|
stroptvalue = _keycheck('foooo', config, str, '', optional=True)
|
|
self.assertIsNone(stroptvalue)
|
|
intvalue = _keycheck('bar', config, int, '')
|
|
self.assertEqual(intvalue, 42)
|
|
listvalue = _keycheck('baz', config, list, '')
|
|
self.assertEqual(listvalue, [])
|
|
objvalue = _keycheck('qux', config, dict, '')
|
|
self.assertEqual(objvalue, {})
|
|
|
|
def test_keycheck_missing(self) -> None:
|
|
config = {
|
|
'foo': 'bar',
|
|
}
|
|
with self.assertRaises(KeyError):
|
|
_keycheck('baz', config, str, '')
|
|
|
|
def test_keycheck_type(self) -> None:
|
|
config = {
|
|
'foo': 'bar',
|
|
}
|
|
with self.assertRaises(TypeError):
|
|
_keycheck('foo', config, int, '')
|
|
|
|
def test_keycheck_value(self) -> None:
|
|
config = {
|
|
'foo': '1337',
|
|
}
|
|
with self.assertRaises(ValueError):
|
|
_keycheck('foo', config, str, '', valid_values=['foo', 'bar', 'baz'])
|
|
|
|
def test_parse_timedelta(self) -> None:
|
|
config = {
|
|
'pos': 'PT42S',
|
|
'neg': '-PT42S'
|
|
}
|
|
self.assertEqual(_parse_timedelta('zero', config, '', default_value='PT'), timedelta(seconds=0))
|
|
self.assertEqual(_parse_timedelta('pos', config, ''), timedelta(seconds=42))
|
|
self.assertEqual(_parse_timedelta('neg', config, ''), timedelta(seconds=-42))
|
|
self.assertEqual(_parse_timedelta('pos', config, '', force_positive=True), timedelta(seconds=42))
|
|
with self.assertRaises(ValueError):
|
|
_parse_timedelta('neg', config, '', force_positive=True)
|
|
|
|
def test_parse_timezone(self) -> None:
|
|
config = {
|
|
'tz': 'Europe/Zurich',
|
|
'notz': 'North/Winterfell'
|
|
}
|
|
self.assertEqual(_parse_timezone('tz', config, ''), tz.gettz('Europe/Zurich'))
|
|
self.assertEqual(_parse_timezone('def', config, '', default_value='Europe/Berlin'),
|
|
tz.gettz('Europe/Berlin'))
|
|
with self.assertRaises(ValueError):
|
|
_parse_timezone('notz', config, '')
|
|
|
|
def test_parse_full_config_valid(self):
|
|
config = Config(json.loads(_CONFIG_VALID))
|
|
self.assertEqual(config.addr, '127.0.0.1')
|
|
self.assertEqual(config.port, 8090)
|
|
self.assertEqual(config.start_delta, Duration(hours=-3))
|
|
self.assertEqual(config.end_delta, Duration(days=30))
|
|
self.assertEqual(config.tz, tz.gettz('Europe/Zurich'))
|
|
|
|
def test_parse_calendars(self):
|
|
config = Config(json.loads(_CONFIG_VALID))
|
|
self.assertEqual({'public', 'private', 'confidential'}, config.calendars.keys())
|
|
|
|
self.assertEqual('https://example.cloud/dav/me/public.ics', config.calendars['public'].url)
|
|
self.assertEqual(Duration(days=1), config.calendars['public'].interval)
|
|
self.assertEqual('none', config.calendars['public']._authtype)
|
|
|
|
self.assertEqual('https://example.cloud/dav/me/private.ics', config.calendars['private'].url)
|
|
self.assertEqual(Duration(minutes=15), config.calendars['private'].interval)
|
|
self.assertEqual('basic', config.calendars['private']._authtype)
|
|
self.assertEqual('Basic bWU6bXlzdXBlcnNlY3VyZXBhc3N3b3Jk',
|
|
config.calendars['private']._request_headers['Authorization'])
|
|
|
|
self.assertEqual('https://example.cloud/dav/me/confidential.ics', config.calendars['confidential'].url)
|
|
self.assertEqual(Duration(minutes=5), config.calendars['confidential'].interval)
|
|
self.assertEqual('tls', config.calendars['confidential']._authtype)
|
|
self.assertEqual('/etc/ssl/client.pem', config.calendars['confidential']._tls_keyfile)
|
|
self.assertEqual('mysupersecurepassword', config.calendars['confidential']._tls_passphrase)
|