add cattrs tests

Author: getzze

pyproject.toml

@@ -70,6 +70,8 @@ tests = [
     "pytest>=6.0",
     "pytest-cov",
     "pytest-xdist",
+    "cattrs",
+    "packaging",
     "sympy",
     "vcrpy>=5",  # keep synchronized with docs dependencies
     "importlib_metadata>=4.6; python_version<'3.10'",

tests/test_subtitle.py

@@ -1,11 +1,14 @@
 from __future__ import annotations
 
 import os
+from importlib.metadata import version as get_version
 from textwrap import dedent
 from typing import TYPE_CHECKING
 
 import pytest
 from babelfish import Language  # type: ignore[import-untyped]
+from cattrs import structure, unstructure
+from packaging.version import Version
 
 from subliminal.subtitle import (
     EmbeddedSubtitle,
@@ -566,3 +569,24 @@ def test_embedded_subtitle_info_foreign_only(monkeypatch: pytest.MonkeyPatch) ->
     assert subtitle.foreign_only is True
     assert isinstance(subtitle.id, str)
     assert isinstance(subtitle.info, str)
+
+
+@pytest.mark.skipif(
+    Version(get_version('babelfish')) <= Version('0.6.1'),
+    reason='babelfish.Language needs to be a hashable dataclass',
+)
+def test_serialize_subtitle() -> None:
+    content = b'1\n00:01:00 --> 00:02:00\nA long text\n\n'
+
+    subtitle = Subtitle(
+        language=Language('eng'),
+        hearing_impaired=False,
+        page_link=None,
+        encoding=None,
+    )
+    subtitle.content = content
+
+    ser = unstructure(subtitle)
+    new_subtitle = structure(ser, Subtitle)
+
+    assert subtitle == new_subtitle

tests/test_video.py

@@ -3,10 +3,13 @@
 
 import os
 from datetime import datetime, timedelta, timezone
+from importlib.metadata import version as get_version
 from typing import TYPE_CHECKING
 from unittest.mock import Mock
 
 import pytest
+from cattrs import structure, unstructure
+from packaging.version import Version
 
 from subliminal.utils import sanitize, timestamp
 from subliminal.video import Episode, Movie, Video
@@ -209,3 +212,29 @@ def test_episode_fromname(episodes: dict[str, Episode]) -> None:
     assert video.title is None
     assert video.year is None
     assert video.tvdb_id is None
+
+
+@pytest.mark.skipif(
+    Version(get_version('babelfish')) <= Version('0.6.1'),
+    reason='babelfish.Language needs to be a hashable dataclass',
+)
+def test_serialize_movie(movies: dict[str, Movie]) -> None:
+    video = movies['man_of_steel']
+
+    ser = unstructure(video)
+    new_video = structure(ser, Movie)
+
+    assert video == new_video
+
+
+@pytest.mark.skipif(
+    Version(get_version('babelfish')) <= Version('0.6.1'),
+    reason='babelfish.Language needs to be a hashable dataclass',
+)
+def test_serialize_episode(episodes: dict[str, Episode]) -> None:
+    video = episodes['bbt_s07e05']
+
+    ser = unstructure(video)
+    new_video = structure(ser, Episode)
+
+    assert video == new_video