import pytest from unittest.mock import patch, MagicMock from src.core.media_uploader import MediaUploader import os def test_upload_success(tmp_path): # Create dummy file f = tmp_path / "test.jpg" f.write_text("dummy image data") uploader = MediaUploader() with patch("src.core.media_uploader.requests.put") as mock_put: mock_put.return_value.status_code = 200 result = uploader.upload_file("http://mock-upload-url.com", str(f)) assert result is True mock_put.assert_called_once() def test_upload_failure(tmp_path): f = tmp_path / "test.jpg" f.write_text("dummy") uploader = MediaUploader() with patch("src.core.media_uploader.requests.put") as mock_put: mock_put.return_value.status_code = 403 result = uploader.upload_file("http://mock-upload-url.com", str(f)) assert result is False