def send_start_flow_msg(dag_name, dag_run_id):
step_finish_event = {
"dag_name": dag_name,
"dag_run_id": dag_run_id,
"operation": EventOperation.Start_Flow,
"time": time.time(),
}
print(step_finish_event)
mq_broker = get_MQ_Broker_Object()
mq_broker.send_msg(step_finish_event)
def send_finish_step_msg(dag_name,
dag_run_id,
step_name,
status=None,
message=None,
result=None):
if not status:
status = StepStatus.Succeeded
step_finish_event = {
"dag_name": dag_name,
"dag_run_id": dag_run_id,
"operation": EventOperation.Finish_Step,
"time": time.time(),
"step_name": step_name,
"status": status,
"message": message,
"result": result,
}
print(step_finish_event)
mq_broker = get_MQ_Broker_Object()
mq_broker.send_msg(step_finish_event)
__author__ = 'godq'
import unittest
import time
from dagflow.loader import get_MQ_Broker_Object, get_DagRepo_Object
from dagflow.event import EventOperation
from dagflow.flow_operation import send_finish_step_msg, send_start_flow_msg
mq_broker = get_MQ_Broker_Object()
dag_repo = get_DagRepo_Object()
class StartFlowAsyncEventTest(unittest.TestCase):
def test_start_async_flow(self):
dag_name = "dag_def_test_{}".format(time.time())
dag = {
"name":
dag_name,
"steps": [
{
"name": "step1",
"task_func": "hello_plugin",
"args": {
"name": "hello world1"
},
"upstreams": [],
"downstreams": ["step2"]
},
{
"name": "step2",
def send_event_message(event):
if isinstance(event, dict):
event = json.dumps(event)
# message_queue.put_message(event)
mq_broker = get_MQ_Broker_Object()
mq_broker.send_msg(event)