# Copyright 2015 Google Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import contextlib

import bookshelf
from conftest import flaky_filter
from flaky import flaky
import mock
from oauth2client.client import OAuth2Credentials
import pytest


@pytest.fixture
def client_with_credentials(app):
    """This fixture provides a Flask app test client that has a session
    pre-configured with use credentials."""
    credentials = OAuth2Credentials(
        'access_token',
        'client_id',
        'client_secret',
        'refresh_token',
        '3600',
        None,
        'Test',
        id_token={'sub': '123', 'email': 'user@example.com'},
        scopes=('email', 'profile'))

    @contextlib.contextmanager
    def inner():
        with app.test_client() as client:
            with client.session_transaction() as session:
                session['profile'] = {'id': 'abc', 'displayName': 'Test User'}
                session['google_oauth2_credentials'] = credentials.to_json()
            yield client

    return inner


# Mark all test cases in this class as flaky, so that if errors occur they
# can be retried. This is useful when databases are temporarily unavailable.
@flaky(rerun_filter=flaky_filter)
# Tell pytest to use both the app and model fixtures for all test cases.
# This ensures that configuration is properly applied and that all database
# resources created during tests are cleaned up. These fixtures are defined
# in conftest.py
@pytest.mark.usefixtures('app', 'model')
class TestAuth(object):
    def test_not_logged_in(self, app):
        with app.test_client() as c:
            rv = c.get('/books/')

        assert rv.status == '200 OK'
        body = rv.data.decode('utf-8')
        assert 'Login' in body

    def test_logged_in(self, client_with_credentials):
        with client_with_credentials() as c:
            rv = c.get('/books/')

        assert rv.status == '200 OK'
        body = rv.data.decode('utf-8')
        assert 'Test User' in body

    def test_add_anonymous(self, app):
        data = {
            'title': 'Test Book',
        }

        with app.test_client() as c:
            rv = c.post('/books/add', data=data, follow_redirects=True)

        assert rv.status == '200 OK'
        body = rv.data.decode('utf-8')
        assert 'Test Book' in body
        assert 'Added by Anonymous' in body

    def test_add_logged_in(self, client_with_credentials):
        data = {
            'title': 'Test Book',
        }

        with client_with_credentials() as c:
            rv = c.post('/books/add', data=data, follow_redirects=True)

        assert rv.status == '200 OK'
        body = rv.data.decode('utf-8')
        assert 'Test Book' in body
        assert 'Added by Test User' in body

    def test_mine(self, model, client_with_credentials):
        # Create two books, one created by the logged in user and one
        # created by another user.
        model.create({
            'title': 'Book 1',
            'createdById': 'abc'
        })

        model.create({
            'title': 'Book 2',
            'createdById': 'def'
        })

        # Check the "My Books" page and make sure only one of the books
        # appears.
        with client_with_credentials() as c:
            rv = c.get('/books/mine')

        assert rv.status == '200 OK'
        body = rv.data.decode('utf-8')
        assert 'Book 1' in body
        assert 'Book 2' not in body

    @mock.patch("httplib2.Http")
    def test_request_user_info(self, HttpMock):
        httpObj = mock.MagicMock()
        responseMock = mock.MagicMock(status=200)
        httpObj.request = mock.MagicMock(
            return_value=(responseMock, b'{"name": "bill"}'))
        HttpMock.return_value = httpObj
        credentials = mock.MagicMock()
        bookshelf._request_user_info(credentials)