У меня есть следующее утверждение в одном из тестируемых методов.

db_employees = self.db._session.query(Employee).filter(Employee.dept ==   
    new_employee.dept).all()

Я хочу, чтобы db_employees получал фиктивный список сотрудников. Я пытался добиться этого с помощью:

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

Где сотрудники - это список сотрудников объекта. Но это не сработало. Когда я пытаюсь напечатать значение какого-либо атрибута, он имеет фиктивное значение. Вот как выглядит код:

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  

У меня есть query_results, поскольку тестируемый метод вызывает запрос дважды. Сначала таблица сотрудников, а затем таблица отделов.

Как мне посмеяться над этим вызовом функции?

13
Pradeep 16 Дек 2015 в 12:40

3 ответа

Лучший ответ

Вы должны исправить метод query() атрибута _session Database и настроить его так, чтобы он дал правильный ответ. Вы можете сделать это разными способами, но ИМХО более чистым способом является исправление DBSession статической ссылки query. Я не знаю, из какого модуля вы импортировали DBSession, поэтому я исправлю локальную ссылку.

Другой аспект - это фиктивная конфигурация: мы установим возвращаемое значение query, которое в вашем случае станет объектом с методом filter().

class TestEmployee(MyEmployee):
    def setUp(self):
        self.db = emp.database.Database(db=options.connection)
        self.db._session._autoflush()
        self.log_add = {}

    @mock.patch.object(__name__.'DBSession.add')     
    @mock.patch.object(__name__.'DBSession.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)
        .... your test here

    def add_side_effect(self, instance, _warn=True):
        # ... storing data
        self.log_add[...] = [...]

    def query_results(self):  
        m = MagicMock()
        value = "[department]"
        if not self.count:  
             value = "[employee]"  
        m.filter.return_value.all.return_value = value 
        return m
2
Michele d'Amico 16 Дек 2015 в 22:41

Я нашел решение аналогичной проблемы, где мне нужно было макетировать вложенный набор фильтрации вызовов.

Данный тестируемый код похож на следующее:

interesting_cats = (session.query(Cats)
                           .filter(Cat.fur_type == 'furry')
                           .filter(Cat.voice == 'meowrific')
                           .filter(Cat.color == 'orande')
                           .all())

Вы можете настроить макеты, как показано ниже:

mock_session_response = MagicMock()
# This is the magic - create a mock loop
mock_session_response.filter.return_value = mock_session_response
# We can exit the loop with a call to 'all'
mock_session_response.all.return_value = provided_cats

mock_session = MagicMock(spec=Session)
mock_session.query.return_value = mock_session_response
4
Ron Dahlgren 8 Мар 2018 в 06:33
m = MagickMock()
m.session.query().filter().all.return_value = employees

https://docs.python.org/3/library/unittest.mock.html

17
Tate Thurston 30 Ноя 2017 в 04:37