You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
22 lines
597 B
22 lines
597 B
# -*- coding: utf-8 -*-
|
|
# Copyright 2017 LasLabs Inc.
|
|
# License LGPL-3.0 or later (http://www.gnu.org/licenses/lgpl).
|
|
|
|
from contextlib import contextmanager
|
|
from mock import MagicMock
|
|
|
|
from odoo.tests.common import TransactionCase
|
|
|
|
|
|
class Common(TransactionCase):
|
|
|
|
@contextmanager
|
|
def _mock_method(self, method_name, method_obj=None):
|
|
if method_obj is None:
|
|
method_obj = self.record
|
|
magic = MagicMock()
|
|
method_obj._patch_method(method_name, magic)
|
|
try:
|
|
yield magic
|
|
finally:
|
|
method_obj._revert_method(method_name)
|