Python append_additional_labels_to_tasks Examples

Example #1

File: test_runners_measurement.py Project: ipuustin/workload-collocation-agent

def test_append_additional_labels_to_tasks__overwriting_label(log_mock):
    """Should not ovewrite existing previously label."""
    task1 = task('/t1', labels={'source_key': '__val__'})
    append_additional_labels_to_tasks(
        {
            'source_key':
            TaskLabelRegexGenerator('__(.*)__', '\\1', 'non_existing_key')
        }, [task1])
    assert task1.labels['source_key'] == '__val__'
    log_mock.debug.assert_called_once()

Example #2

File: test_runners_measurement.py Project: ipuustin/workload-collocation-agent

def test_append_additional_labels_to_tasks__generate_returns_None(log_mock):
    """Generate method for generator returns None."""
    class TestTaskLabelGenerator(TaskLabelGenerator):
        def generate(self, task):
            return None

    task1 = task('/t1', labels={'source_key': 'source_val'})
    append_additional_labels_to_tasks({'target_key': TestTaskLabelGenerator()},
                                      [task1])
    log_mock.debug.assert_called_once()