Python Headers.set方法代码示例

# 需要导入模块: from werkzeug.datastructures import Headers [as 别名]
# 或者: from werkzeug.datastructures.Headers import set [as 别名]
def test_200_with_credentials(client, test_user):
    headers = Headers()
    headers.set("Authorization", basic_auth_encode("[email protected]", "secret123"))

    response = client.get("/user", headers=headers)
    assert response.status_code == 200
    assert response.json.get("user") == test_user.id

# 需要导入模块: from werkzeug.datastructures import Headers [as 别名]
# 或者: from werkzeug.datastructures.Headers import set [as 别名]
def test_200_with_authorization_header(client, access_token, test_user):
    headers = Headers()
    headers.set("Authorization", "Bearer " + access_token)

    response = client.get("/user", headers=headers)
    assert response.status_code == 200
    assert response.json.get("user") == test_user.id

# 需要导入模块: from werkzeug.datastructures import Headers [as 别名]
# 或者: from werkzeug.datastructures.Headers import set [as 别名]
def test_200_with_credentials(client):
    headers = Headers()
    headers.set("Authorization", basic_auth_encode("[email protected]", "secret123"))

    response = client.get("/secrets", headers=headers)
    assert response.status_code == 200
    assert isinstance(response.json.get("secrets"), list)

# 需要导入模块: from werkzeug.datastructures import Headers [as 别名]
# 或者: from werkzeug.datastructures.Headers import set [as 别名]
def test_401_with_invalid_token(client):
    headers = Headers()
    headers.set('Authorization', 'Bearer invalid-token')

    response = client.get('/user', headers=headers)
    assert response.status_code == 401
    assert response.json.get('error') == 'invalid_token'

# 需要导入模块: from werkzeug.datastructures import Headers [as 别名]
# 或者: from werkzeug.datastructures.Headers import set [as 别名]
def test_401_with_invalid_credentials(client):
    headers = Headers()
    headers.set('Authorization', 'Basic ' + base64.b64encode('invalid-username:invalid-password'))

    response = client.get('/user', headers=headers)
    assert response.status_code == 401
    assert response.json.get('error') == 'invalid_token'

# 需要导入模块: from werkzeug.datastructures import Headers [as 别名]
# 或者: from werkzeug.datastructures.Headers import set [as 别名]
def test_200_with_authorization_header(client, access_token):
    headers = Headers()
    headers.set('Authorization', 'Bearer ' + access_token)

    response = client.get('/secrets', headers=headers)
    assert response.status_code == 200
    assert isinstance(response.json.get('secrets'), list)

# 需要导入模块: from werkzeug.datastructures import Headers [as 别名]
# 或者: from werkzeug.datastructures.Headers import set [as 别名]
def test_401_with_unsupported_authorization_header(client):
    headers = Headers()
    headers.set('Authorization', 'MAC 123456789')

    response = client.get('/user', headers=headers)
    assert response.status_code == 401
    assert response.json.get('error') == 'invalid_token'

# 需要导入模块: from werkzeug.datastructures import Headers [as 别名]
# 或者: from werkzeug.datastructures.Headers import set [as 别名]
def test_200_with_credentials(client, test_user):
    headers = Headers()
    headers.set('Authorization', 'Basic ' + base64.b64encode('[email protected]:secret123'))

    response = client.get('/user', headers=headers)
    assert response.status_code == 200
    assert response.json.get('user') == test_user.id

# 需要导入模块: from werkzeug.datastructures import Headers [as 别名]
# 或者: from werkzeug.datastructures.Headers import set [as 别名]
def test_400_without_grant_type(client):
    headers = Headers()
    headers.set('Origin', ORIGIN)

    response = client.post('/oauth/token', headers=headers)
    assert response.status_code == 400
    assert response.headers.get('Access-Control-Allow-Origin') == ORIGIN
    assert response.headers.get('Access-Control-Allow-Credentials') == 'true'
    assert response.json.get('error') == 'unsupported_grant_type'

# 需要导入模块: from werkzeug.datastructures import Headers [as 别名]
# 或者: from werkzeug.datastructures.Headers import set [as 别名]
def test_cors(client):
    headers = Headers()
    headers.set('Origin', ORIGIN)

    response = client.get('/', headers=headers)
    assert response.headers.get('Access-Control-Allow-Origin') == ORIGIN
    assert response.headers.get('Access-Control-Allow-Credentials') == 'true'
    assert 'Access-Control-Allow-Methods' not in response.headers
    assert 'Access-Control-Allow-Headers' not in response.headers

# 需要导入模块: from werkzeug.datastructures import Headers [as 别名]
# 或者: from werkzeug.datastructures.Headers import set [as 别名]
def test_401_with_expired_token(client, access_token):
    headers = Headers()
    headers.set('Authorization', 'Bearer ' + access_token)

    with freeze_time('2099-01-14'):
        response = client.get('/user', headers=headers)

    assert response.status_code == 401
    assert response.json.get('error') == 'invalid_token'

# 需要导入模块: from werkzeug.datastructures import Headers [as 别名]
# 或者: from werkzeug.datastructures.Headers import set [as 别名]
def test_401_with_invalid_credentials(client):
    headers = Headers()
    headers.set(
        "Authorization", basic_auth_encode("invalid-username", "invalid-password")
    )

    response = client.get("/secrets", headers=headers)
    assert response.status_code == 401
    assert response.json.get("error") == "invalid_token"

# 需要导入模块: from werkzeug.datastructures import Headers [as 别名]
# 或者: from werkzeug.datastructures.Headers import set [as 别名]
def test_401_with_expired_token(client, access_token):
    headers = Headers()
    headers.set("Authorization", "Bearer " + access_token)

    with immobilus("2099-01-14"):
        response = client.get("/secrets", headers=headers)

    assert response.status_code == 401
    assert response.json.get("error") == "invalid_token"

# 需要导入模块: from werkzeug.datastructures import Headers [as 别名]
# 或者: from werkzeug.datastructures.Headers import set [as 别名]
def test_missing_user_agent_returns_403(client):
    headers = Headers()
    headers.set('User-Agent', '')

    res = client.get('/', headers=headers)
    assert res.status_code == 403

    json = res.json
    assert isinstance(json, dict)
    assert 'User-Agent header' in json.get('message')

# 需要导入模块: from werkzeug.datastructures import Headers [as 别名]
# 或者: from werkzeug.datastructures.Headers import set [as 别名]
def return_hydro_info(scenario_id):
    '''
    This function is used to return all the hydro information
    '''
    # TODO extract hydro data based on id
    # the id is an list separate by ---
    final_csv_raw_data = []

    id_list = scenario_id.split('---')
    for temp_id in id_list:
        scenarios = Scenario.objects(id=temp_id).first()
        hydro_data_list = scenarios.hydrograph.streamflow_array
        hydro_time_list = scenarios.hydrograph.time_array

        final_csv_raw_data.append([temp_id, hydro_time_list, hydro_data_list])

    if len(hydro_time_list) != len(hydro_data_list):
        return 'time and data arrays are not the same length'
    else:
        # idea is from
        # http://stackoverflow.com/questions/28011341/create-and-download-a-csv-file
        def generate():
            data = StringIO.StringIO()
            w = csv.writer(data)

            # write header
            w.writerow(('timestamp', 'HydroValue'))
            yield data.getvalue()
            data.seek(0)
            data.truncate(0)

            # write each log item
            # TODO this part final_csv_raw_data[0], coz create a csv file only based on
            # the first scenario
            for count in range(len(final_csv_raw_data[0][1])):
                w.writerow((
                    # format datetime as string
                    final_csv_raw_data[0][1][count].isoformat(),
                    final_csv_raw_data[0][2][count]
                ))
                yield data.getvalue()
                data.seek(0)
                data.truncate(0)

        # add a filename
        headers = Headers()
        headers.set('Content-Disposition', 'attachment', filename='hydro.csv')

        # stream the response as the data is generated
        return Response(
            stream_with_context(generate()),
            mimetype='text/csv', headers=headers
        )