feat(main): main

tests/test_source_rules.py

@@ -0,0 +1,116 @@
+import pytest
+from datetime import date, timedelta, datetime, timezone
+from django.utils import timezone as django_timezone
+
+from core.models import AppUser, Library, Channel, ScheduleTemplate, ScheduleBlock, MediaSource, MediaItem, ChannelSourceRule, Airing
+from core.services.scheduler import ScheduleGenerator
+
+@pytest.mark.django_db
+def test_repeat_gap_hours():
+    # Setup user, library, channel
+    user = AppUser.objects.create_user(username="rule_tester", password="pw")
+    library = Library.objects.create(owner_user=user, name="Rule Test Lib")
+    channel = Channel.objects.create(
+        owner_user=user, library=library, name="Rule TV", slug="r-tv", channel_number=99, timezone_name="UTC"
+    )
+
+    source = MediaSource.objects.create(
+        library=library,
+        name="Gap Source",
+        source_type="local_directory",
+        uri="/tmp/fake",
+        min_repeat_gap_hours=5
+    )
+    
+    # Create 6 items (each 1 hour long) -> 6 hours total duration
+    # This means a 5 hour gap is mathematically satisfiable without breaking cooldowns.
+    for i in range(6):
+        MediaItem.objects.create(
+            media_source=source,
+            title=f"Vid {i}",
+            item_kind="movie",
+            runtime_seconds=3600, # 1 hour
+            file_path=f"/tmp/fake/{i}.mp4",
+            cached_file_path=f"/tmp/fake/{i}.mp4",
+            is_active=True
+        )
+
+    ChannelSourceRule.objects.create(channel=channel, media_source=source, rule_mode="allow", weight=1.0)
+    template = ScheduleTemplate.objects.create(channel=channel, name="T", priority=10, is_active=True)
+    block = ScheduleBlock.objects.create(
+        schedule_template=template, name="All Day", block_type="programming",
+        start_local_time="00:00:00", end_local_time="12:00:00", day_of_week_mask=127
+    )
+    
+    target_date = django_timezone.now().date()
+    gen = ScheduleGenerator(channel)
+    created = gen.generate_for_date(target_date)
+    
+    assert created == 12  # 12 1-hour slots
+    
+    airings = list(Airing.objects.filter(channel=channel).order_by('starts_at'))
+    
+    # Assert that NO item is repeated within 5 hours of itself
+    last_played = {}
+    for a in airings:
+        if a.media_item_id in last_played:
+            gap = (a.starts_at - last_played[a.media_item_id]).total_seconds() / 3600.0
+            assert gap >= 5.0, f"Item {a.media_item_id} played too soon. Gap: {gap} hours"
+        last_played[a.media_item_id] = a.starts_at
+
+@pytest.mark.django_db
+def test_youtube_age_filter():
+    from unittest.mock import patch
+    from core.services.youtube import sync_source
+    user = AppUser.objects.create_user(username="rule_tester2", password="pw")
+    library = Library.objects.create(owner_user=user, name="Rule Test Lib 2")
+
+    source = MediaSource.objects.create(
+        library=library,
+        name="Old Videos Only",
+        source_type="youtube_playlist",
+        uri="https://www.youtube.com/playlist?list=fake_playlist",
+        max_age_days=10 # Anything older than 10 days is skipped
+    )
+    
+    # Mock yt-dlp extract_info to return 3 videos
+    # 1. 2 days old (keep)
+    # 2. 15 days old (skip)
+    # 3. 5 days old (keep)
+    from datetime import date, timedelta
+    today = date.today()
+    
+    with patch("core.services.youtube.yt_dlp.YoutubeDL") as mock_ydl:
+        mock_instance = mock_ydl.return_value.__enter__.return_value
+        mock_instance.extract_info.return_value = {
+            "entries": [
+                {
+                    "id": "vid1",
+                    "title": "New Video",
+                    "duration": 600,
+                    "upload_date": (today - timedelta(days=2)).strftime("%Y%m%d")
+                },
+                {
+                    "id": "vid2",
+                    "title": "Old Video",
+                    "duration": 600,
+                    "upload_date": (today - timedelta(days=15)).strftime("%Y%m%d")
+                },
+                {
+                    "id": "vid3",
+                    "title": "Medium Video",
+                    "duration": 600,
+                    "upload_date": (today - timedelta(days=5)).strftime("%Y%m%d")
+                }
+            ]
+        }
+        
+        res = sync_source(source, max_videos=5)
+        
+        assert res['created'] == 2 # Only vid1 and vid3
+        assert res['skipped'] == 1 # vid2 skipped due to age
+        
+        # Verify the items in DB
+        items = list(MediaItem.objects.filter(media_source=source).order_by('title'))
+        assert len(items) == 2
+        assert "vid2" not in [item.title for item in items]