test: update

Author: add-n2x

test/data/navidrome.db

@@ -1,12 +1,38 @@
-import pytest
+"""
+Tests for the Beets integration.
+"""
 
-from ndtoolbox.beets import BeetsClient
+import io
+import sys
+
+from beets.ui import main
+
+from ndtoolbox.beets import beets
 from ndtoolbox.config import config
 
 config.set_file("test/config/config.yaml")
 
 
+def test_beets_ui():
+    """Test Beets UI."""
+    captured_output = io.StringIO()
+    sys.stdout = captured_output
+    main(["-c", "config/beets/config.yaml", "stats"])
+    sys.stdout = sys.__stdout__
+    results = captured_output.getvalue()
+    captured_output.close()
+    print("Beets stats:" + results)
+
+
 def test_beets_client():
     """Test Beets client."""
-    client = BeetsClient()
-    assert client is not None
+    info = beets.get_album_info("/music/Calibre/Second Sun/")
+    assert info is not None
+    for i in info:
+        print("Subprocess Query - Beets album info: " + i)
+
+    # client = BeetsClient(1)
+    # info = client.get_album_info("/music/Calibre/Second Sun/")
+    # assert info is not None
+    # for i in info:
+    #     print("Stdio Capture - Beets album info: " + i)

test/test_beets.py

@@ -102,7 +102,7 @@ def test_album_folder_dirty_beets(subprocess_out, mocker):
 
     media = MediaFile(
         "1",
-        "/music/library/artist/album11/track.mp3",
+        "/music/library/artist/album1144/track.mp3",
         "title",
         2003,
         1,
@@ -113,10 +113,10 @@ def test_album_folder_dirty_beets(subprocess_out, mocker):
         "2",
         "album",
         "mbz3",
-        beets_path="/music/artist/album11/track.mp",
+        beets_path="/music/artist/album1144/track.mp",
     )
     folder = media.folder
-    assert folder.beets_path == "/music/artist/album11"
+    assert folder.beets_path == "/music/artist/album1144"
     assert folder.has_keepable is False
     assert folder.is_dirty is True
     assert folder.type == Folder.Type.UNKNOWN