1

Pythonで、moxユニットテストライブラリを使用してwithステートメントで作成されたオブジェクトをモックするにはどうすればよいですか？

コード

class MyCode:
    def generate_gzip_file(self):
        with gzip.GzipFile('file_name.txt.gz','wb') as f:
             f.write('data')

単体テスト

class MyCodeTest(unittest.TestCase):
    def test_generate_gzip_file(self):
        mox = mox.Mox()
        mock_gzip_file = self.mox.CreateMock(gzip.GzipFile)
        mox.StubOutWithMock(gzip, 'GzipFile')
        gzip.GzipFile('file_name.txt.gz','wb').AndReturn(mock_file)
        mock_gzip_file.write('data')
        mox.ReplayAll()
        MyCode().generate_gzip_file()
        mox.VerifyAll()

AttributeError: __exit__オンラインでエラーが発生します

with gzip.GzipFile('file_name.txt.gz','wb') as f:
4

1 に答える 1

3

のモック化されたインスタンスが何らかの理由でメソッドでgzip.GzipFile終わらないという DSM は正しいです。ステートメントで使用するクラス__exit__を定義するのを忘れると、まったく同じエラーが発生します。例えば：__exit__with

>>> class C(object):
...   def __enter__(self):
...     return self
... 
>>> with C() as c:
...   pass
... 
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AttributeError: __exit__

CreateMockAnything()幸いなことに、Mox の方法を使用mock_gzip_fileして、特定のインターフェイスを強制しないオブジェクトを作成することで、この問題を回避できます。オブジェクトの期待値を正しく設定するように注意する必要がありますmock_gzip_file(つまり、__enter__()および__exit__(...)メソッドがいつ、どのように呼び出されるかについての期待値を設定します)。これが私のために働いた例です：

import gzip
import mox
import unittest

class MyCode:
    def generate_gzip_file(self):
        with gzip.GzipFile('file_name.txt.gz', 'wb') as f:
             f.write('data')

class MyCodeTest(unittest.TestCase):
    def test_generate_gzip_file(self):
        mymox = mox.Mox()
        mock_gzip_file = mymox.CreateMockAnything()
        mymox.StubOutWithMock(gzip, 'GzipFile')
        gzip.GzipFile('file_name.txt.gz', 'wb').AndReturn(mock_gzip_file)
        mock_gzip_file.__enter__().AndReturn(mock_gzip_file)
        mock_gzip_file.write('data')
        mock_gzip_file.__exit__(None, None, None).AndReturn(None)
        mymox.ReplayAll()

        MyCode().generate_gzip_file()
        mymox.VerifyAll()

if __name__ == '__main__':
    unittest.main()

これを実行すると、次のようになります。

.
----------------------------------------------------------------------
Ran 1 test in 0.001s

OK
于 2012-01-19T01:23:38.960 に答える