forked from orbit-oss/flask
Merge pull request #2311 from bovarysme/use-yield-syntax
Use the yield syntax in pytest's fixtures
This commit is contained in:
commit
a037762781
3 changed files with 25 additions and 28 deletions
|
|
@ -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):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue