33 lines
937 B
Python
33 lines
937 B
Python
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
|