1 files changed, 122 insertions, 0 deletions
diff --git a/tests/test_subscleaner.py b/tests/test_subscleaner.py
new file mode 100644
index 0000000..73e7453
--- /dev/null
+++ b/tests/test_subscleaner.py
@@ -0,0 +1,122 @@
+"""Unit tests for the subscleaner module."""
+
+from io import StringIO
+from unittest.mock import patch
+
+import pytest
+
+from src.subscleaner.subscleaner import ads_in_line, main, process_file
+
+
+@pytest.fixture
+def sample_srt_content():
+    """Return a sample SRT content."""
+    return """1
+00:00:01,000 --> 00:00:03,000
+This is a sample subtitle.
+
+2
+00:00:04,000 --> 00:00:06,000
+OpenSubtitles
+
+3
+00:00:07,000 --> 00:00:09,000
+Another sample subtitle.
+"""
+
+
+def create_sample_srt_file(tmpdir, content):
+    """Create a sample SRT file with the given content."""
+    file_path = tmpdir.join("sample.srt")
+    file_path.write(content)
+    return str(file_path)
+
+
+@pytest.mark.parametrize(
+    "line, expected",
+    [
+        ("This is a normal line", False),
+        ("This line contains OpenSubtitles", True),
+        ("Subtitles by XYZ", True),
+    ],
+)
+def test_ads_in_line(line, expected):
+    """
+    Test the ads_in_line function with different input lines and expected results.
+
+    Args:
+        line (str): The input line to be tested.
+        expected (bool): The expected result (True if the line contains an ad, False otherwise).
+    """
+    assert ads_in_line(line) is expected
+
+
+def test_process_file_no_modification(tmpdir, sample_srt_content):
+    """
+    Test the process_file function when the file does not require modification.
+
+    Args:
+        tmpdir (pytest.fixture): A temporary directory for creating the sample SRT file.
+        sample_srt_content (str): The sample SRT content.
+    """
+    file_path = create_sample_srt_file(tmpdir, sample_srt_content)
+    with patch("src.subscleaner.subscleaner.os.path.getctime", return_value=0):
+        assert process_file(file_path) is False
+
+
+def test_process_file_with_modification(tmpdir, sample_srt_content):
+    """
+    Test the process_file function when the file requires modification.
+
+    Args:
+        tmpdir (pytest.fixture): A temporary directory for creating the sample SRT file.
+        sample_srt_content (str): The sample SRT content.
+    """
+    file_path = create_sample_srt_file(tmpdir, sample_srt_content)
+    with patch("src.subscleaner.subscleaner.os.path.getctime", return_value=9999999999):
+        assert process_file(file_path) is True
+
+
+def test_process_file_error(tmpdir):
+    """
+    Test the process_file function when an error occurs (e.g., file not found).
+
+    Args:
+        tmpdir (pytest.fixture): A temporary directory.
+    """
+    file_path = tmpdir.join("nonexistent.srt")
+    assert process_file(str(file_path)) is False
+
+
+def test_main_no_modification(tmpdir, sample_srt_content):
+    """
+    Test the main function when no files require modification.
+
+    Args:
+        tmpdir (pytest.fixture): A temporary directory for creating the sample SRT file.
+        sample_srt_content (str): The sample SRT content.
+    """
+    file_path = create_sample_srt_file(tmpdir, sample_srt_content)
+    with (
+        patch("sys.stdin", StringIO(file_path)),
+        patch("src.subscleaner.subscleaner.process_file", return_value=False) as mock_process_file,
+    ):
+        main()
+        mock_process_file.assert_called_once_with(file_path)
+
+
+def test_main_with_modification(tmpdir, sample_srt_content):
+    """
+    Test the main function when files require modification.
+
+    Args:
+        tmpdir (pytest.fixture): A temporary directory for creating the sample SRT file.
+        sample_srt_content (str): The sample SRT content.
+    """
+    file_path = create_sample_srt_file(tmpdir, sample_srt_content)
+    with (
+        patch("sys.stdin", StringIO(file_path)),
+        patch("src.subscleaner.subscleaner.process_file", return_value=True) as mock_process_file,
+    ):
+        main()
+        mock_process_file.assert_called_once_with(file_path)