Python SafeConfigParser嘲笑

Question

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

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

谷歌使用SafeConfigParser的模拟后，我已经知道如何测试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)

P.S。您可以查看此网址的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