From 16b6fe3473dbaa2db45a29a5adff06c0dadb12df Mon Sep 17 00:00:00 2001
From: Laurent Monin <github@norz.org>
Date: Thu, 3 Jun 2021 12:41:48 +0200
Subject: [PATCH] Add tests for File._guess_tracknumber_and_title()

---
 test/test_file.py | 47 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)

diff --git a/test/test_file.py b/test/test_file.py
index 2ab62d9f8..cfea35636 100644
--- a/test/test_file.py
+++ b/test/test_file.py
@@ -290,3 +290,50 @@ class FileNamingTest(PicardTestCase):
         self.assertEqual(
             os.path.realpath('/media/music/_somealbum/_sometitle.mp3'),
             filename)
+
+
+class FileGuessTracknumberAndTitleTest(PicardTestCase):
+    def setUp(self):
+        super().setUp()
+        self.set_config_values({
+            'guess_tracknumber_and_title': True,
+        })
+
+    def test_no_guess(self):
+        f = File('/somepath/01 somefile.mp3')
+        metadata = Metadata({
+            'album': 'somealbum',
+            'title': 'sometitle',
+            'tracknumber': '2',
+        })
+        f._guess_tracknumber_and_title(metadata)
+        self.assertEqual(metadata['tracknumber'], '2')
+        self.assertEqual(metadata['title'], 'sometitle')
+
+    def test_guess_title(self):
+        f = File('/somepath/01 somefile.mp3')
+        metadata = Metadata({
+            'album': 'somealbum',
+            'tracknumber': '2',
+        })
+        f._guess_tracknumber_and_title(metadata)
+        self.assertEqual(metadata['tracknumber'], '2')
+        self.assertEqual(metadata['title'], 'somefile')
+
+    def test_guess_tracknumber(self):
+        f = File('/somepath/01 somefile.mp3')
+        metadata = Metadata({
+            'album': 'somealbum',
+            'title': 'sometitle',
+        })
+        f._guess_tracknumber_and_title(metadata)
+        self.assertEqual(metadata['tracknumber'], '1')
+
+    def test_guess_title_tracknumber(self):
+        f = File('/somepath/01 somefile.mp3')
+        metadata = Metadata({
+            'album': 'somealbum',
+        })
+        f._guess_tracknumber_and_title(metadata)
+        self.assertEqual(metadata['tracknumber'], '1')
+        self.assertEqual(metadata['title'], 'somefile')