summaryrefslogtreecommitdiff
path: root/tests/mach2_test.py
blob: c33fd9228f0212e7db3d3e3fd8f4e3e75c1ad9ef (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
import json
import unittest

import pytest
import six

from mach2 import create_app


@pytest.mark.usefixtures("app")
class Mach2TestCase(unittest.TestCase):

    def setUp(self):
        app = create_app(database=self.db, library=self.library_db)
        app.config['TESTING'] = True
        self.app = app.test_client()

    def login(self, username, password):
        return self.app.post('/login', data=dict(
            username=username,
            password=password
        ), follow_redirects=True)

    def logout(self):
        return self.app.get('/logout', follow_redirects=True)

    def test_login(self):
        rv = self.login("admin", "testpass")
        assert six.b("Log out") in rv.data
        self.logout()
        rv = self.login("wrong", "definitelywrong")
        assert six.b("Log out") not in rv.data
        self.logout()

    def test_album(self):
        self.login("admin", "testpass")

        rv = self.app.get("/albums/1")
        assert six.b("Album 1") in rv.data

        self.logout()

    def test_artists(self):
        self.login("admin", "testpass")
        rv = self.app.get("/artists")

        assert six.b("Artist 1") in rv.data
        assert six.b("Artist 2") in rv.data

        artists = json.loads(rv.data.decode("utf-8"))
        assert artists

        self.logout()