test_app.py

# -*- coding: utf-8 -*-
importmock
fromflaskimportredirect, session
fromappimportgithub


deftest_app(client):
assertclient.get("/").status_code==200


deftest_login(client):
defmock_post(*args, **kwargs):
response=mock.Mock()
response.content=b'access_token=mocktoken&token_type=bearer'
returnresponse

defmock_get_user(*args, **kwargs):
return {'login': 'mock_userid'}

@mock.patch.object(github, 'get', new=mock_get_user)
@mock.patch.object(github.session, 'post', new=mock_post)
defmock_authorize(*args, **kwargs):
client.get('/github/callback?code=mockcode')
returnredirect('/')

@mock.patch.object(github, 'authorize', new=mock_authorize)
deftest():
client.get('/login')
assertsession['oauth_token'] =='mocktoken'
assertsession['u_id']['id'] =='mock_userid'

test()


deftest_logout(client):
client.get('/logout')
assertsession.get('u_id') isNone