from flask import g, url_for
from test_base import BaseTestCase


class AccessScreenTests(BaseTestCase):


    def setUp(self):
        g.locale = 'en'

    def test_should_redirect_when_access_root(self):
        assert '302 FOUND' == self.client.get('/').status

#    def test_english_home_screen_is_up_and_running(self):
#        response = self.client.get('/en/')
#        self.assert_200(response, message=None)

#    def test_portuguese_home_screen_is_up_and_running(self):
#        response = self.client.get('/pt/')
#        self.assert_200(response)

    def test_home_screen_access(self):
        response = self.client.get('/')
        self.assertRedirects(response, url_for('general.home'))

    def test_help_screen_access(self):
        response = self.client.get(url_for('help.index'))
        self.assert_200(response)

    def test_search_screen_access(self):
        response = self.client.get(url_for('general.search'))
        self.assert_200(response)

    def test_rankin_screen_access(self):
        response = self.client.get(url_for('rankings.index'))
        self.assert_200(response)

    def test_buid_graph_screen_access(self):
        response = self.client.get(url_for('build_graph.index'))
        self.assert_200(response)

    def test_buid_graph_costummer_screen_access(self):
        response = self.client.get(g.locale + 
        	'/build_graph/rais/all/all/all?view=Employment%20by%20Municipality&graph=geo_map')
        self.assert_200(response)
from flask import g, url_for
from test_base import BaseTestCase


class AccessScreenTests(BaseTestCase):


    def setUp(self):
        g.locale = 'en'

    def test_should_redirect_when_access_root(self):
        assert '302 FOUND' == self.client.get('/').status

#    def test_english_home_screen_is_up_and_running(self):
#        response = self.client.get('/en/')
#        self.assert_200(response, message=None)

#    def test_portuguese_home_screen_is_up_and_running(self):
#        response = self.client.get('/pt/')
#        self.assert_200(response)

    def test_home_screen_access(self):
        response = self.client.get('/')
        self.assertRedirects(response, url_for('general.home'))

    def test_help_screen_access(self):
        response = self.client.get(url_for('help.index'))
        self.assert_200(response)

    def test_search_screen_access(self):
        response = self.client.get(url_for('general.search'))
        self.assert_200(response)

    def test_rankin_screen_access(self):
        response = self.client.get(url_for('rankings.index'))
        self.assert_200(response)

    def test_buid_graph_screen_access(self):
        response = self.client.get(url_for('build_graph.index'))
        self.assert_200(response)

    def test_buid_graph_costummer_screen_access(self):
        response = self.client.get(g.locale + 
        	'/build_graph/rais/all/all/all?view=Employment%20by%20Municipality&graph=geo_map')
        self.assert_200(response)

    def test_data_screen_access(self):
        response = self.client.get(url_for('data.index'))
        self.assert_200(response)
Add build data access test.