Skip to content

Commit 7dff42d

Browse files
committed
update test to use mock
1 parent a781555 commit 7dff42d

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

test-requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,3 +3,4 @@ pytest==2.8.7
33
pytest-cov==1.8.1
44
requests==2.5.1
55
responses==0.3.0
6+
mock==2.0.0

test/test_transmissions.py

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
import os
44
import tempfile
55
import warnings
6+
from mock import patch
67

78
import pytest
89
import responses
@@ -298,7 +299,8 @@ def test_fail_get():
298299

299300

300301
@responses.activate
301-
def test_success_list():
302+
@patch.object(warnings, 'warn')
303+
def test_success_list(mock_warn):
302304
responses.add(
303305
responses.GET,
304306
'https://api.sparkpost.com/api/v1/transmissions',
@@ -307,12 +309,9 @@ def test_success_list():
307309
body='{"results": []}'
308310
)
309311
sp = SparkPost('fake-key')
310-
with warnings.catch_warnings(record=True) as warns:
311-
warnings.simplefilter("always")
312-
response = sp.transmission.list()
313-
assert response == []
314-
assert len(warns) == 1
315-
assert "deprecated" in str(warns[-1].message)
312+
response = sp.transmission.list()
313+
assert mock_warn.called
314+
assert response == []
316315

317316

318317
@responses.activate

0 commit comments

Comments
 (0)