def next(self,itemType): result = None try: req = GetRequest(self.url) req.query("status", "QUEUED").query("type", itemType) req.header('Authorization', self.api_auth) resp = req.open() obj = json.load(resp) if len(obj["objects"]) > 0: result = obj["objects"][0] except urllib2.HTTPError, e: #Got an error None
def test_get_request(self): req = GetRequest("http://example.com/") req.header("Test", "test") real = req._request() self.assertEqual(real.get_method(),"GET") self.assertEqual(real.get_header("Test"),"test")