From 193a7869efcc8390796283006e6ebcf8dd49a718 Mon Sep 17 00:00:00 2001
From: moonshinerd <vitorhls15@gmail.com>
Date: Sun, 8 Jun 2025 09:43:10 -0300
Subject: [PATCH] =?UTF-8?q?[test:update]=20Add=20unit=20tests=20for=20=5Fg?=
 =?UTF-8?q?et=5Fversion=5Finfo=20(CT1=E2=80=93CT5)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 test/test_update.py | 57 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 57 insertions(+)

diff --git a/test/test_update.py b/test/test_update.py
index 23c12d38c..a3d70e9ee 100644
--- a/test/test_update.py
+++ b/test/test_update.py
@@ -272,6 +272,63 @@ def test(target, expected, current_version=None, current_commit=None, identifier
         test('testing', None, current_commit='9' * 40)
         test('testing', UpdateInfo('testing', commit='9' * 40))
 
+    def test_get_version_info(self):
+        updater = FakeUpdater(FakeYDL(), 'stable')
+
+        # CT1 - Tag Already in Required Format
+        updater._identifier = 'zip'
+        updater.requested_repo = 'yt-dlp/yt-dlp'
+        version, commit = updater._get_version_info('2025.04.06', {})
+        self.assertEqual(version, '2025.04.06')
+        self.assertIsNone(commit)
+
+        # CT2 - Fetch Latest Release Without Commit Hash
+        tag = 'latest'
+        updater.requested_repo = 'yt-dlp/yt-dlp'
+        updater._call_api = lambda _: {
+            'tag_name': 'v2025.04.06',
+            'name': 'Release 2025.04.06',
+            'target_commitish': '',
+            'body': '- Adds support for the new URL parser\n- Fixes bug in simultaneous downloads\n',
+        }
+        version, commit = updater._get_version_info(tag, updater._call_api(tag))
+        self.assertEqual(version, 'v2025.04.06')
+        self.assertIsNone(commit)
+
+        # CT3 - Fallback to Commit Hash When Tag Is Empty
+        updater._call_api = lambda _: {
+            'tag_name': 'v2024.02.01',
+            'name': 'Release candidate',
+            'target_commitish': 'a1b2c3d4e5f60718293a4b5c6d7e8f9012ab3cd4',
+            'body': '- Implements extra parameter validation\n- Optimizes performance in date parser\n',
+        }
+        version, commit = updater._get_version_info('', updater._call_api(''))
+        self.assertIsNone(version)
+        self.assertEqual(commit, 'a1b2c3d4e5f60718293a4b5c6d7e8f9012ab3cd4')
+
+        # CT4 - Extract Version from Name with Empty Tag and Body Hash
+        updater._call_api = lambda _: {
+            'tag_name': 'v2024.02.01',
+            'name': 'Release 2024.02.01',
+            'target_commitish': '',
+            'body': 'a1b2c3d4e5f60718293a4b5c6d7e8f9012ab3cd4\n- Implements extra parameter validation\n- Optimizes performance in date parser\n',
+        }
+        version, commit = updater._get_version_info('', updater._call_api(''))
+        self.assertEqual(version, '2024.02.01')
+        self.assertIsNone(commit)
+
+        # CT5 - Error on Missing Version and Commit Hash
+        updater._call_api = lambda _: {
+            'tag_name': 'v2024.02.01',
+            'name': 'Release candidate',
+            'target_commitish': '',
+            'body': '- Implements extra parameter validation\n- Optimizes performance in date parser\n',
+        }
+        with self.assertLogs(level='WARNING') as cm:
+            version, commit = updater._get_version_info('', updater._call_api(''))
+            self.assertIsNone(version)
+            self.assertIsNone(commit)
+            self.assertTrue(any('One of either version or commit hash must be available' in m for m in cm.output))
 
 if __name__ == '__main__':
     unittest.main()