b4tman/sync_ics2gcal
1
0
Fork 0
mirror of https://github.com/b4tman/sync_ics2gcal synced 2024-10-31 18:56:54 +00:00
Code Issues Releases Activity

add shuffle to sync tests

Browse Source
This commit is contained in:
Dmitry Belyaev 2018-04-09 15:08:45 +03:00
parent 0c145bca07
commit e2d92458d3
Signed by: b4tman
GPG Key ID: 014E87EC54B77673
1 changed files with 37 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
tests/test_sync.py
View File

@ -2,6 +2,8 @@ import datetime
import hashlib import hashlib
import operator import operator
import unittest import unittest
from copy import deepcopy
from random import shuffle
import dateutil.parser import dateutil.parser
from pytz import timezone, utc from pytz import timezone, utc
@ -22,12 +24,13 @@ class TestCalendarSync(unittest.TestCase):
def gen_events(start, stop, start_time): def gen_events(start, stop, start_time):
one_hour = datetime.datetime( one_hour = datetime.datetime(
1, 1, 1, 2) - datetime.datetime(1, 1, 1, 1) 1, 1, 1, 2) - datetime.datetime(1, 1, 1, 1)
result = []
for i in range(start, stop): for i in range(start, stop):
event_start = start_time + (one_hour * i) event_start = start_time + (one_hour * i)
event_end = event_start + one_hour event_end = event_start + one_hour
updated = utc.normalize( updated = utc.normalize(
event_start.astimezone(utc)).replace(tzinfo=None) event_start.astimezone(utc)).replace(tzinfo=None)
yield { result.append({
'summary': 'test event __ {}'.format(i), 'summary': 'test event __ {}'.format(i),
'location': 'la la la {}'.format(i), 'location': 'la la la {}'.format(i),
'description': 'test TEST -- test event {}'.format(i), 'description': 'test TEST -- test event {}'.format(i),
@ -40,12 +43,15 @@ class TestCalendarSync(unittest.TestCase):
"iCalUID": "{}@test.com".format(TestCalendarSync.sha1("test - event {}".format(i))), "iCalUID": "{}@test.com".format(TestCalendarSync.sha1("test - event {}".format(i))),
"updated": updated.isoformat() + 'Z', "updated": updated.isoformat() + 'Z',
"created": updated.isoformat() + 'Z' "created": updated.isoformat() + 'Z'
} })
return result
@staticmethod @staticmethod
def gen_list_to_compare(start, stop): def gen_list_to_compare(start, stop):
result = []
for i in range(start, stop): for i in range(start, stop):
yield {'iCalUID': 'test{}'.format(i)} result.append({'iCalUID': 'test{}'.format(i)})
return result
@staticmethod @staticmethod
def get_start_date(event): def get_start_date(event):
@ -58,19 +64,31 @@ class TestCalendarSync(unittest.TestCase):
return dateutil.parser.parse(start_date) return dateutil.parser.parse(start_date)
def test_compare(self): def test_compare(self):
lst_src = list(TestCalendarSync.gen_list_to_compare(1, 5)) lst_src = TestCalendarSync.gen_list_to_compare(1, 11)
lst_dst = list(TestCalendarSync.gen_list_to_compare(3, 7)) lst_dst = TestCalendarSync.gen_list_to_compare(6, 16)
lst_src_rnd = deepcopy(lst_src)
lst_dst_rnd = deepcopy(lst_dst)
shuffle(lst_src_rnd)
shuffle(lst_dst_rnd)
to_ins, to_upd, to_del = CalendarSync._events_list_compare( to_ins, to_upd, to_del = CalendarSync._events_list_compare(
lst_src, lst_dst) lst_src_rnd, lst_dst_rnd)
self.assertEqual(len(to_ins), 2) self.assertEqual(len(to_ins), 5)
self.assertEqual(len(to_upd), 2) self.assertEqual(len(to_upd), 5)
self.assertEqual(len(to_del), 2) self.assertEqual(len(to_del), 5)
self.assertEqual(to_ins, lst_src[:2]) self.assertEqual(
self.assertEqual(to_upd, list(zip(lst_src[2:4], lst_dst[:2]))) sorted(to_ins, key=lambda x: x['iCalUID']), lst_src[:5])
self.assertEqual(to_del, lst_dst[2:]) self.assertEqual(
sorted(to_del, key=lambda x: x['iCalUID']), lst_dst[5:])
to_upd_ok = list(zip(lst_src[5:], lst_dst[:5]))
self.assertEqual(len(to_upd), len(to_upd_ok))
for item in to_upd_ok:
self.assertIn(item, to_upd)
def test_filter_events_by_date(self): def test_filter_events_by_date(self):
msk = timezone('Europe/Moscow') msk = timezone('Europe/Moscow')
@ -81,7 +99,9 @@ class TestCalendarSync(unittest.TestCase):
1, 1, 1, 2) - datetime.datetime(1, 1, 1, 1) 1, 1, 1, 2) - datetime.datetime(1, 1, 1, 1)
date_cmp = msk_now + (one_hour * 5) date_cmp = msk_now + (one_hour * 5)
events = list(TestCalendarSync.gen_events(1, 11, msk_now)) events = TestCalendarSync.gen_events(1, 11, msk_now)
shuffle(events)
events_pending = CalendarSync._filter_events_by_date( events_pending = CalendarSync._filter_events_by_date(
events, date_cmp, operator.ge) events, date_cmp, operator.ge)
events_past = CalendarSync._filter_events_by_date( events_past = CalendarSync._filter_events_by_date(
@ -91,7 +111,8 @@ class TestCalendarSync(unittest.TestCase):
self.assertEqual(len(events_past), 4) self.assertEqual(len(events_past), 4)
for event in events_pending: for event in events_pending:
self.assertGreaterEqual(TestCalendarSync.get_start_date(event), date_cmp) self.assertGreaterEqual(
TestCalendarSync.get_start_date(event), date_cmp)
for event in events_past: for event in events_past:
self.assertLess(TestCalendarSync.get_start_date(event), date_cmp) self.assertLess(TestCalendarSync.get_start_date(event), date_cmp)
@ -106,8 +127,8 @@ class TestCalendarSync(unittest.TestCase):
date_upd = msk_now + (one_hour * 5) date_upd = msk_now + (one_hour * 5)
count = 10 count = 10
events_old = list(TestCalendarSync.gen_events(1, 1 + count, msk_now)) events_old = TestCalendarSync.gen_events(1, 1 + count, msk_now)
events_new = list(TestCalendarSync.gen_events(1, 1 + count, date_upd)) events_new = TestCalendarSync.gen_events(1, 1 + count, date_upd)
sync1 = CalendarSync(None, None) sync1 = CalendarSync(None, None)
sync1.to_update = list(zip(events_new, events_old)) sync1.to_update = list(zip(events_new, events_old))