From 06d64ec6dfa0a84804f48c43f1053d9c13449f63 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sybren=20A=2E=20St=C3=BCvel?= <sybren@stuvel.eu>
Date: Fri, 29 Jul 2016 17:43:58 +0200
Subject: [PATCH] Added unit test

---
 attract_server/modules.py      |  3 ++-
 runserver.py                   | 11 +++++++++++
 tests/abstract_attract_test.py | 12 ++++++++++++
 tests/test_jemoeder.py         |  7 +++++++
 4 files changed, 32 insertions(+), 1 deletion(-)
 create mode 100755 runserver.py
 create mode 100644 tests/abstract_attract_test.py
 create mode 100644 tests/test_jemoeder.py

diff --git a/attract_server/modules.py b/attract_server/modules.py
index 0c0b317..d784bde 100644
--- a/attract_server/modules.py
+++ b/attract_server/modules.py
@@ -1,8 +1,9 @@
 from flask import Blueprint
+from pillar_server.utils import jsonify
 
 blueprint = Blueprint('attract', __name__)
 
 
 @blueprint.route('/jemoeder')
 def jemoeder():
-    return 'je moeder'
+    return jsonify({'je': 'moeder'})
diff --git a/runserver.py b/runserver.py
new file mode 100755
index 0000000..bd9eebe
--- /dev/null
+++ b/runserver.py
@@ -0,0 +1,11 @@
+#!/usr/bin/env python
+
+from pillar_server import PillarServer
+from attract_server import AttractExtension
+
+app = PillarServer('.')
+app.load_extension(AttractExtension(), '/attract')
+app.process_extensions()
+
+if __name__ == '__main__':
+    app.run('::0', 5000, debug=True)
diff --git a/tests/abstract_attract_test.py b/tests/abstract_attract_test.py
new file mode 100644
index 0000000..09d0349
--- /dev/null
+++ b/tests/abstract_attract_test.py
@@ -0,0 +1,12 @@
+from pillar_server.tests import TestPillarServer, AbstractPillarTest
+from attract_server import AttractExtension
+
+class TestAttractServer(TestPillarServer):
+    def __init__(self, *args, **kwargs):
+        super(TestAttractServer, self).__init__(*args, **kwargs)
+
+        self.load_extension(AttractExtension(), '/attract')
+
+
+class AbstractAttractTest(AbstractPillarTest):
+    pillar_server_class = TestAttractServer
diff --git a/tests/test_jemoeder.py b/tests/test_jemoeder.py
new file mode 100644
index 0000000..ddd3e9f
--- /dev/null
+++ b/tests/test_jemoeder.py
@@ -0,0 +1,7 @@
+from abstract_attract_test import AbstractAttractTest
+
+
+class JeMoederTest(AbstractAttractTest):
+    def test_je_moeder(self):
+        resp = self.get('/attract/jemoeder')
+        self.assertEqual({'je': 'moeder'}, resp.json())