mox generator とか欲しい

最近、テストを書くのがとても面倒になってきたので、mox で簡単にスタブできないか考えてみた。

まずは、クラスを。

mox_generator.py

import mox
class MoxGenerator(object):

    def __init__(self):
        self.mox = mox.Mox()

まだ思いつきの書き始めなので、とりあえず１つ書いてみました。なんでもいいから、True を返すスタブします。引数すら考慮せず。これはひどい（汗）。

mox_generator.py

    def true_stub(self, stub_class, stub_method, num_args):
        self.mox.StubOutWithMock(stub_class, stub_method)
        stubbed = getattr(stub_class, stub_method)
        args = tuple([mox.IgnoreArg() for i in range(num_args)])
        stubbed(*args).AndReturn(True)

呼び出すときは、こんな感じで。ここでは、MyClass.hoge(x,x,x) が True を返すように。最後の数字は、MyClass.hoge に渡す引数の個数です。

example_test.py

    moxgen= MoxGenerator()
    moxgen.true_stub(MyClass, 'hoge', 3)

もちろん Replay も面倒なので、名前が変だけどジェネレータに入れてしまえ！

mox_generator.py

    import contextlib
    @contextlib.contextmanager
    def mox_replay(self):
        self.mox.ReplayAll()
        yield
        self.mox.VerifyAll()

テスト実行側。

example_test.py

    with moxgen.mox_replay():
        MyClass.method_which_should_call_hoge()

github にも上げてみました。https://github.com/norobust/mox_generator なぜか本名でコミットされとるやないかー。どうやってユーザ名に変えるんだろう？