Python Response.status_code方法代码示例

# 需要导入模块: from requests import Response [as 别名]
# 或者: from requests.Response import status_code [as 别名]
def responses(code, path=None, redirection=None, data=None,
              url=None,
              headers=None):
    if headers is None:
        headers = {'Content-Type': 'text/xml'}
    response = Response()
    response.status_code = code
    if path is not None and redirection is None:
        with open(data_file(path), 'rb') as f:
            response.raw = BytesIO(f.read())
    elif data is not None:
        response._content = data.encode('utf-8')
    if redirection is not None:
        temp = Response()
        temp.status_code = 301 if 'permanent' in redirection else 302
        temp.url = path
        response.history.append(temp)
        response.url = redirection
        headers['location'] = path
    if url is None:
        if redirection is not None:
            url = redirection
        else:
            url = 'https://example.com/{}'.format(str(uuid4()))
    response.url = url
    response.headers = headers
    return response

# 需要导入模块: from requests import Response [as 别名]
# 或者: from requests.Response import status_code [as 别名]
    def test_upload_app_ignore_non_ok_responses(self):
        mock_response = self.mock_post.return_value
        mock_response.ok = True
        mock_response.return_value = {
            'success': True,
            'status_url': 'http://example.com/status/',
        }

        ok_response = Response()
        ok_response.status_code = 200
        ok_response.encoding = 'utf-8'
        ok_response._content = json.dumps(
            {'completed': True, 'revision': 14}).encode('utf-8')
        nok_response = Response()
        nok_response.status_code = 503

        self.mock_get.side_effect = [nok_response, nok_response, ok_response]
        response = upload_app(self.package_name, self.data)
        self.assertEqual(response, {
            'success': True,
            'errors': [],
            'application_url': '',
            'revision': 14,
        })
        self.assertEqual(self.mock_get.call_count, 3)

# 需要导入模块: from requests import Response [as 别名]
# 或者: from requests.Response import status_code [as 别名]
    def test_store(self):
        self.rp.decrypt_survey = MagicMock(return_value=valid_json)
        self.rp.validate_survey = MagicMock()
        # <store>
        self.rp.send_receipt = MagicMock()
        self.rp.send_notification = MagicMock()

        r = Response()
        with mock.patch('app.response_processor.ResponseProcessor.remote_call') as call_mock:
            call_mock.return_value = r

            # 500 - retry
            r.status_code = 500
            with self.assertRaises(RetryableError):
                self._process()

            # 400 - bad
            r.status_code = 400
            with self.assertRaises(QuarantinableError):
                self._process()

            # 200 - ok
            r.status_code = 200
            r.encoding = 'utf-8'
            self._process()

# 需要导入模块: from requests import Response [as 别名]
# 或者: from requests.Response import status_code [as 别名]
def test_reset_rabbit_connection_errors():

    rabbit_manager = Mock()

    with patch('nameko.testing.utils.get_rabbit_connections') as connections:
        connections.return_value = [{
            'vhost': 'vhost_name',
            'name': 'connection_name'
        }]

        # 500 error
        response_500 = Response()
        response_500.status_code = 500
        error_500 = HTTPError(response=response_500)
        rabbit_manager.delete_connection.side_effect = error_500

        with pytest.raises(HTTPError):
            reset_rabbit_connections("vhost_name", rabbit_manager)

        # 404 error
        response_404 = Response()
        response_404.status_code = 404
        error_404 = HTTPError(response=response_404)
        rabbit_manager.delete_connection.side_effect = error_404

        # does not raise
        reset_rabbit_connections("vhost_name", rabbit_manager)

# 需要导入模块: from requests import Response [as 别名]
# 或者: from requests.Response import status_code [as 别名]
    def send(self, request, stream=None, timeout=None, verify=None, cert=None, proxies=None):
        pathname = url_to_path(request.url)

        resp = Response()
        resp.status_code = 200
        resp.url = request.url

        try:
            stats = stat(pathname)
        except OSError as exc:
            resp.status_code = 404
            resp.raw = exc
        else:
            modified = formatdate(stats.st_mtime, usegmt=True)
            content_type = guess_type(pathname)[0] or "text/plain"
            resp.headers = CaseInsensitiveDict({
                "Content-Type": content_type,
                "Content-Length": stats.st_size,
                "Last-Modified": modified,
            })

            resp.raw = open(pathname, "rb")
            resp.close = resp.raw.close

        return resp

# 需要导入模块: from requests import Response [as 别名]
# 或者: from requests.Response import status_code [as 别名]
    def test_avi_json(self):
        rsp = Response()
        rsp.status_code = 404
        rsp._content = 'Not found'
        try:
            avi_rsp = ApiResponse(rsp)
            avi_rsp.json()
            assert False
        except ObjectNotFound:
            pass
        except Exception:
            assert False

        rsp.status_code = 501
        try:
            avi_rsp = ApiResponse(rsp)
            avi_rsp.json()
            assert False
        except APIError:
            pass
        except Exception:
            assert False

        rsp.status_code = 200
        rsp._content = json.dumps({'count': 3, 'results': ['a', 'b', 'c']})
        try:
            avi_rsp = ApiResponse(rsp)
            obj = avi_rsp.json()
            assert obj['count']
            assert avi_rsp.count() == 3
            assert len(obj['results']) == 3
        except Exception as e:
            log.debug('exception %s', str(e))
            log.debug('%s', traceback.format_exc())
            assert False

# 需要导入模块: from requests import Response [as 别名]
# 或者: from requests.Response import status_code [as 别名]
    def test_upload_app_poll_status(self):
        mock_response = self.mock_post.return_value
        mock_response.ok = True
        mock_response.return_value = {
            'success': True,
            'status_url': 'http://example.com/status/'
        }

        response_not_completed = Response()
        response_not_completed.status_code = 200
        response_not_completed.encoding = 'utf-8'
        response_not_completed._content = json.dumps(
            {'completed': False, 'application_url': ''}).encode('utf-8')
        response_completed = Response()
        response_completed.status_code = 200
        response_completed.encoding = 'utf-8'
        response_completed._content = json.dumps(
            {'completed': True, 'revision': 14,
             'application_url': 'http://example.org'}).encode('utf-8')
        self.mock_get.side_effect = [
            response_not_completed,
            response_not_completed,
            response_completed,
        ]
        response = upload_app(self.package_name, self.data)
        self.assertEqual(response, {
            'success': True,
            'errors': [],
            'application_url': 'http://example.org',
            'revision': 14,
        })
        self.assertEqual(self.mock_get.call_count, 3)

# 需要导入模块: from requests import Response [as 别名]
# 或者: from requests.Response import status_code [as 别名]
    def test_decrypt(self):
        # <decrypt>
        self.rp.validate_survey = MagicMock()
        self.rp.store_survey = MagicMock()
        self.rp.send_receipt = MagicMock()
        self.rp.send_notification = MagicMock()
        self.rp.send_to_dap_queue = MagicMock()

        self.rp_invalid.validate_survey = MagicMock()
        self.rp_invalid.store_survey = MagicMock()
        self.rp_invalid.send_receipt = MagicMock()

        r = Response()

        with mock.patch('app.response_processor.ResponseProcessor.remote_call') as call_mock:
            call_mock.return_value = r
            r._content = valid_decrypted.encode('utf-8')

            # 500 - retry
            r.status_code = 500
            with self.assertRaises(RetryableError):
                self._process()

            # 400 - bad
            r.status_code = 400
            with self.assertRaises(QuarantinableError):
                self._process()

            # 200 - ok
            r.status_code = 200
            self._process()

            with self.assertRaises(QuarantinableError):
                self._process_invalid()

# 需要导入模块: from requests import Response [as 别名]
# 或者: from requests.Response import status_code [as 别名]
 def download_func(request, **kwargs):
     self.assertEqual(12, kwargs['timeout'])
     response = Response()
     response._content = "Content"
     if request[0] == 'http://mocktracker.com/3':
         raise Exception("Some strange exception")
     if request[0] == 'http://mocktracker.com/4':
         response.status_code = 500
         return response, None
     response.status_code = 200
     return response, request[1]

# 需要导入模块: from requests import Response [as 别名]
# 或者: from requests.Response import status_code [as 别名]
    def test_does_not_back_off_on_forbidden(self, mock_request, mock_sleep):
        data_set = DataSet('', None)

        good = Response()
        good.status_code = 200
        bad = Response()
        bad.status_code = 403

        mock_request.side_effect = [bad]
        mock_request.__name__ = 'request'

        assert_raises(HTTPError, data_set.post, [{'key': 'foo'}])

# 需要导入模块: from requests import Response [as 别名]
# 或者: from requests.Response import status_code [as 别名]
    def request(self, url, method, **kwargs):

        resp = Response()

        if url == '/role_assignments?scope.domain.id=default':
            json_data = open(UNIT_TEST_RESOURCES_FOLDER + LIST_ROLE_ASSIGNMENTS_RESPONSE_FILE_EXTENDED1).read()
            resp.status_code = OK
            resp._content = json_data
        elif url == '/role_assignments?user.id=fake3&scope.domain.id=default':
            json_data = open(UNIT_TEST_RESOURCES_FOLDER + LIST_USERS_RESPONSE_FILE2).read()
            resp.status_code = OK
            resp._content = json_data
        return resp

# 需要导入模块: from requests import Response [as 别名]
# 或者: from requests.Response import status_code [as 别名]
    def send(self, request, stream=None, timeout=None, verify=None, cert=None, proxies=None):

        resp = Response()
        resp.status_code = 200
        resp.url = request.url

        try:
            import boto
        except ImportError:
            stderrlog.info('\nError: boto is required for S3 channels. '
                           'Please install it with `conda install boto`\n'
                           'Make sure to run `source deactivate` if you '
                           'are in a conda environment.\n')
            resp.status_code = 404
            return resp

        conn = boto.connect_s3()

        bucket_name, key_string = url_to_s3_info(request.url)

        # Get the bucket without validation that it exists and that we have
        # permissions to list its contents.
        bucket = conn.get_bucket(bucket_name, validate=False)

        try:
            key = bucket.get_key(key_string)
        except boto.exception.S3ResponseError as exc:
            # This exception will occur if the bucket does not exist or if the
            # user does not have permission to list its contents.
            resp.status_code = 404
            resp.raw = exc
            return resp

        if key and key.exists:
            modified = key.last_modified
            content_type = key.content_type or "text/plain"
            resp.headers = CaseInsensitiveDict({
                "Content-Type": content_type,
                "Content-Length": key.size,
                "Last-Modified": modified,
            })

            _, self._temp_file = mkstemp()
            key.get_contents_to_filename(self._temp_file)
            f = open(self._temp_file, 'rb')
            resp.raw = f
            resp.close = resp.raw.close
        else:
            resp.status_code = 404

        return resp

# 需要导入模块: from requests import Response [as 别名]
# 或者: from requests.Response import status_code [as 别名]
    def test_backs_off_on_service_unavailable(self, mock_request, mock_sleep):
        data_set = DataSet('', None)

        good = Response()
        good.status_code = 200
        good._content = '[]'
        bad = Response()
        bad.status_code = 503

        mock_request.side_effect = [bad, bad, good]
        mock_request.__name__ = 'request'

        # No exception should be raised
        data_set.post([{'key': 'foo'}])

# 需要导入模块: from requests import Response [as 别名]
# 或者: from requests.Response import status_code [as 别名]
    def test_init(self, requests):
        response = Response()
        response.status_code = 200
        requests.get.return_value = response
        volume = annual.Volume(1010, 12, 4)
        self.assertEqual(True, volume.exists)

        response.status_code = 404
        volume = annual.Volume(1010, 12, 4)
        self.assertEqual(False, volume.exists)

        self.assertTrue('1010' in requests.get.call_args[0][0])
        self.assertTrue('12' in requests.get.call_args[0][0])
        self.assertTrue('4' in requests.get.call_args[0][0])

# 需要导入模块: from requests import Response [as 别名]
# 或者: from requests.Response import status_code [as 别名]
 def do_call(self, url, **kwargs):
     if url.endswith('/test-project/info'):
         response = Response()
         response.raw = BytesIO(
             json.dumps(self.kwargs.get('info', "")).encode('utf-8')
         )
         response.status_code = 200
         return response
     if url.endswith('/test-project/download/all.zip'):
         response = create_zip_response(self.kwargs.get('zip', {}))
         response.status_code = 200
         return response
     else:
         assert NotImplementedError