예제 #1
0
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