Added more testing for ApiRequest using mocks

This commit is contained in:
Rudi Grinberg 2012-10-29 23:34:26 -04:00
parent ccb4925779
commit a33a1e6e2a
1 changed files with 15 additions and 1 deletions

View File

@ -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()