message_handlers_test.py 文件源码

def test_handle_schedule_command(self, talk_class_mock, send_schedule_mock):
        messaging_event = self.generate_postback('schedule payload')
        talks_mock = [MagicMock(talk_id=1), MagicMock(talk_id=2)]
        server.db_session.query().all = MagicMock(return_value=talks_mock)
        talk_like_numbers_mock = {}
        for mock_index, talk_mock in enumerate(talks_mock):
            talk_mock.id = mock_index
            talk_mock.count_likes = MagicMock(return_value=1)
            talk_like_numbers_mock[talk_mock.id] = 1
        talk_like_ids_mock = []
        message_handlers.handle_schedule_command(messaging_event, self.access_token, server.db_session)
        send_schedule_mock.assert_called_once_with(
            self.access_token,
            self.sender_id,
            talks_mock,
            talk_like_numbers_mock,
            talk_like_ids_mock  
            )