Merge pull request #2311 from bovarysme/use-yield-syntax

Use the yield syntax in pytest's fixtures
This commit is contained in:
David Lord 2017-05-23 12:00:34 -07:00 committed by GitHub
commit a037762781
3 changed files with 25 additions and 28 deletions

View file

@ -15,18 +15,16 @@ from minitwit import minitwit
@pytest.fixture
def client(request):
def client():
db_fd, minitwit.app.config['DATABASE'] = tempfile.mkstemp()
client = minitwit.app.test_client()
with minitwit.app.app_context():
minitwit.init_db()
def teardown():
"""Get rid of the database again after each test."""
os.close(db_fd)
os.unlink(minitwit.app.config['DATABASE'])
request.addfinalizer(teardown)
return client
yield client
os.close(db_fd)
os.unlink(minitwit.app.config['DATABASE'])
def register(client, username, password, password2=None, email=None):