Back to Repositories

Testing Age Restriction Validation in youtube-dl

This test suite implements age restriction validation for video downloads in youtube-dl. It verifies that content with age restrictions is properly handled by checking download attempts with different age parameters across multiple video platforms.

Test Coverage Overview

The test suite provides comprehensive coverage of age-restricted content handling across different video platforms. Key areas tested include:

Age restriction validation for YouTube videos
Adult content verification for YouPorn platform
Edge cases with different age thresholds
Download attempt validation with valid and invalid age parameters

Implementation Analysis

The testing approach utilizes Python’s unittest framework with a helper function _download_restricted that simulates download attempts. The implementation features:

Parameterized test helper for reusable age restriction checks
JSON-based metadata validation
Clean test file management with automatic cleanup
Platform-specific test cases with different age thresholds

Technical Details

The test suite leverages:

YoutubeDL class for download operations
Custom helper functions for file management
unittest assertion methods
JSON metadata extraction
Platform-specific URL handling
Temporary file cleanup mechanisms

Best Practices Demonstrated

The test implementation showcases several testing best practices:

Isolated test cases for different platforms
Proper cleanup of test artifacts
Reusable test helpers
Clear assertion messages
Efficient test setup and teardown
Platform-independent file path handling

ytdl-org/youtube-dl

test/test_age_restriction.py

            
#!/usr/bin/env python
from __future__ import unicode_literals

# Allow direct execution
import os
import sys
import unittest
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from test.helper import try_rm


from youtube_dl import YoutubeDL
from youtube_dl.utils import DownloadError


def _download_restricted(url, filename, age):
    """ Returns true if the file has been downloaded """

    params = {
        'age_limit': age,
        'skip_download': True,
        'writeinfojson': True,
        'outtmpl': '%(id)s.%(ext)s',
    }
    ydl = YoutubeDL(params)
    ydl.add_default_info_extractors()
    json_filename = os.path.splitext(filename)[0] + '.info.json'
    try_rm(json_filename)
    try:
        ydl.download([url])
    except DownloadError:
        try_rm(json_filename)
    res = os.path.exists(json_filename)
    try_rm(json_filename)
    return res


class TestAgeRestriction(unittest.TestCase):
    def _assert_restricted(self, url, filename, age, old_age=None):
        self.assertTrue(_download_restricted(url, filename, old_age))
        self.assertFalse(_download_restricted(url, filename, age))

    def test_youtube(self):
        self._assert_restricted('HtVdAasjOgU', 'HtVdAasjOgU.mp4', 10)

    def test_youporn(self):
        self._assert_restricted(
            'https://www.youporn.com/watch/16715086/sex-ed-in-detention-18-asmr/',
            '16715086.mp4', 2, old_age=25)


if __name__ == '__main__':
    unittest.main()