Esempio n. 1
0
    def setUp(self):
        self.ip = '127.0.0.1'
        self.response_key = 'result'

        self.kvs_client = kvs_client.MockAnnaClient()
        self.socket = zmq_utils.MockZmqSocket()
        self.pusher_cache = zmq_utils.MockPusherCache()
        self.user_library = CloudburstUserLibrary(zmq_utils.MockZmqContext(),
                                                  self.pusher_cache, self.ip,
                                                  0, self.kvs_client)
Esempio n. 2
0
    def setUp(self):
        self.pusher_cache = zmq_utils.MockPusherCache()
        self.socket = zmq_utils.MockZmqSocket()
        self.pin_socket = zmq_utils.MockZmqSocket()

        self.kvs_client = kvs_client.MockAnnaClient()
        self.ip = '127.0.0.1'

        self.policy = DefaultCloudburstSchedulerPolicy(self.pin_socket,
                                                       self.pusher_cache,
                                                       self.kvs_client,
                                                       self.ip,
                                                       random_threshold=0)
Esempio n. 3
0
    def setUp(self):
        self.kvs_client = kvs_client.MockAnnaClient()
        self.socket = zmq_utils.MockZmqSocket()
        self.pusher_cache = zmq_utils.MockPusherCache()

        self.ip = '127.0.0.1'
        self.status = ThreadStatus()
        self.status.ip = self.ip
        self.status.tid = 0
        self.status.running = True

        self.pinned_functions = {}
        self.runtimes = {}
        self.exec_counts = {}
Esempio n. 4
0
    def setUp(self):
        self.kvs_client = kvs_client.MockAnnaClient()
        self.socket = zmq_utils.MockZmqSocket()
        self.pusher_cache = zmq_utils.MockPusherCache()

        self.ip = '127.0.0.1'
        self.status = ThreadStatus()
        self.status.ip = self.ip
        self.status.tid = 0
        self.status.running = True

        self.pinned_functions = {}
        self.runtimes = {}
        self.exec_counts = {}

        self.user_library = CloudburstUserLibrary(zmq_utils.MockZmqContext(),
                                                  self.pusher_cache, self.ip,
                                                  0, self.kvs_client)
Esempio n. 5
0
    def setUp(self):
        self.pusher_cache = zmq_utils.MockPusherCache()
        self.socket = zmq_utils.MockZmqSocket()
        self.pin_socket = zmq_utils.MockZmqSocket()

        self.kvs_client = kvs_client.MockAnnaClient()
        self.ip = '127.0.0.1'

        self.policy = DefaultCloudburstSchedulerPolicy(self.pin_socket,
                                                       self.pusher_cache,
                                                       self.kvs_client,
                                                       self.ip,
                                                       random_threshold=0)

        # Add an executor to the policy engine by default.
        status = ThreadStatus()
        status.ip = self.ip
        status.tid = 0
        self.executor_key = (status.ip, status.tid)
        self.policy.unpinned_executors.add(self.executor_key)