correctly mock packaged requests library
This commit is contained in:
parent
8e8c0154d6
commit
d36592e1c5
1 changed files with 7 additions and 3 deletions
|
@ -3,6 +3,7 @@
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
import unittest
|
import unittest
|
||||||
|
from wakatime.packages.requests.models import Response
|
||||||
|
|
||||||
try:
|
try:
|
||||||
from mock import patch
|
from mock import patch
|
||||||
|
@ -12,20 +13,23 @@ except:
|
||||||
from wakatime.base import main
|
from wakatime.base import main
|
||||||
|
|
||||||
|
|
||||||
@patch('requests.post')
|
@patch('wakatime.packages.requests.adapters.HTTPAdapter.send')
|
||||||
class BaseTestCase(unittest.TestCase):
|
class BaseTestCase(unittest.TestCase):
|
||||||
|
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
# disable logging while testing
|
# disable logging while testing
|
||||||
logging.disable(logging.CRITICAL)
|
logging.disable(logging.CRITICAL)
|
||||||
|
|
||||||
def test_help_contents(self, mock_post):
|
def test_help_contents(self, mock_requests):
|
||||||
with self.assertRaises(SystemExit):
|
with self.assertRaises(SystemExit):
|
||||||
args = ['', '--help']
|
args = ['', '--help']
|
||||||
retval = main(args)
|
retval = main(args)
|
||||||
self.assertEquals(retval, 0)
|
self.assertEquals(retval, 0)
|
||||||
|
|
||||||
def test_argument_parsing(self, mock_post):
|
def test_argument_parsing(self, mock_requests):
|
||||||
|
response = Response()
|
||||||
|
response.status_code = 201
|
||||||
|
mock_requests.return_value = response
|
||||||
args = ['', '--file', 'tests/samples/emptyfile.txt', '--key', '123']
|
args = ['', '--file', 'tests/samples/emptyfile.txt', '--key', '123']
|
||||||
retval = main(args)
|
retval = main(args)
|
||||||
self.assertEquals(retval, 0)
|
self.assertEquals(retval, 0)
|
||||||
|
|
Loading…
Reference in a new issue