コード例 #1
0
ファイル: tests.py プロジェクト: dgelvin/minio-py
def main():
    """
    Functional testing of minio python library.
    """
    fake = Factory.create()
    client = Minio('s3.amazonaws.com',
                   os.getenv('ACCESS_KEY'),
                   os.getenv('SECRET_KEY'))

    _http = urllib3.PoolManager(
        cert_reqs='CERT_REQUIRED',
        ca_certs=certifi.where()
    )

    # Get unique bucket_name, object_name.
    bucket_name = uuid.uuid4().__str__()
    object_name = uuid.uuid4().__str__()

    # Enable trace
    # client.trace_on(sys.stderr)

    # Make a new bucket.
    bucket_name = 'minio-pytest'

    print(client.make_bucket(bucket_name))
    print(client.make_bucket(bucket_name+'.unique',
                             location='us-west-1'))

    ## Check if return codes a valid from server.
    try:
        client.make_bucket(bucket_name+'.unique',
                           location='us-west-1')
    except ResponseError as err:
        if str(err.code) in ['BucketAlreadyOwnedByYou', 'BucketAlreadyExists']:
            pass
        else:
            raise

    # Check if bucket was created properly.
    print(client.bucket_exists(bucket_name))
    print(client.bucket_exists(bucket_name+'.unique'))

    # Set bucket name to private.
    print(client.set_bucket_acl(bucket_name, Acl.private()))
    print(client.set_bucket_acl(bucket_name+'.unique', Acl.private()))

    # Verify current bucket acl.
    acl = client.get_bucket_acl(bucket_name)
    if acl != 'private':
        raise ValueError('Invalid acl type found: ' + acl)
    acl = client.get_bucket_acl(bucket_name+'.unique')
    if acl != 'private':
        raise ValueError('Invalid acl type found: ' + acl)

    # Set bucket name to public-read.
    print(client.set_bucket_acl(bucket_name, Acl.public_read()))
    print(client.set_bucket_acl(bucket_name+'.unique', Acl.public_read()))

    # Verify current bucket acl.
    acl = client.get_bucket_acl(bucket_name)
    if acl != 'public-read':
        raise ValueError('Invalid acl type found: ' + acl)
    acl = client.get_bucket_acl(bucket_name+'.unique')
    if acl != 'public-read':
        raise ValueError('Invalid acl type found: ' + acl)

    # Set bucket name to public-read-write.
    print(client.set_bucket_acl(bucket_name, Acl.public_read_write()))
    print(client.set_bucket_acl(bucket_name+'.unique', Acl.public_read_write()))

    # Verify current bucket acl.
    acl = client.get_bucket_acl(bucket_name)
    if acl != 'public-read-write':
        raise ValueError('Invalid acl type found: ' + acl)
    acl = client.get_bucket_acl(bucket_name+'.unique')
    if acl != 'public-read-write':
        raise ValueError('Invalid acl type found: ' + acl)

    # List all buckets.
    buckets = client.list_buckets()
    for bucket in buckets:
        print(bucket.name, bucket.creation_date)

    with open('testfile', 'wb') as file_data:
        file_data.write(fake.text().encode('utf-8'))
    file_data.close()

    # Put a file
    file_stat = os.stat('testfile')
    with open('testfile', 'rb') as file_data:
        client.put_object(bucket_name, object_name, file_data, file_stat.st_size)
    file_data.close()

    # Fput a file
    print(client.fput_object(bucket_name, object_name+'-f', 'testfile'))

    # Fetch stats on your object.
    print(client.stat_object(bucket_name, object_name))

    # Get a full object
    object_data = client.get_object(bucket_name, object_name)
    with open('newfile', 'wb') as file_data:
        for data in object_data:
            file_data.write(data)
    file_data.close()

    # Get a full object locally.
    print(client.fget_object(bucket_name, object_name, 'newfile-f'))

    # List all object paths in bucket.
    objects = client.list_objects(bucket_name, recursive=True)
    for obj in objects:
        print(obj.bucket_name, obj.object_name, obj.last_modified, \
            obj.etag, obj.size, obj.content_type)

    presigned_get_object_url = client.presigned_get_object(bucket_name, object_name)
    response = _http.urlopen('GET', presigned_get_object_url)
    if response.status != 200:
        response_error = ResponseError(response)
        raise response_error.get(bucket_name, object_name)

    presigned_put_object_url = client.presigned_put_object(bucket_name, object_name)
    value = fake.text().encode('utf-8')
    data = io.BytesIO(value).getvalue()
    response = _http.urlopen('PUT', presigned_put_object_url, body=data)
    if response.status != 200:
        response_error = ResponseError(response)
        raise response_error.put(bucket_name, object_name)

    object_data = client.get_object(bucket_name, object_name)
    if object_data.read() != value:
        raise ValueError('Bytes not equal')

    # Post policy.
    policy = PostPolicy()
    policy.set_bucket_name(bucket_name)
    policy.set_key_startswith('objectPrefix/')

    expires_date = datetime.utcnow()+timedelta(days=10)
    policy.set_expires(expires_date)
    print(client.presigned_post_policy(policy))

    # Remove an object.
    print(client.remove_object(bucket_name, object_name))
    print(client.remove_object(bucket_name, object_name+'-f'))

    # Remove a bucket. This operation will only work if your bucket is empty.
    print(client.remove_bucket(bucket_name))
    print(client.remove_bucket(bucket_name+'.unique'))

    # Remove temporary files.
    os.remove('testfile')
    os.remove('newfile')
    os.remove('newfile-f')