From caa1c3ccdf94ee20140b3964aab0ad3058e03699 Mon Sep 17 00:00:00 2001
From: Michaël Ball <michael.ball@gmail.com>
Date: Sun, 7 Feb 2016 15:28:56 +0000
Subject: Create test framework

---
 tests/mach2_test.py | 52 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)
 create mode 100644 tests/mach2_test.py

(limited to 'tests/mach2_test.py')

diff --git a/tests/mach2_test.py b/tests/mach2_test.py
new file mode 100644
index 0000000..c45ff2a
--- /dev/null
+++ b/tests/mach2_test.py
@@ -0,0 +1,52 @@
+import json
+import unittest
+
+import pytest
+
+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 bytes("Log out", "utf-8") in rv.data
+        self.logout()
+        rv = self.login("wrong", "definitelywrong")
+        assert bytes("Log out", "utf-8") not in rv.data
+        self.logout()
+
+    def test_album(self):
+        self.login("admin", "testpass")
+
+        rv = self.app.get("/albums/1")
+        assert bytes("Album 1", "utf-8") in rv.data
+
+        self.logout()
+
+    def test_artists(self):
+        self.login("admin", "testpass")
+        rv = self.app.get("/artists")
+
+        assert bytes("Artist 1", "utf-8") in rv.data
+        assert bytes("Artist 2", "utf-8") in rv.data
+
+        artists = json.loads(rv.data.decode("utf-8"))
+        assert artists
+
+        self.logout()
-- 
cgit v1.2.3