from flask import json, url_for from openhdi.app import app, g, QUIZ, url_for from openhdi.mongo import get_db import openhdi.model as model db = get_db() def setup(): pass class TestApp(): def setup(self): self.app = app.test_client() def teardown(self): db.weighting.drop() def test_index(self): res = self.app.get('/') assert 'Which country is closest' in res.data, res.data def test_quiz(self): with app.test_client() as c: res = c.get('/quiz/1') assert 'Economy' in res.data, res.data assert 'Education' in res.data, res.data assert 'Step 1' in res.data, res.data assert '__dimension__' in res.data, res.data print g.user_id assert g.user_id in res.data, res.data