light refact

src/tests/crawler_test.py

@@ -41,24 +41,13 @@ def setUp(self):
 
         def _api_req(method, url, **kwargs):
             if url.startswith('feed') and len(url) == 6:
-                class Proxy:
-                    status_code = self.resp_status_code
-                    headers = self.resp_headers
-                    history = []
-
-                    @classmethod
-                    def raise_for_status(cls):
-                        if self.resp_raise is None:
-                            return
-                        raise self.resp_raise
-
-                    @property
-                    def content(self):
-                        with open(atom_file_path) as f:
-                            return f.read()
-
-                    text = content
-                return Proxy()
+                with open(atom_file_path) as f:
+                    content = f.read()
+                    resp = Mock(status_code=self.resp_status_code,
+                                headers=self.resp_headers,
+                                content=content, text=content, history=[])
+                resp.raise_for_status.return_value = self.resp_raise
+                return resp
 
             url = url.split(conf.API_ROOT)[1].strip('/')
             kwargs.pop('allow_redirects', None)