使用pytest.fixture选择退出功能

Question

I have a function that I don't want to run every time I run tests in my Flask-RESTFul API. This is an example of the setup:

class function(Resource):
    def post(self):
        print 'test'
        do_action()
        return {'success':True}

in my test I want to run this function, but ignore do_action(). How would I make this happen using pytest?

Answer 1

This seems like a good opportunity to mark the tests

@pytest.mark.foo_test
class function(Resource):
    def post(self):
        print 'test'
        do_action()
        return {'success':True}

Then if you call with

py.test -v -m foo_test

It will run only those tests marked "foo_test"

If you call with

py.test -v -m "not foo_test"

It will run all tests not marked "foo_test"

Answer 2

You can mock do_action in your test:

def test_post(resource, mocker):
    m = mocker.patch.object(module_with_do_action, 'do_action')
    resource.post()
    assert m.call_count == 1

So the actual function will not be called in this test, with the added benefit that you can check if the post implementation is actuall calling the function.

This requires pytest-mocker to be installed (shameless plug).