Python SafeConfigParser 模拟

Question

我正在研究如何在我的单元测试程序中使用模拟。

现在我有一个 SafeConfigParser 对象，我想测试我编写的内容是否正确。

在google了SafeConfigParser的mocking用法之后，我已经知道如何测试SafeConfigParser的读取了。但我还是不知道如何验证SafeConfigParser的写入。

我的想法是:

1. 创建一个空缓冲区。
2. 考虑一种可以将缓冲区设置为 SafeConfigParser 的方法。
3. 调用包含SafeConfigParser.write()的函数
4. 用我的答案验证缓冲区。

我需要测试的程序如下:

def write_tokens_to_config(self):
    """Write token values to the config
    """

    parser = SafeConfigParser()
    with open(self.CONFIG_PATH) as fp:
        parser.readfp(fp)
        if not parser.has_section('Token'):
            parser.add_section('Token')
    parser.set('Token', 'access_token', self._access_token)
    parser.set('Token', 'refresh_token', self._refresh_token)
    with open(self.CONFIG_PATH, 'wb') as fp:
        parser.write(fp)

附注您可以从此网址检查read部分:http://www.snip2code.com/Snippet/4347/

Answer 1

我终于找到了解决方案:)。

我将我的程序(例如:program.py)修改为以下内容:

class Program(): 

    def __init__(self):
        self._access_token = None
        self._refresh_token = None
        self.CONFIG_PATH = 'test.conf'

    def write_tokens_to_config(self):
        """Write token value to the config
        """

        parser = SafeConfigParser()
        parser.read(self.CONFIG_PATH)
        if not parser.has_section('Token'):
            parser.add_section('Token')
        parser.set('Token', 'access_token', self._access_token)
        parser.set('Token', 'refresh_token', self._refresh_token)
        with open(self.CONFIG_PATH, 'wb') as f:
            parser.write(f)

我的测试程序是这样的:

class TestMyProgram(unittest.TestCase):

    def setUp(self):
        from program import Program
        self.program = Program()

    def test_write_tokens_to_config(self):
        from mock import mock_open
        from mock import call
        self.program._access_token = 'aaa'
        self.program._refresh_token = 'bbb'
        with mock.patch('program.ConfigParser.SafeConfigParser.read'):
            m = mock_open()
            with mock.patch('__builtin__.open', m, create=True):
                self.program.write_tokens_to_config()
                m.assert_called_once_with(self.program.CONFIG_PATH, 'wb')
                handle = m()
                handle.write.assert_has_calls(
                    [
                        call('[Token]\n'),
                        call('access_token = aaa\n'),
                        call('refresh_token = bbb\n'),
                    ]
                )

引用号:http://docs.python.org/dev/library/unittest.mock#mock-open