base.py 2.7 KB
Newer Older
1 2
from django import test
from responsediff.test import ResponseDiffTestMixin
3
from memopol_scores.models import RepresentativeScore
4 5 6


class BaseTest(ResponseDiffTestMixin, test.TestCase):
7
    fixtures = ['data_sample.json']
8 9 10

    """
    Common queries
11
    - 1 for settings
Nicolas Joyard's avatar
Nicolas Joyard committed
12 13 14 15 16 17 18 19 20
    - 5 for search forms
        - 1 for chambers
        - 1 for countries
        - 1 for parties
        - 1 for committees
        - 1 for delegations
    - 2 for the position form
        - 1 for representatives
        - 1 for themes
21
    """
22
    left_pane_queries = 8
23

24 25 26
    def setUp(self):
        RepresentativeScore.refresh()

Nicolas Joyard's avatar
Nicolas Joyard committed
27 28 29
    def request_test(self, url=None):
        self.assertResponseDiffEmpty(self.client.get(url or self.url))

Nicolas Joyard's avatar
Nicolas Joyard committed
30 31 32
    def selector_test(self, selector, url=None):
        self.assertResponseDiffEmpty(self.client.get(url or self.url),
                                     selector)
33 34 35 36


class RepresentativeBaseTest(BaseTest):
    tab = 'none'
Nicolas Joyard's avatar
Nicolas Joyard committed
37
    base_url = '/representatives/olivier-dussopt-1978-08-16/%s/'
38 39 40

    """
    Common queries plus:
41
    - 1 for session key
42 43 44 45 46 47 48 49
    - 1 for chamber abbreviations (helper for chamber websites)
    - 1 for the representative
    - 1 for the main mandate
    - 1 for emails
    - 1 for social websites
    - 1 for chamber websites
    - 1 for other websites
    - 1 for addresses
50 51 52
    - 1 for address country
    - 1 for phone numbers related to addresses
    - 1 for other phone numbers
53
    - 2 for themes and theme scores
54
    """
55
    queries = BaseTest.left_pane_queries + 14
56 57 58 59 60 61 62 63 64 65 66

    @property
    def url(self):
        return self.base_url % self.tab

    def do_query_test(self):
        # First query to set session variables
        self.client.get(self.url)

        with self.assertNumQueries(self.queries):
            self.client.get(self.url)
Nicolas Joyard's avatar
Nicolas Joyard committed
67 68 69 70


class ThemeBaseTest(BaseTest):
    tab = 'none'
Nicolas Joyard's avatar
Nicolas Joyard committed
71
    base_url = '/themes/etat-durgence/%s/'
Nicolas Joyard's avatar
Nicolas Joyard committed
72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88

    """
    Common queries plus:
    - 1 for the theme
    """
    queries = BaseTest.left_pane_queries + 1

    @property
    def url(self):
        return self.base_url % self.tab

    def do_query_test(self):
        # First query to set session variables
        self.client.get(self.url)

        with self.assertNumQueries(self.queries):
            self.client.get(self.url)
Nicolas Joyard's avatar
Nicolas Joyard committed
89 90 91 92


class DossierBaseTest(BaseTest):
    tab = 'none'
Nicolas Joyard's avatar
Nicolas Joyard committed
93
    base_url = '/dossiers/15409/%s/'
Nicolas Joyard's avatar
Nicolas Joyard committed
94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111

    """
    Common queries plus:
    - 1 for the dossier
    - 1 for related themes
    """
    queries = BaseTest.left_pane_queries + 2

    @property
    def url(self):
        return self.base_url % self.tab

    def do_query_test(self):
        # First query to set session variables
        self.client.get(self.url)

        with self.assertNumQueries(self.queries):
            self.client.get(self.url)