def test_to_json_string(self): scaling_request = DynamiteScalingRequest() scaling_request.failure_counter = 7 scaling_request.service_instance_name = "apache_service_instance_name" scaling_request.service_name = "apache_service_name" scaling_request.command = DynamiteScalingCommand.SCALE_DOWN json = scaling_request.to_json_string() copy = DynamiteScalingRequest.from_json_string(json) assert copy == scaling_request
def test_to_json_string(self): scaling_request = DynamiteScalingRequest() scaling_request.failure_counter = 7 scaling_request.service_instance_name = "apache_service_instance_name" scaling_request.service_name = "apache_service_name" scaling_request.command = DynamiteScalingCommand.SCALE_DOWN json = scaling_request.to_json_string() copy = DynamiteScalingRequest.from_json_string(json) assert copy == scaling_request
def receive(self): try: received_scaling_request_string = self._process_queue.get(block=False) scaling_request = DynamiteScalingRequest.from_json_string( received_scaling_request_string, message_processed_callback=lambda: None ) self._logger.debug("Received scaling request:{:s}".format(received_scaling_request_string)) return scaling_request except Empty: self._logger.debug("No request received.") return None
def receive(self): try: received_scaling_request_string = self._process_queue.get( block=False) scaling_request = DynamiteScalingRequest.from_json_string( received_scaling_request_string, message_processed_callback=lambda: None) self._logger.debug("Received scaling request:{:s}".format( received_scaling_request_string)) return scaling_request except Empty: self._logger.debug("No request received.") return None
def test_init_from_json_string(self): json_string = """ { "service_instance_name": "apache_service_instance", "service_name": "apache_service", "command": "scale_up", "failure_counter": 8 } """ request = DynamiteScalingRequest.from_json_string(json_string) assert request.command == DynamiteScalingCommand.SCALE_UP assert request.service_name == "apache_service" assert request.service_instance_name == "apache_service_instance" assert request.failure_counter == 8
def test_init_from_json_string(self): json_string = """ { "service_instance_name": "apache_service_instance", "service_name": "apache_service", "command": "scale_up", "failure_counter": 8 } """ request = DynamiteScalingRequest.from_json_string(json_string) assert request.command == DynamiteScalingCommand.SCALE_UP assert request.service_name == "apache_service" assert request.service_instance_name == "apache_service_instance" assert request.failure_counter == 8
def receive(self): if self._queue_channel is None: raise ValueError("You need to connect before you can receive any message!") method_frame, header_frame, message_body = self._queue_channel.basic_get(queue=self._queue_name, no_ack=False) if self._no_message_delivered(method_frame, header_frame): self._logger.debug("Received no scaling request") return None received_scaling_request_string = message_body.decode("utf-8") message_processed_callback = lambda: self._on_message_processed(method_frame.delivery_tag) scaling_request = DynamiteScalingRequest.from_json_string( received_scaling_request_string, message_processed_callback=message_processed_callback ) self._logger.debug("Received scaling request:{:s}".format(received_scaling_request_string)) return scaling_request
def check_if_content_in_queue(self, content): for content_message in content: method_frame, header_frame, message_body = self._channel.basic_get(queue=self.RABBITMQ_SCALING_REQUEST_QUEUE_NAME, no_ack=False) request = DynamiteScalingRequest.from_json_string(message_body.decode("utf-8")) success_or_error = "[SUCCESS]" if request == content_message else "[ERROR]" print("{} Received message {}".format(success_or_error, message_body))