supysonic/tests/frontend/test_login.py

#!/usr/bin/env python
# coding: utf-8
#
# This file is part of Supysonic.
# Supysonic is a Python implementation of the Subsonic server API.
#
# Copyright (C) 2013-2017 Alban 'spl0k' Féron
#                    2017 Óscar García Amor
#
# Distributed under terms of the GNU AGPLv3 license.

import uuid

from pony.orm import db_session

from supysonic.db import User

from .frontendtestbase import FrontendTestBase

class LoginTestCase(FrontendTestBase):
    def test_unauthorized_request(self):
        # Unauthorized request
        rv = self.client.get('/', follow_redirects=True)
        self.assertIn('Please login', rv.data)

    def test_login_with_bad_data(self):
        # Login with not blank user or password
        rv = self._login('', '')
        self.assertIn('Missing user name', rv.data)
        self.assertIn('Missing password', rv.data)
        # Login with not valid user or password
        rv = self._login('nonexistent', 'nonexistent')
        self.assertIn('Wrong username or password', rv.data)
        rv = self._login('alice', 'badpassword')
        self.assertIn('Wrong username or password', rv.data)

    def test_login_admin(self):
        # Login with a valid admin user
        rv = self._login('alice', 'Alic3')
        self.assertIn('Logged in', rv.data)
        self.assertIn('Users', rv.data)
        self.assertIn('Folders', rv.data)

    def test_login_non_admin(self):
        # Login with a valid non-admin user
        rv = self._login('bob', 'B0b')
        self.assertIn('Logged in', rv.data)
        # Non-admin user cannot acces to users and folders
        self.assertNotIn('Users', rv.data)
        self.assertNotIn('Folders', rv.data)

    def test_root_with_valid_session(self):
        # Root with valid session
        with db_session:
            with self.client.session_transaction() as sess:
                sess['userid'] = User.get(name = 'alice').id
        rv = self.client.get('/', follow_redirects=True)
        self.assertIn('alice', rv.data)
        self.assertIn('Log out', rv.data)
        self.assertIn('There\'s nothing much to see here.', rv.data)

    def test_root_with_non_valid_session(self):
        # Root with a no-valid session
        with self.client.session_transaction() as sess:
            sess['userid'] = uuid.uuid4()
        rv = self.client.get('/', follow_redirects=True)
        self.assertIn('Please login', rv.data)

    def test_multiple_login(self):
        self._login('alice', 'Alic3')
        rv = self._login('bob', 'B0b')
        self.assertIn('Already logged in', rv.data)
        self.assertIn('alice', rv.data)

if __name__ == '__main__':
    unittest.main()