from sauce.tests import TestController

__all__ = ['TestSite']


class TestSite(TestController):
    sites = ['/', '/about', '/contact']

    def _do_site(self, site):
        response = self.app.get(site)
        assert response.status.startswith('2')

    def test_sites(self):
        for site in self.sites:
            yield self._do_site, site