From a33a1e6e2abe57ba2d0e8879cdd61fc847ceca7e Mon Sep 17 00:00:00 2001 From: Rudi Grinberg Date: Mon, 29 Oct 2012 23:34:26 -0400 Subject: [PATCH] Added more testing for ApiRequest using mocks --- python_apps/api_clients/tests/test_apirequest.py | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/python_apps/api_clients/tests/test_apirequest.py b/python_apps/api_clients/tests/test_apirequest.py index a7b58b8a7..bde16ea5a 100644 --- a/python_apps/api_clients/tests/test_apirequest.py +++ b/python_apps/api_clients/tests/test_apirequest.py @@ -1,7 +1,21 @@ import unittest +from mock import MagicMock, patch from .. api_client import ApcUrl, ApiRequest class TestApiRequest(unittest.TestCase): def test_init(self): - u = ApiRequest("request_name", ApcUrl("/test/ing")) + u = ApiRequest('request_name', ApcUrl('/test/ing')) self.assertEquals(u.name, "request_name") + + def test_call(self): + ret = 'ok' + read = MagicMock() + read.read = MagicMock(return_value=ret) + u = '/testing' + with patch('urllib2.urlopen') as mock_method: + mock_method.return_value = read + request = ApiRequest('mm', ApcUrl(u))() + self.assertEquals(request, ret) + mock_method.assert_called_once_with(u) + +if __name__ == '__main__': unittest.main()