pjht/buku
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

chg: test: simplify test_get_PoolManager

Browse Source
This commit is contained in:
rachmadaniHaryono 2018-08-06 15:51:36 +08:00
parent 9d2b414384
commit f823a6aecc
1 changed files with 2 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/test_buku.py
View File

@ -67,18 +67,10 @@ def test_gen_headers():
@pytest.mark.parametrize('m_myproxy', [None, mock.Mock()])
def test_get_PoolManager(m_myproxy):
"""test func."""
with mock.patch('buku.urllib3') as m_ul3:
with mock.patch('buku.urllib3'):
import buku
buku.myproxy = m_myproxy
res = buku.get_PoolManager()
if m_myproxy:
m_ul3.ProxyManager.assert_called_once_with(
m_myproxy, num_pools=1, timeout=15, headers=buku.myheaders)
assert res == m_ul3.ProxyManager.return_value
else:
m_ul3.PoolManager.assert_called_once_with(
num_pools=1, timeout=15, headers=buku.myheaders)
assert res == m_ul3.PoolManager.return_value
assert buku.get_PoolManager()
@pytest.mark.parametrize(