예제 #1
0
def msg(config, filter_):
    '''Creates a new :py:class:`pymco.message.Message` instance.'''
    # Importing here since py-cov will ignore code imported on conftest files
    # imports
    from pymco import message
    with mock.patch('time.time') as time:
        with mock.patch('hashlib.sha1') as sha1:
            time.return_value = ctxt.MSG['msgtime']
            sha1.return_value.hexdigest.return_value = ctxt.MSG['requestid']
            msg_ = message.Message(body=ctxt.MSG['body'],
                                   agent=ctxt.MSG['agent'],
                                   filter_=filter_,
                                   config=config)
            time.assert_called_once_with()
            sha1.return_value.hexdigest.assert_called_once_with()
    return msg_
예제 #2
0
def msg(config, filter_):
    '''Creates a new :py:class:`pymco.message.Message` instance.'''
    # Importing here since py-cov will ignore code imported on conftest files
    # imports
    from pymco import message
    with mock.patch('time.time') as time:
        with mock.patch('hashlib.sha1') as sha1:
            time.return_value = ctxt.MSG['msgtime']
            sha1.return_value.hexdigest.return_value = ctxt.MSG['requestid']
            msg_ = message.Message(body=ctxt.MSG['body'],
                                   agent=ctxt.MSG['agent'],
                                   filter_=filter_,
                                   config=config)
            time.assert_called_once_with()
            sha1.return_value.hexdigest.assert_called_once_with()
    return msg_
예제 #3
0
def msg_with_data(config, filter_):
    """Creates :py:class:`pymco.message.Message` instance with some data."""
    # Importing here since py-cov will ignore code imported on conftest files
    # imports
    from pymco import message
    with mock.patch('time.time') as time:
        with mock.patch('hashlib.sha1') as sha1:
            time.return_value = ctxt.MSG['msgtime']
            sha1.return_value.hexdigest.return_value = ctxt.MSG['requestid']
            body = {
                ':action': 'runonce',
                ':data': {':noop': True, ':process_results': True},
                ':ssl_msgtime': 1421878604,
                ':ssl_ttl': 60,
            }
            return message.Message(body=body,
                                   agent='puppet',
                                   filter_=filter_,
                                   config=config)
예제 #4
0
def msg_with_data(config, filter_):
    """Creates :py:class:`pymco.message.Message` instance with some data."""
    # Importing here since py-cov will ignore code imported on conftest files
    # imports
    from pymco import message
    with mock.patch('time.time') as time:
        with mock.patch('hashlib.sha1') as sha1:
            time.return_value = ctxt.MSG['msgtime']
            sha1.return_value.hexdigest.return_value = ctxt.MSG['requestid']
            body = {
                ':action': 'runonce',
                ':data': {
                    ':noop': True,
                    ':process_results': True
                },
                ':ssl_msgtime': 1421878604,
                ':ssl_ttl': 60,
            }
            return message.Message(body=body,
                                   agent='puppet',
                                   filter_=filter_,
                                   config=config)