import unittest
from gcal_sync import CalendarConverter

ics_empty = """BEGIN:VCALENDAR
VERSION:2.0
PRODID:-//test//test//ES
CALSCALE:GREGORIAN
METHOD:PUBLISH
END:VCALENDAR
"""

ics_empty_event = """BEGIN:VCALENDAR
BEGIN:VEVENT
END:VEVENT
END:VCALENDAR
"""

ics_event_no_end = """BEGIN:VCALENDAR
BEGIN:VEVENT
UID:uisgtr8tre93wewe0yr8wqy@test.com
DTSTART;VALUE=DATE:20180215
END:VEVENT
END:VCALENDAR
"""

ics_event_date_val = """BEGIN:VCALENDAR
BEGIN:VEVENT
UID:uisgtr8tre93wewe0yr8wqy@test.com
DTSTART;VALUE=DATE:20180215
DTEND;VALUE=DATE:20180217
END:VEVENT
END:VCALENDAR
"""

ics_event_datetime_utc_val = """BEGIN:VCALENDAR
BEGIN:VEVENT
UID:uisgtr8tre93wewe0yr8wqy@test.com
DTSTART;VALUE=DATE-TIME:20180319T092001Z
DTEND:20180319T102001Z
END:VEVENT
END:VCALENDAR
"""

ics_event_date_duration = """BEGIN:VCALENDAR
BEGIN:VEVENT
UID:uisgtr8tre93wewe0yr8wqy@test.com
DTSTART;VALUE=DATE:20180215
DURATION:P3D
END:VEVENT
END:VCALENDAR
"""

ics_event_datetime_utc_duration = """BEGIN:VCALENDAR
BEGIN:VEVENT
UID:uisgtr8tre93wewe0yr8wqy@test.com
DTSTART;VALUE=DATE-TIME:20180319T092001Z
DURATION:P2DT1H5M
END:VEVENT
END:VCALENDAR
"""

ics_event_created_updated = """BEGIN:VCALENDAR
BEGIN:VEVENT
UID:uisgtr8tre93wewe0yr8wqy@test.com
DTSTART:20180215
DTEND:20180217
CREATED:20180320T071155Z
LAST-MODIFIED:20180326T120235Z
END:VEVENT
END:VCALENDAR
"""

class TestCalendarConverter(unittest.TestCase):
    def test_empty_calendar(self):
      converter = CalendarConverter()
      converter.loads(ics_empty)
      evnts = converter.events_to_gcal()
      self.assertEqual(len(evnts), 0)
    
    def test_empty_event(self):
      converter = CalendarConverter()
      converter.loads(ics_empty_event)
      with self.assertRaises(KeyError):
        converter.events_to_gcal()
    
    def test_event_no_end(self):
      converter = CalendarConverter()
      converter.loads(ics_event_no_end)
      with self.assertRaises(ValueError):
        converter.events_to_gcal()

    def test_event_date_values(self):
      converter = CalendarConverter()
      converter.loads(ics_event_date_val)
      events = converter.events_to_gcal()
      self.assertEqual(len(events), 1)
      event = events[0]
      self.assertEqual(event['start'], {
        'date': '2018-02-15'
      })
      self.assertEqual(event['end'], {
        'date': '2018-02-17'
      })

    def test_event_datetime_utc_values(self):
      converter = CalendarConverter()
      converter.loads(ics_event_datetime_utc_val)
      events = converter.events_to_gcal()
      self.assertEqual(len(events), 1)
      event = events[0]
      self.assertEqual(event['start'], {
        'dateTime': '2018-03-19T09:20:01.000001Z'
      })
      self.assertEqual(event['end'], {
        'dateTime': '2018-03-19T10:20:01.000001Z'
      })
      
    def test_event_date_duration(self):
      converter = CalendarConverter()
      converter.loads(ics_event_date_duration)
      events = converter.events_to_gcal()
      self.assertEqual(len(events), 1)
      event = events[0]
      self.assertEqual(event['start'], {
        'date': '2018-02-15'
      })
      self.assertEqual(event['end'], {
        'date': '2018-02-18'
      })

    def test_event_datetime_utc_duration(self):
      converter = CalendarConverter()
      converter.loads(ics_event_datetime_utc_duration)
      events = converter.events_to_gcal()
      self.assertEqual(len(events), 1)
      event = events[0]
      self.assertEqual(event['start'], {
        'dateTime': '2018-03-19T09:20:01.000001Z'
      })
      self.assertEqual(event['end'], {
        'dateTime': '2018-03-21T10:25:01.000001Z'
      })

    def test_event_created_updated(self):
      converter = CalendarConverter()
      converter.loads(ics_event_created_updated)
      events = converter.events_to_gcal()
      self.assertEqual(len(events), 1)
      event = events[0]
      self.assertEqual(event['created'], '2018-03-20T07:11:55.000001Z')
      self.assertEqual(event['updated'], '2018-03-26T12:02:35.000001Z')

if __name__ == '__main__':
    unittest.main()