From 8cfcfff83ab8b1ee596066e1d71848cfd07c5342 Mon Sep 17 00:00:00 2001
From: Dr-Blank <64108942+Dr-Blank@users.noreply.github.com>
Date: Tue, 10 Oct 2023 02:34:31 -0400
Subject: [PATCH] add tests for models

---
 tests/test_api_models.py | 110 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 110 insertions(+)
 create mode 100644 tests/test_api_models.py

diff --git a/tests/test_api_models.py b/tests/test_api_models.py
new file mode 100644
index 0000000..a2a5d37
--- /dev/null
+++ b/tests/test_api_models.py
@@ -0,0 +1,110 @@
+""" Tests for the API models. """
+
+from datetime import datetime
+
+from lrclib.models import (
+    Lyrics,
+    LyricsMinimal,
+    ErrorResponse,
+    SearchResult,
+    CryptographicChallenge,
+)
+
+minimized_lyrics = {
+    "id": 123,
+    "name": "test_name",
+    "trackName": "test_track_name",
+    "artistName": "test_artist_name",
+    "albumName": "test_album_name",
+    "duration": 180,
+    "instrumental": False,
+    "plainLyrics": "test_plain_lyrics",
+    "syncedLyrics": "test_synced_lyrics",
+}
+
+full_lyrics = {
+    **minimized_lyrics,
+    "lang": "test_lang",
+    "isrc": "test_isrc",
+    "spotifyId": "test_spotify_id",
+    "releaseDate": "2023-08-10T00:00:00Z",
+}
+
+sample_search_result = [minimized_lyrics, minimized_lyrics]
+
+
+def test_lyrics_minimal_from_dict() -> None:
+    """Test the LyricsMinimal.from_dict method"""
+    # Create a LyricsMinimal object from a dictionary
+    lyrics = LyricsMinimal.from_dict(minimized_lyrics)
+
+    # Check that the LyricsMinimal object has the expected type
+    assert isinstance(lyrics, LyricsMinimal)
+
+    # Check that the LyricsMinimal object has the expected attributes
+    assert lyrics.id == minimized_lyrics["id"]
+    assert lyrics.name == minimized_lyrics["name"]
+    assert lyrics.track_name == minimized_lyrics["trackName"]
+    assert lyrics.artist_name == minimized_lyrics["artistName"]
+    assert lyrics.album_name == minimized_lyrics["albumName"]
+    assert lyrics.duration == minimized_lyrics["duration"]
+    assert lyrics.instrumental == minimized_lyrics["instrumental"]
+    assert lyrics.plain_lyrics == minimized_lyrics["plainLyrics"]
+    assert lyrics.synced_lyrics == minimized_lyrics["syncedLyrics"]
+
+
+def test_lyrics_from_dict() -> None:
+    lyrics = Lyrics.from_dict(full_lyrics)
+
+    assert isinstance(lyrics, Lyrics)
+
+    assert lyrics.id == full_lyrics["id"]
+    assert lyrics.name == full_lyrics["name"]
+    assert lyrics.track_name == full_lyrics["trackName"]
+    assert lyrics.artist_name == full_lyrics["artistName"]
+    assert lyrics.album_name == full_lyrics["albumName"]
+    assert lyrics.duration == full_lyrics["duration"]
+    assert lyrics.instrumental == full_lyrics["instrumental"]
+    assert lyrics.plain_lyrics == full_lyrics["plainLyrics"]
+    assert lyrics.synced_lyrics == full_lyrics["syncedLyrics"]
+    assert lyrics.lang == full_lyrics["lang"]
+    assert lyrics.isrc == full_lyrics["isrc"]
+    assert lyrics.spotify_id == full_lyrics["spotifyId"]
+    assert lyrics.release_date == datetime.fromisoformat(
+        full_lyrics["releaseDate"]
+    )
+
+    assert isinstance(lyrics.release_date, datetime)
+
+
+def test_error_response_from_dict() -> None:
+    error_response = ErrorResponse.from_dict(
+        {"error": "test_error", "message": "test_message"}
+    )
+
+    assert isinstance(error_response, ErrorResponse)
+
+    assert error_response.error == "test_error"
+    assert error_response.message == "test_message"
+
+
+def test_search_result_from_dict() -> None:
+    search_result = SearchResult.from_list(sample_search_result)
+
+    assert isinstance(search_result, SearchResult)
+
+    assert len(search_result) == len(sample_search_result)
+
+    for lyrics in search_result:
+        assert isinstance(lyrics, LyricsMinimal)
+
+
+def test_cryptographic_challenge_from_dict() -> None:
+    challenge = CryptographicChallenge.from_dict(
+        {"prefix": "test_prefix", "target": "test_target"}
+    )
+
+    assert isinstance(challenge, CryptographicChallenge)
+
+    assert challenge.prefix == "test_prefix"
+    assert challenge.target == "test_target"