def test_get_geo_ip_url_error(Client_mock):
Client_mock.get_soap_client_service.return_value = service = MagicMock()
service.GetGeoIP.return_value = geo_ip = Mock()
geo_ip.CountryName = 'Country Name'
print service, Client_mock
result = get_geo_ip('1.2.3.4')
Client_mock.get_soap_client_service.assert_called_once()
service.GetGeoIP.assert_called_once_with('1.2.3.4')
assert 'Country Name' == result
def test_get_geo_ip_url_error(Client_mock):
Client_mock.get_soap_client_service.return_value = service = MagicMock()
service.GetGeoIP.return_value = geo_ip = Mock()
geo_ip.CountryName = 'Country Name'
print service, Client_mock
result = get_geo_ip('1.2.3.4')
Client_mock.get_soap_client_service.assert_called_once()
service.GetGeoIP.assert_called_once_with('1.2.3.4')
assert 'Country Name' == result
from lib.geoipserverice import get_geo_ip
import logging
__author__ = 'Hossein Zolfi <*****@*****.**>'
do_log = False
if do_log:
logging.basicConfig(level=logging.INFO)
logging.getLogger('suds.client').setLevel(logging.DEBUG)
print get_geo_ip('4.2.2.4')
def test_get_geo_ip_timeout(Client_mock):
Client_mock.get_soap_client_service.side_effect = timeout
result = get_geo_ip('1.2.3.4')
Client_mock.get_soap_client_service.assert_called_once()
assert None == result
def test_get_geo_ip_url_error(Client_mock):
Client_mock.get_soap_client_service.side_effect = URLError('')
result = get_geo_ip('1.2.3.4')
Client_mock.get_soap_client_service.assert_called_once()
assert None == result
def test_get_geo_ip_timeout(Client_mock):
Client_mock.get_soap_client_service.side_effect = timeout
result = get_geo_ip('1.2.3.4')
Client_mock.get_soap_client_service.assert_called_once()
assert None == result
def test_get_geo_ip_url_error(Client_mock):
Client_mock.get_soap_client_service.side_effect = URLError('')
result = get_geo_ip('1.2.3.4')
Client_mock.get_soap_client_service.assert_called_once()
assert None == result
from lib.geoipserverice import get_geo_ip
import logging
__author__ = 'Hossein Zolfi <*****@*****.**>'
do_log = False
if do_log:
logging.basicConfig(level=logging.INFO)
logging.getLogger('suds.client').setLevel(logging.DEBUG)
print get_geo_ip('4.2.2.4')
