melbic · August 29, 2015 14:27 · May 11, 2011 · May 11, 2011
diff --git a/tests.py b/tests.py
@@ -2,6 +2,7 @@
 
 from django.core.urlresolvers import reverse
 from django.test import TestCase
+from django.utils import simplejson as json
 
 from MockResponse import MockResponse
 

diff --git a/MockResponse.py b/MockResponse.py
@@ -0,0 +1,13 @@
+class MockResponse(object):
+
+    def __init__(self, resp_data, code=200, msg='OK'):
+        self.resp_data = resp_data
+        self.code = code
+        self.msg = msg
+        self.headers = {'content-type': 'text/plain; charset=utf-8'}
+
+    def read(self):
+        return self.resp_data
+
+    def getcode(self):
+        return self.code
diff --git a/tests.py b/tests.py
@@ -0,0 +1,35 @@
+from mock import patch
+
+from django.core.urlresolvers import reverse
+from django.test import TestCase
+
+from MockResponse import MockResponse
+
+class MyTest(TestCase):
+
+    def setUp(self):
+        "Mock urllib2.urlopen"
+        self.patcher = patch('urllib2.urlopen')
+        self.urlopen_mock = self.patcher.start()
+
+    def test_one(self):
+        """
+        Assuming that url-name view actually use urllib2.open to
+        retrieve data from another server:
+        json.loads(urllib2.urlopen(url, data).read())
+        """
+        ret = {
+            'username': 'username',
+            'password': 'password',
+            ...
+        }
+
+        self.urlopen_mock.return_value =  MockResponse(json.dumps(ret))
+
+        data = {'username': 'your-username'}
+        response = self.client.post(reverse("url-name"), data)
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(response.content, 'returned-data')
+
+    def tearDown(self):
+        self.patcher.stop()
No results found