Python get_user_token Examples

Example #1

def test_fail_upload_model():
    headers = {'Authorization': get_user_token()['id_token']}
    params = {
        'model_name': MODEL_NAME,
        'model_version': 1,
        'file_path': os.path.abspath('non-existing-model.pb'),
    }
    url = f"{MANAGEMENT_API_URL}/tenants/{TENANT_NAME}"
    with pytest.raises(Exception):
        upload_model(url, params, headers, UPLOAD_PART_SIZE)

Example #2

File: test_inference_endpoints.py Project: kbalka/inference-model-manager

def test_upload_model():
    try:
        response = requests.get('https://storage.googleapis.com/inference-eu/models_zoo/'
                                'resnet_V1_50/saved_model/saved_model.pb')
        with open('saved_model.pb', 'wb') as f:
            f.write(response.content)

        headers = {'Authorization': get_user_token()['id_token']}
        params = {
            'model_name': MODEL_NAME,
            'model_version': 1,
            'file_path': os.path.abspath('saved_model.pb'),
        }
        url = f"{MANAGEMENT_API_URL}/tenants/{TENANT_NAME}"

        upload_model(url, params, headers, 30)
        os.remove('saved_model.pb')

    except Exception as e:
        pytest.fail(f"Unexpected error during upload test {e}")

Example #3

def test_upload_models():
    try:
        headers = {'Authorization': get_user_token()['id_token']}
        params = {
            'model_name': MODEL_NAME,
            'model_version': 1,
            'file_path': os.path.abspath('saved_model.pb'),
        }
        url = f"{MANAGEMENT_API_URL}/tenants/{TENANT_NAME}"

        # resnet_v1_50 upload
        download_saved_model_from_path(
            'https://storage.googleapis.com/inference-eu/models_zoo/'
            'resnet_V1_50/saved_model/saved_model.pb')

        upload_model(url, params, headers, UPLOAD_PART_SIZE)
        os.remove('saved_model.pb')

        # resnet_v2_50 upload
        download_saved_model_from_path(
            'https://storage.googleapis.com/inference-eu/models_zoo/'
            'resnet_V2_50/saved_model/saved_model.pb')
        params['model_version'] = 2
        upload_model(url, params, headers, UPLOAD_PART_SIZE)
        os.remove('saved_model.pb')

        # resnet model from tarball
        file_name = 'resnet_v2_fp16_savedmodel_NCHW.tar.gz'
        params = {
            'model_name': MODEL_NAME + '-tar',
            'model_version': 1,
            'file_path': os.path.abspath(file_name),
        }
        download_saved_model_from_path(
            'http://download.tensorflow.org/models/official'
            '/20181001_resnet/savedmodels'
            '/resnet_v2_fp16_savedmodel_NCHW.tar.gz', file_name)
        upload_model(url, params, headers, UPLOAD_PART_SIZE)
        os.remove(file_name)
    except Exception as e:
        pytest.fail(f"Unexpected error during upload test: {e.text}")

Example #4

File: config.py Project: poussa/inference-model-manager

from management_api_tests.authenticate import get_user_token, get_admin_token, \
    get_token

MINIO_ACCESS_KEY_ID = os.environ.get('MINIO_ACCESS_KEY',
                                     'AKIAIOSFODNN7EXAMPLE')
MINIO_SECRET_ACCESS_KEY = os.environ.get(
    'MINIO_SECRET_KEY', 'wJalrXUtnFEMI/K7MDENG/bPxRfiCYEXAMPLEKEY')
MINIO_ENDPOINT_ADDR = os.environ.get('MINIO_ENDPOINT_ADDR',
                                     'http://127.0.0.1:9000')
MINIO_REGION = os.environ.get('MINIO_REGION', 'us-east-1')
SIGNATURE_VERSION = 's3v4'
PORTABLE_SECRETS_PATHS = ['mgt-api/minio-access-info', 'mgt-api/tls-secret']

DEFAULT_HEADERS = {
    'accept': 'application/json',
    'Authorization': get_user_token()['id_token'],
    'Content-Type': 'application/json',
}

ADMIN_HEADERS = {
    'accept': 'application/json',
    'Authorization': get_admin_token()['id_token'],
    'Content-Type': 'application/json',
}

USER1_HEADERS = {
    'accept': 'application/json',
    'Authorization': get_token("mars")['id_token'],
    'Content-Type': 'application/json',
}