1
0

vdir_test.py 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130
  1. # ical2vdir - convert .ics file to vdir directory
  2. #
  3. # Copyright (C) 2020 Fabian Peter Hammerle <fabian@hammerle.me>
  4. #
  5. # This program is free software: you can redistribute it and/or modify
  6. # it under the terms of the GNU General Public License as published by
  7. # the Free Software Foundation, either version 3 of the License, or
  8. # (at your option) any later version.
  9. #
  10. # This program is distributed in the hope that it will be useful,
  11. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13. # GNU General Public License for more details.
  14. #
  15. # You should have received a copy of the GNU General Public License
  16. # along with this program. If not, see <https://www.gnu.org/licenses/>.
  17. import copy
  18. import os
  19. import pathlib
  20. import unittest.mock
  21. import icalendar.cal
  22. import pytest
  23. import ical2vdir
  24. def _normalize_ical(ical: bytes) -> bytes:
  25. return ical.replace(b"\n", b"\r\n")
  26. _SINGLE_EVENT_ICAL = _normalize_ical(
  27. b"""BEGIN:VEVENT
  28. SUMMARY:party
  29. DTSTART:20201024T100000Z
  30. DTEND:20201026T120000Z
  31. DTSTAMP:20200205T160640Z
  32. UID:1qa2ws3ed4rf5tg@google.com
  33. SEQUENCE:0
  34. CREATED:20191231T103841Z
  35. DESCRIPTION:
  36. LAST-MODIFIED:20191231T103841Z
  37. LOCATION:
  38. STATUS:CONFIRMED
  39. TRANSP:OPAQUE
  40. END:VEVENT
  41. """
  42. )
  43. # pylint: disable=protected-access
  44. # tmp_path fixture: https://github.com/pytest-dev/pytest/blob/5.4.3/src/_pytest/tmpdir.py#L191
  45. def test__write_event_cleanup(tmp_path: pathlib.Path):
  46. event = icalendar.cal.Event.from_ical(_SINGLE_EVENT_ICAL)
  47. with unittest.mock.patch("os.unlink") as unlink_mock:
  48. with pytest.raises(IsADirectoryError):
  49. ical2vdir._write_event(event, tmp_path)
  50. # assert_called_once new in python3.6
  51. assert unlink_mock.call_count == 1
  52. unlink_args, _ = unlink_mock.call_args
  53. os.unlink(unlink_args[0])
  54. @pytest.mark.parametrize(
  55. ("event_ical", "expected_filename"),
  56. [
  57. (
  58. _SINGLE_EVENT_ICAL,
  59. "1qa2ws3ed4rf5tg@google.com.ics",
  60. ),
  61. (
  62. b"""BEGIN:VEVENT
  63. SUMMARY:work
  64. DTSTART;TZID=Europe/Vienna:20150924T090000
  65. DTEND;TZID=Europe/Vienna:20150924T123000
  66. DTSTAMP:20200205T160640Z
  67. UID:1qa2ws3ed4rf5tg@google.com
  68. RECURRENCE-ID;TZID=Europe/Vienna:20150924T090000
  69. SEQUENCE:5
  70. CREATED:20140228T212925Z
  71. DESCRIPTION:
  72. LAST-MODIFIED:20150908T181423Z
  73. LOCATION:
  74. STATUS:CONFIRMED
  75. TRANSP:TRANSPARENT
  76. END:VEVENT
  77. """,
  78. "1qa2ws3ed4rf5tg@google.com.20150924T090000+0200.ics",
  79. ),
  80. ],
  81. )
  82. def test__event_vdir_filename(event_ical, expected_filename):
  83. event = icalendar.cal.Event.from_ical(event_ical)
  84. assert ical2vdir._event_vdir_filename(event) == expected_filename
  85. @pytest.mark.parametrize("event_ical", [_SINGLE_EVENT_ICAL])
  86. def test__sync_event_create(tmp_path: pathlib.Path, event_ical):
  87. event = icalendar.cal.Event.from_ical(event_ical)
  88. ical2vdir._sync_event(event, tmp_path)
  89. (ics_path,) = tmp_path.iterdir()
  90. assert ics_path.name == "1qa2ws3ed4rf5tg@google.com.ics"
  91. assert ics_path.read_bytes() == _SINGLE_EVENT_ICAL
  92. @pytest.mark.parametrize("event_ical", [_SINGLE_EVENT_ICAL])
  93. def test__sync_event_update(tmp_path: pathlib.Path, event_ical):
  94. event = icalendar.cal.Event.from_ical(event_ical)
  95. ical2vdir._sync_event(event, tmp_path)
  96. event["SUMMARY"] += " suffix"
  97. ical2vdir._sync_event(event, tmp_path)
  98. (ics_path,) = tmp_path.iterdir()
  99. assert ics_path.name == event["UID"] + ".ics"
  100. assert ics_path.read_bytes() == _SINGLE_EVENT_ICAL.replace(
  101. b"party", b"party suffix"
  102. )
  103. @pytest.mark.parametrize("event_ical", [_SINGLE_EVENT_ICAL])
  104. def test__sync_event_unchanged(tmp_path: pathlib.Path, event_ical):
  105. event = icalendar.cal.Event.from_ical(event_ical)
  106. ical2vdir._sync_event(event, tmp_path)
  107. (ics_path,) = tmp_path.iterdir()
  108. old_stat = copy.deepcopy(ics_path.stat())
  109. ical2vdir._sync_event(event, tmp_path)
  110. assert ics_path.stat() == old_stat
  111. assert ics_path.read_bytes() == _SINGLE_EVENT_ICAL