Esempio n. 1
0
    def setUp(self):
        surpressResourceWarning()

        configuration = ais_client.Configuration()
        configuration.debug = False
        api_client = ais_client.ApiClient(configuration)

        self.sort = ais_client.api.sort_api.SortApi(api_client)
        self.bucket = ais_client.api.bucket_api.BucketApi(api_client)
        self.object = ais_client.api.object_api.ObjectApi(api_client)
        self.models = ais_client.models

        # Create local bucket
        input_params = self.models.InputParameters(self.models.Actions.CREATELB)
        self.bucket.perform_operation(self.BUCKET_NAME, input_params)

        # Create and send tars
        for i in range(0, self.SHARDS):
            out = io.BytesIO()
            object_name = "%s%d.tar" % (self.PREFIX, i)
            with tarfile.open(mode="w", fileobj=out) as tar:
                for j in range(0, 100):
                    b = "Hello world!".encode("ascii")
                    t = tarfile.TarInfo("%d.txt" % j)
                    t.size = len(b)
                    tar.addfile(t, io.BytesIO(b))

            self.object.put(self.BUCKET_NAME, object_name, body=bytestring(out.getvalue()))
Esempio n. 2
0
    def setUp(self):
        surpressResourceWarning()

        configuration = ais_client.Configuration()
        configuration.debug = False
        api_client = ais_client.ApiClient(configuration)
        self.daemon = ais_client.api.daemon_api.DaemonApi(api_client)
        self.models = ais_client.models
Esempio n. 3
0
        def createAisApiClients(daemonlist):
            """ Create openapi client API handles for a list of daemons (does not initiate connection yet). """

            for d in daemonlist:
                if d['pod'].status.pod_ip is None:
                    d['aisClientApi'] = None
                    continue

                ais_config = ais_client.Configuration()
                ais_config.debug = False
                ais_config.host = "http://%s:%s/v1" % (d['pod'].status.pod_ip, d['pod'].spec.containers[0].ports[0].container_port)
                d['aisClientApi'] = ais_client.ApiClient(ais_config)
Esempio n. 4
0
    def setUp(self):
        surpressResourceWarning()

        configuration = ais_client.Configuration()
        configuration.debug = False
        api_client = ais_client.ApiClient(configuration)
        self.bucket = ais_client.api.bucket_api.BucketApi(api_client)
        self.object = ais_client.api.object_api.ObjectApi(api_client)
        self.models = ais_client.models
        self.BUCKET_NAME = os.environ["BUCKET"]
        self.FILE_SIZE = 128
        self.SLEEP_LONG_SECONDS = 15
        self.created_objects = []
        self.created_buckets = []
Esempio n. 5
0
    def setUp(self):
        surpressResourceWarning()

        configuration = ais_client.Configuration()
        configuration.debug = False
        api_client = ais_client.ApiClient(configuration)
        self.object = ais_client.api.object_api.ObjectApi(api_client)
        self.bucket = ais_client.api.bucket_api.BucketApi(api_client)
        self.models = ais_client.models
        self.BUCKET_NAME = os.environ["BUCKET"]
        self.assertTrue(self.BUCKET_NAME,
                        "Environment variable 'BUCKET' not set.")
        self.FILE_SIZE = 128
        self.created_objects = []
        self.created_buckets = []