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))
