File tree Expand file tree Collapse file tree 2 files changed +7
-7
lines changed Expand file tree Collapse file tree 2 files changed +7
-7
lines changed Original file line number Diff line number Diff line change @@ -3,3 +3,4 @@ pytest==2.8.7
33pytest-cov==1.8.1
44requests==2.5.1
55responses==0.3.0
6+ mock==2.0.0
Original file line number Diff line number Diff line change 33import os
44import tempfile
55import warnings
6+ from mock import patch
67
78import pytest
89import 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
You can’t perform that action at this time.
0 commit comments