def test_format_data_with_unjsonable_dict(self): class A: pass data = {'a': 'something', 'b': A()} expected = str(data) self.assertEqual(CaseLogger._format_data(data), expected)
def test_create_event_entry(self): event = WalkoffEvent.WorkflowExecutionStart uid = uuid.uuid4() data = {'a': 'something', 'b': 5} expected_data = json.dumps(data) event_entry = CaseLogger._create_event_entry(event, uid, data) self.assertEqual(event_entry.type, EventType.workflow.name) self.assertEqual(event_entry.originator, uid) self.assertEqual(event_entry.message, event.value.message) self.assertEqual(event_entry.data, expected_data)
def __init__(self, config): """Initializes a new Context object. This acts as an interface for objects to access other event specific variables that might be needed. Args: config (Config): A config object """ self.execution_db = walkoff.executiondb.ExecutionDatabase( config.EXECUTION_DB_TYPE, config.EXECUTION_DB_PATH) self.case_db = walkoff.case.database.CaseDatabase( config.CASE_DB_TYPE, config.CASE_DB_PATH) self.subscription_cache = SubscriptionCache() self.case_logger = CaseLogger(self.case_db, self.subscription_cache) self.cache = walkoff.cache.make_cache(config.CACHE) self.executor = executor.MultiprocessedExecutor( self.cache, self.case_logger) self.scheduler = walkoff.scheduler.Scheduler(self.case_logger)
def __init__(self, id_, config_path): """Initialize a Workfer object, which will be managing the execution of Workflows Args: id_ (str): The ID of the worker config_path (str): The path to the configuration file to be loaded """ logger.info('Spawning worker {}'.format(id_)) self.id_ = id_ self._lock = Lock() signal.signal(signal.SIGINT, self.exit_handler) signal.signal(signal.SIGABRT, self.exit_handler) if os.name == 'nt': walkoff.config.initialize(config_path=config_path) else: walkoff.config.Config.load_config(config_path) self.execution_db = ExecutionDatabase( walkoff.config.Config.EXECUTION_DB_TYPE, walkoff.config.Config.EXECUTION_DB_PATH) self.case_db = CaseDatabase(walkoff.config.Config.CASE_DB_TYPE, walkoff.config.Config.CASE_DB_PATH) @WalkoffEvent.CommonWorkflowSignal.connect def handle_data_sent(sender, **kwargs): self.on_data_sent(sender, **kwargs) self.handle_data_sent = handle_data_sent self.thread_exit = False server_secret_file = os.path.join( walkoff.config.Config.ZMQ_PRIVATE_KEYS_PATH, "server.key_secret") server_public, server_secret = auth.load_certificate( server_secret_file) client_secret_file = os.path.join( walkoff.config.Config.ZMQ_PRIVATE_KEYS_PATH, "client.key_secret") client_public, client_secret = auth.load_certificate( client_secret_file) socket_id = u"Worker-{}".format(id_).encode("ascii") key = PrivateKey( client_secret[:nacl.bindings.crypto_box_SECRETKEYBYTES]) server_key = PrivateKey( server_secret[:nacl.bindings.crypto_box_SECRETKEYBYTES]).public_key self.cache = walkoff.cache.make_cache(walkoff.config.Config.CACHE) self.capacity = walkoff.config.Config.NUMBER_THREADS_PER_PROCESS self.subscription_cache = SubscriptionCache() case_logger = CaseLogger(self.case_db, self.subscription_cache) self.workflow_receiver = WorkflowReceiver(key, server_key, walkoff.config.Config.CACHE) self.workflow_results_sender = WorkflowResultsHandler( socket_id, client_secret, client_public, server_public, walkoff.config.Config.ZMQ_RESULTS_ADDRESS, self.execution_db, case_logger) self.workflow_communication_receiver = WorkflowCommunicationReceiver( socket_id, client_secret, client_public, server_public, walkoff.config.Config.ZMQ_COMMUNICATION_ADDRESS) self.comm_thread = threading.Thread(target=self.receive_communications) self.comm_thread.start() self.workflows = {} self.threadpool = ThreadPoolExecutor(max_workers=self.capacity) self.receive_workflows()
def test_format_data_with_jsonable_dict(self): data = {'a': 'something', 'b': 5} expected = json.dumps(data) self.assertEqual(CaseLogger._format_data(data), expected)
def test_format_data_with_string(self): string = 'something' self.assertEqual(CaseLogger._format_data(string), string)
def test_format_data_with_none(self): self.assertEqual(CaseLogger._format_data(None), '')
def get_case_logger(subscriptions): repo = create_autospec(CaseDatabase) return CaseLogger(repo, subscriptions=subscriptions)