Issue #140: kotatogram's upload speed boost big.

tests/test_client/test_telegram_upload_client.py

@@ -2,7 +2,9 @@
 import os
 import sys
 import unittest
-from unittest.mock import patch, mock_open, Mock, MagicMock, call
+from unittest.mock import patch, mock_open, Mock, MagicMock, call, AsyncMock
+
+from telethon import types
 
 from telegram_upload.client.telegram_upload_client import TelegramUploadClient
 from telegram_upload.exceptions import TelegramUploadDataLoss, MissingFileError
@@ -19,7 +21,7 @@ def __eq__(a, b):
         return True
 
 
-class TestTelegramUploadClient(unittest.TestCase):
+class TestTelegramUploadClient(unittest.IsolatedAsyncioTestCase):
     @patch('builtins.open', mock_open(read_data=json.dumps(CONFIG_DATA)))
     @patch('telegram_upload.client.telegram_upload_client.TelegramClient.__init__', return_value=None)
     def setUp(self, m1) -> None:
@@ -38,6 +40,20 @@ def test_forward_to(self, mock_forward_messages: MagicMock):
             call(mock_destinations[1], [mock_message]),
         ])
 
+    async def test_send_album_media(self):
+        self.client.get_input_entity = AsyncMock()
+        self.client._call = AsyncMock()
+        self.client._sender = MagicMock()
+        self.client._get_response_message = MagicMock()
+        entity = "entity"
+        mock_media = [MagicMock(), MagicMock()]
+        response_message = await self.client._send_album_media(entity, mock_media)
+        self.assertEqual(self.client._get_response_message.return_value, response_message)
+        self.client._get_response_message.assert_called_once_with(
+            [m.random_id for m in mock_media], self.client._call.return_value,
+            self.client.get_input_entity.return_value,
+        )
+
     @patch('telegram_upload.client.telegram_upload_client.TelegramUploadClient.send_files')
     @patch('telegram_upload.client.telegram_upload_client.TelegramUploadClient._send_album_media')
     @unittest.skipIf(sys.version_info < (3, 8), "TypeError: An asyncio.Future, a coroutine or an awaitable is required")
@@ -70,3 +86,27 @@ def test_send_files_data_loss(self):
         self.client.send_file.return_value.media.document.size = 200
         with self.assertRaises(TelegramUploadDataLoss):
             self.client.send_files('foo', [file])
+
+    @patch('telegram_upload.client.telegram_upload_client.utils')
+    async def test_send_media(self, mock_utils: MagicMock):
+        mock_utils.get_appropriated_part_size.return_value = 512
+        self.client.get_input_entity = AsyncMock()
+        self.client._log = AsyncMock()
+        self.client._call = AsyncMock()
+        self.client._sender = MagicMock()
+        entity = 'entity'
+        mock_progress = MagicMock()
+        file = File(self.upload_file_path)
+        with patch('telegram_upload.client.telegram_upload_client.isinstance', return_value=True), \
+                self.subTest("Test photo"):
+            await self.client._send_media(entity, file, mock_progress)
+        isinstance_result = {types.InputMediaUploadedPhoto: False, types.InputMediaUploadedDocument: True}
+        with patch('telegram_upload.client.telegram_upload_client.isinstance',
+                   side_effect=lambda obj, target: isinstance_result.get(target, isinstance(obj, target))), \
+                self.subTest("Test Document"):
+            await self.client._send_media(entity, file, mock_progress)
+        # self.client.send_file.assert_called_once_with(
+        #     entity, file, thumb=None, file_size=file.file_size,
+        #     caption=os.path.basename(self.upload_file_path).split('.')[0], force_document=False,
+        #     progress_callback=AnyArg(), attributes=[],
+        # )