コード例 #1
0
ファイル: test_actions.py プロジェクト: tmooney/flow-workflow
    def test_environment(self):
        operation_id = 999
        color = 9
        color_descriptor = mock.Mock()
        color_descriptor.color = color

        parent_log_dir = '/happy/log/dir'

        operation_data = OperationData(net_key=self.net.key,
                operation_id=operation_id,
                color=color)

        environment = {
            'foo': 'bar',
            'baz': 'buz',
        }
        expected_environment = {
                'FLOW_WORKFLOW_OPERATION_DATA': operation_data.dumps(),
                'FLOW_PARENT_WORKFLOW_LOG_DIR': parent_log_dir,
        }
        expected_environment.update(environment)

        self.net.constant.return_value = environment

        with mock.patch('flow_workflow.perl_action.actions.factory') as factory:
            operation = mock.Mock()
            operation.log_dir = parent_log_dir
            factory.load_operation.return_value = operation

            result_env = self.action.environment(self.net, color_descriptor)
            factory.load_operation.assert_called_once_with(self.net,
                    self.args['operation_id'])

        self.net.constant.assert_called_once_with('environment', {})
        self.assertEqual(expected_environment, result_env)
コード例 #2
0
ファイル: actions.py プロジェクト: tmooney/flow-workflow
    def environment(self, net, color_descriptor):
        env = net.constant('environment', {})

        operation = self._get_operation(net)
        operation_data = OperationData(net_key=net.key,
                operation_id=self.args['operation_id'],
                color=color_descriptor.color)

        env['FLOW_WORKFLOW_OPERATION_DATA'] = operation_data.dumps()
        env['FLOW_PARENT_WORKFLOW_LOG_DIR'] = operation.log_dir
        return env
コード例 #3
0
class OperationDataTests(TestCase):
    def setUp(self):
        self.net_key = 'some_net_key'
        self.operation_id = 999
        self.color = 888
        self.operation_data = OperationData(net_key=self.net_key,
                operation_id=self.operation_id,
                color=self.color)
        self.string = '{"color": 888, "net_key": "some_net_key", "operation_id": 999}'

    def test_to_dict(self):
        expected_dict = {
                'net_key':self.net_key,
                'operation_id':self.operation_id,
                'color':self.color,
                }
        self.assertItemsEqual(expected_dict, self.operation_data.to_dict)

    def test_from_dict(self):
        some_dict = {
                'net_key':'some_net_key',
                'operation_id':999,
                'color':888,
                }
        operation_data = OperationData.from_dict(some_dict)
        self.assertEqual(self.operation_data, operation_data)

    def test_dumps(self):
        self.assertEqual(self.string, self.operation_data.dumps())

    def test_loads(self):
        operation_data = OperationData.loads(self.string)
        self.assertEqual(self.operation_data, operation_data)
コード例 #4
0
 def setUp(self):
     self.net_key = 'some_net_key'
     self.operation_id = 999
     self.color = 888
     self.operation_data = OperationData(net_key=self.net_key,
             operation_id=self.operation_id,
             color=self.color)
     self.string = '{"color": 888, "net_key": "some_net_key", "operation_id": 999}'
コード例 #5
0
 def test_from_dict(self):
     some_dict = {
             'net_key':'some_net_key',
             'operation_id':999,
             'color':888,
             }
     operation_data = OperationData.from_dict(some_dict)
     self.assertEqual(self.operation_data, operation_data)
コード例 #6
0
ファイル: handler.py プロジェクト: tmooney/flow-workflow
    def _get_message_dict(self, message):
        message_dict = message.to_dict()

        message_dict['status'] = Status(message_dict['status'])

        OPERATION_DATA_FIELDS = ['operation_data', 'parent_operation_data',
                'peer_operation_data']
        for field in OPERATION_DATA_FIELDS:
            if field in message_dict:
                message_dict[field] = OperationData.from_dict(
                        message_dict[field])

        return message_dict
コード例 #7
0
 def test_loads(self):
     operation_data = OperationData.loads(self.string)
     self.assertEqual(self.operation_data, operation_data)
コード例 #8
0
ファイル: launch_base.py プロジェクト: tmooney/flow-workflow
 def operation_data(self):
     string = os.environ.get('FLOW_WORKFLOW_OPERATION_DATA')
     if string:
         return OperationData.loads(string)
     else:
         return False