Updates tests to use pytest-mock.

This commit is contained in:
erinhmclark
2025-02-18 23:32:03 +00:00
parent 657fbd357d
commit f0fd9bf445
19 changed files with 584 additions and 600 deletions

View File

@@ -1,7 +1,6 @@
import hashlib
import json
from datetime import datetime, timezone
from unittest.mock import Mock, patch
import pytest
@@ -44,20 +43,19 @@ class TestURLExpansion:
("https://example.com", "https://example.com"),
("https://t.co/test", "https://expanded.url")
])
def test_expand_url(self, input_url, expected):
mock_response = Mock()
def test_expand_url(self, input_url, expected, mocker):
mock_response = mocker.Mock()
mock_response.url = "https://expanded.url"
with patch('requests.get', return_value=mock_response):
mocker.patch('requests.get', return_value=mock_response)
result = expand_url(input_url)
assert result == expected
result = expand_url(input_url)
assert result == expected
def test_expand_url_handles_errors(self, caplog):
with patch('requests.get', side_effect=Exception("Connection error")):
url = "https://t.co/error"
result = expand_url(url)
assert result == url
assert f"Failed to expand url {url}" in caplog.text
def test_expand_url_handles_errors(self, caplog, mocker):
mocker.patch('requests.get', side_effect=Exception("Connection error"))
url = "https://t.co/error"
result = expand_url(url)
assert result == url
assert f"Failed to expand url {url}" in caplog.text
class TestAttributeHandling:
class Sample: