- Add test for channel_info command;

- Add update channel_info file;
- fix test_base

Author: aninhasalesp

tests/commands/test_base.py

@@ -2,8 +2,6 @@
 import argparse
 import pytest
 
-from io import StringIO
-from datetime import datetime
 from pathlib import Path
 from unittest.mock import MagicMock, patch, mock_open
 from youtool.commands import Command
@@ -65,7 +63,7 @@ def test_data_from_csv_valid(mock_csv_file):
     with patch('pathlib.Path.is_file', return_value=True):
         with patch('builtins.open', mock_open(read_data=mock_csv_file)):
             data_column_name = "URL"
-            file_path = Path("tests/commands/csv_valid.csv")
+            file_path = Path("tests/resources/csv_valid.csv")
             result = Command.data_from_csv(file_path, data_column_name)
             assert len(result) == 2
             assert result[0] == "http://example.com"
@@ -80,10 +78,10 @@ def test_data_from_csv_file_not_found():
 def test_data_from_csv_column_not_found(mock_csv_file):
     with patch('pathlib.Path.is_file', return_value=True):
         with patch('builtins.open', mock_open(read_data=mock_csv_file)):
-            file_path = Path("tests/commands/csv_column_not_found.csv")
+            file_path = Path("tests/resources/csv_column_not_found.csv")
             with pytest.raises(Exception) as exc_info:
                 Command.data_from_csv(file_path, "NonExistentColumn")
-            assert "Column NonExistentColumn not found on tests/commands/csv_column_not_found.csv" in str(exc_info.value), "Exception message should contain column not found error"
+            assert "Column NonExistentColumn not found on tests/resources/csv_column_not_found.csv" in str(exc_info.value)
 
 
 @pytest.fixture
@@ -98,13 +96,13 @@ def test_data_to_csv_with_output_file_path(tmp_path, sample_data):
 
     result_path = Command.data_to_csv(sample_data, str(output_file_path))
 
-    assert result_path == str(output_file_path), "The returned path should match the provided output file path"
-    assert output_file_path.exists(), "The output file should exist"
+    assert result_path == str(output_file_path)
+    assert output_file_path.exists()
     with output_file_path.open('r') as f:
         reader = csv.DictReader(f)
         rows = list(reader)
-        assert len(rows) == 2, "There should be two rows in the output CSV"
-        assert rows[0]["id"] == "123" and rows[1]["id"] == "456", "The IDs should match the sample data"
+        assert len(rows) == 2
+        assert rows[0]["id"] == "123" and rows[1]["id"] == "456"
 
 def test_data_to_csv_without_output_file_path(sample_data):
     csv_content = Command.data_to_csv(sample_data)

tests/commands/test_channel_info.py

@@ -0,0 +1,53 @@
+import pytest
+
+from unittest.mock import patch, Mock, call
+
+from youtool.commands.channel_info import ChannelInfo, YouTube
+
+
+def test_filter_fields():
+    channel_info = {
+        'channel_id': '123456',
+        'channel_name': 'Test Channel',
+        'subscribers': 1000,
+        'videos': 50,
+        'category': 'Tech'
+    }
+    
+    info_columns = ['channel_id', 'channel_name', 'subscribers']
+    filtered_info = ChannelInfo.filter_fields(channel_info, info_columns)
+    
+    expected_result = {
+        'channel_id': '123456',
+        'channel_name': 'Test Channel',
+        'subscribers': 1000
+    }
+    
+    assert filtered_info == expected_result, f"Expected {expected_result}, but got {filtered_info}"
+
+
+def test_channel_ids_from_urls_and_usernames(mocker):
+    urls = ["https://www.youtube.com/@Turicas/featured", "https://www.youtube.com/c/PythonicCaf%C3%A9"]
+    usernames = ["Turicas", "PythonicCafe"]
+
+    ids_from_urls_mock = "id_from_url"
+    ids_from_usernames_mock = "id_from_username"
+    youtube_mock = mocker.patch("youtool.commands.channel_info.YouTube")
+
+    channel_id_from_url_mock = Mock(return_value=ids_from_urls_mock)
+    channel_id_from_username_mock = Mock(return_value=ids_from_usernames_mock)
+    channels_infos_mock = Mock(return_value=[])
+
+    youtube_mock.return_value.channel_id_from_url = channel_id_from_url_mock
+    youtube_mock.return_value.channel_id_from_username = channel_id_from_username_mock
+    youtube_mock.return_value.channels_infos = channels_infos_mock
+
+    ChannelInfo.execute(urls=urls, usernames=usernames)
+
+    channel_id_from_url_mock.assert_has_calls(
+        [call(url) for url in urls]
+    )
+    channel_id_from_username_mock.assert_has_calls(
+        [call(username) for username in usernames]
+    )
+    channels_infos_mock.assert_called_once_with([ids_from_urls_mock, ids_from_usernames_mock])

youtool/commands/channel_info.py

@@ -108,7 +108,9 @@ def execute(cls: Self, **kwargs) -> str:
         ] + [
             youtube.channel_id_from_username(username) for username in (usernames or []) if username
         ]
-        channel_ids = [channel_id for channel_id in channels_ids if channel_id]
+        channel_ids = list(
+            set([channel_id for channel_id in channels_ids if channel_id])
+        )
 
         return cls.data_to_csv(
             data=[
@@ -117,4 +119,4 @@ def execute(cls: Self, **kwargs) -> str:
                 ) for channel_info in (youtube.channels_infos(channel_ids) or [])
             ],
             output_file_path=output_file_path
-        )
+        )