GAE/SE/pyでfetchurlを使うメソッドのテスト

gae_fetchurl_test.py

import unittest
import json

from mock import Mock, patch

from google.appengine.ext import testbed


class FetchTestCase(unittest.TestCase):
    def setUp(self):
        self.testbed = testbed.Testbed()
        self.testbed.activate()
        self.testbed.init_urlfetch_stub()
        uf = self.testbed.get_stub('urlfetch')
        uf._Dynamic_Fetch = Mock()

    def tearDown(self):
        self.testbed.deactivate()
 
 class TestSomeAPI(FetchTestCase):

    @patch(
        'google.appengine.api.urlfetch.urlfetch_service_pb.URLFetchResponse')
    def test_some_api(self, URLFetchResponse):
        response = URLFetchResponse.return_value
        response.contentwastruncated.return_value = False
        response.statuscode.return_value = 200
        expected = {"result": 1}
        response.content.return_value = json.dumps(expected)
        with patch.dict('os.environ', {'API_URL': 'localhost/'}):
            results = some_api("hoge", 1234, "01-001", 2980)
            self.assertEquals(results, expected_some_api_return_value)