test_app.py

# -*- coding: utf-8 -*-
import mock
from flask import redirect, session
from app import github


def test_app(client):
    assert client.get("/").status_code == 200


def test_login(client):
    def mock_post(*args, **kwargs):
        response = mock.Mock()
        response.content = b'access_token=mocktoken&token_type=bearer'
        return response

    def mock_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)
    def mock_authorize(*args, **kwargs):
        client.get('/github/callback?code=mockcode')
        return redirect('/')

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

    test()


def test_logout(client):
    client.get('/logout')
    assert session.get('u_id') is None