Python-сообщество

from netrc import netrc
from transmissionrpc import Client as Transmission

def list_torrents(trans, _args):
    """List torrents."""
    for num, torr in trans.list().iteritems():
        fields = torr.fields
        print num, fields['name'][:30], status(torr), torr.progress, torr.eta

def main():
    nrc = netrc()
    tr_login, _, tr_passwd = nrc.authenticators('localhost')
    trans = Transmission(user=tr_login, password=tr_passwd, timeout=10)
    list_torrents(trans, [])

#from netrc import netrc
#from transmissionrpc import Client as Transmission
def list_torrents(trans, _args):
    # ...
def main(netrc, Transmission):
    # ...

from transmissionrc import client as Transmission
from netrc import netrc

def main(netrc, Transmission):
    ....

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import mock
from StringIO import StringIO
netrc = mock.Mock()
trans = mock.Mock()
 
# сюда нужно будет вписать все торренты?
# пока, просто моки
desired_items = {i: mock.MagicMock() for i in range(10)}
 
with mock.patch.dict('sys.modules',
                     {'transmissionrpc': trans,
                      'netrc': netrc}):
    netrc.netrc.return_value.authenticators.return_value = ['user', '?', 'password']
    trans.Client.return_value.list.return_value = desired_items
 
    with mock.patch('sys.stdout', new_callable=StringIO) as stdout:
        import some_lib
        some_lib.main()
 
    # assert 'desired output' in stdout.getvalue() 
    print stdout.getvalue()

torr.fields['name'][:30]

0 <MagicMock name='mock.fields.__getitem__().__getitem__()' id='26179088'> status? <MagicMock name='mock.progress' id='26473104'> <MagicMock name='mock.eta' id='26473264'>
1 <MagicMock name='mock.fields.__getitem__().__getitem__()' id='26578192'> status? <MagicMock name='mock.progress' id='26603408'> <MagicMock name='mock.eta' id='26624080'>
2 <MagicMock name='mock.fields.__getitem__().__getitem__()' id='26720816'> status? <MagicMock name='mock.progress' id='27098288'> <MagicMock name='mock.eta' id='27098448'>
3 <MagicMock name='mock.fields.__getitem__().__getitem__()' id='27215696'> status? <MagicMock name='mock.progress' id='27240912'> <MagicMock name='mock.eta' id='27241072'>
...

т.е. fields['name'] вызывает __getitem__ и возвращает magic_mock, также и magic_mock[:30] возвращает мок.

def status(x):
    return 'status?'

class FakeTransmission(object):
    def __init__(self, user, password, timeout=10):
        pass

    def list(self):
        return {1: transmissionrpc.Torrent(self,
            {'id': 1, 'name': 'My torrent', 'status': 0,
             'sizeWhenDone': 100, 'leftUntilDone': 50,
             'eta': 20})}

class FakeNetrc(object):
    def authenticators(self, _host):
        return 1, 2, 3

def test_ls():
    mod.Transmission = FakeTransmission
    mod.netrc = FakeNetrc
    mod.sys.stdout = StringIO()
    mod.main()
    mod.sys.stdout.seek(0)
    assert mod.sys.stdout.read() == "1 My torrent stopped 50.0 0:00:20\n"