Python TaskManagerAgentBase.log示例

示例#1

文件： Test_Agent_TransformationSystem.py 项目： marianne013/DIRAC

def test__fillTheQueue(mocker, operationsOnTransformationsDict, expected):
  mocker.patch('DIRAC.TransformationSystem.Agent.TaskManagerAgentBase.AgentModule', side_effect=mockAM)
  mocker.patch('DIRAC.TransformationSystem.Agent.TaskManagerAgentBase.FileReport', side_effect=MagicMock())
  tmab = TaskManagerAgentBase()
  tmab.log = gLogger
  tmab.am_getOption = mockAM
  tmab._fillTheQueue(operationsOnTransformationsDict)
  assert tmab.transInQueue == expected[0]
  assert tmab.transQueue.qsize() == expected[1]

示例#2

def test__fillTheQueue(mocker, operationsOnTransformationsDict, expected):
    mocker.patch(
        'DIRAC.TransformationSystem.Agent.TaskManagerAgentBase.AgentModule',
        side_effect=mockAM)
    mocker.patch(
        'DIRAC.TransformationSystem.Agent.TaskManagerAgentBase.FileReport',
        side_effect=MagicMock())
    tmab = TaskManagerAgentBase()
    tmab.log = gLogger
    tmab.am_getOption = mockAM
    tmab._fillTheQueue(operationsOnTransformationsDict)
    assert tmab.transInQueue == expected[0]
    assert tmab.transQueue.qsize() == expected[1]

示例#3

from mock import MagicMock

from DIRAC import gLogger
# sut
from DIRAC.TransformationSystem.Agent.TaskManagerAgentBase import TaskManagerAgentBase
from DIRAC.TransformationSystem.Agent.TransformationAgent import TransformationAgent

gLogger.setLevel('DEBUG')

mockAM = MagicMock()
tmab_m = importlib.import_module(
    'DIRAC.TransformationSystem.Agent.TaskManagerAgentBase')
tmab_m.AgentModule = mockAM
tmab_m.FileReport = MagicMock()
tmab = TaskManagerAgentBase()
tmab.log = gLogger
tmab.am_getOption = mockAM
tmab.log.setLevel('DEBUG')


@pytest.mark.parametrize("operationsOnTransformationsDict, expected",
                         [({
                             1: {
                                 'Operations': ['op1', 'op2'],
                                 'Body': 'veryBigBody'
                             }
                         }, ([1], 1)),
                          ({
                              2: {
                                  'Operations': ['op3', 'op2'],
                                  'Body': 'veryveryBigBody'