python链式函数_Python模拟链式函数调用

我在单元测试中的一个方法中有以下语句。db_employees = self.db._session.query(Employee).filter(Employee.dept ==

new_employee.dept).all()

我想让德银员工拿到模拟员工名单。我试图通过以下方式实现这一目标：m = MagickMock()

m.return_value.filter().all().return_value = employees

其中employees是employee对象的列表。但这没用。当我试图打印任何属性的值时，它有一个模拟值。这就是代码的外观：class Database(object):

def __init__(self, user=None, passwd=None, db="sqlite:tmp/emp.db"):

try:

engine = create_engine(db)

except Exception:

raise ValueError("Database '%s' does not exist." % db)

def on_connect(conn, record):

conn.execute('pragma foreign_keys=ON')

if 'sqlite://' in db:

event.listen(engine, 'connect', on_connect)

Base.metadata.bind = engine

DBSession = sessionmaker(bind=engine)

self._session = DBSession()

class TestEmployee(MyEmployee):

def setUp(self):

self.db = emp.database.Database(db=options.connection)

self.db._session._autoflush()

@mock.patch.object(session.Session, 'add')

@mock.patch.object(session.Session, 'query')

def test_update(self, mock_query, mock_add):

employees = [{'id': 1,

'name': 'Pradeep',

'department': 'IT',

'manager': 'John'}]

mock_add.side_effect = self.add_side_effect

mock_query.return_value = self.query_results()

self.update_employees(employees)

def add_side_effect(self, instance, _warn=True):

// Code to mock add

// Values will be stored in a dict which will be used to

// check with expected value.

def query_results(self):

m = MagicMock()

if self.count == 0:

m.return_value.filter.return_value.all.return_value = [employee]

elif:

m.return_value.filter.return_value.all.return_value = [department]

return m

我将查询结果作为测试调用查询两次的方法。首先是employee表，然后是department表。

如何模拟此链接函数调用？