def test_coalition__multiple_coalitions_in_different_districts(self): from nuorisovaalit.models import Coalition from nuorisovaalit.models import District from nuorisovaalit.models import Party district_x = District(u'District X', 1) district_y = District(u'District Y', 2) party = Party(u'Foobar') session = DBSession() session.add_all([district_x, district_y, party]) session.flush() coalition_x = Coalition(u'Reds', district_x) coalition_y = Coalition(u'Blues', district_y) session.add_all([coalition_x, coalition_y]) coalition_x.parties.append(party) coalition_y.parties.append(party) session.flush() self.assertEquals(coalition_x, party.coalition(district_x)) self.assertEquals(coalition_y, party.coalition(district_y))
def test_coalition__no_coalition(self): from nuorisovaalit.models import District from nuorisovaalit.models import Party district = District(u'District X', 1) party = Party(u'Foobar') session = DBSession() session.add_all([district, party]) session.flush() self.assertEquals(None, party.coalition(district))
def test_coalition__single_coalition(self): from nuorisovaalit.models import Coalition from nuorisovaalit.models import District from nuorisovaalit.models import Party district = District(u'District X', 1) party = Party(u'Foobar') session = DBSession() session.add_all([district, party]) session.flush() coalition = Coalition(u'Reds', district) session.add(coalition) coalition.parties.append(party) session.flush() self.assertEquals(coalition, party.coalition(district))