def test_casa_legislativa_periodos_semestrais(self):
retriever = utils.PeriodosRetriever(self.conv, models.SEMESTRE)
periodos = retriever.get_periodos()
self.assertEquals(len(periodos), 3)
d = periodos[0].ini
self.assertEqual(1989, d.year)
self.assertEqual(1, d.month)
d = periodos[0].fim
self.assertEqual(1989, d.year)
self.assertEqual(6, d.month)
d = periodos[1].ini
self.assertEqual(1989, d.year)
self.assertEqual(7, d.month)
d = periodos[1].fim
self.assertEqual(1989, d.year)
self.assertEqual(12, d.month)
d = periodos[2].ini
self.assertEqual(1990, d.year)
self.assertEqual(1, d.month)
d = periodos[2].fim
self.assertEqual(1990, d.year)
self.assertEqual(6, d.month)
self.assertEqual(periodos[0].string, '1989 1o Semestre')
self.assertEqual(periodos[1].string, '1989 2o Semestre')
self.assertEqual(periodos[2].string, '1990 1o Semestre')
def test_legislative_house_semiannual_periods(self):
# Stores the objects of conv and the semester from models
retriever = utils.PeriodosRetriever(self.conv, models.SEMESTRE)
# Stores the retriever periods
retriever_periods = retriever.get_periods()
self.assertEquals(len(retriever_periods), 2)
# Temporary variable that stores the initial period of conv legislative house
conv_period = retriever_periods[0].ini
self.assertEqual(1989, conv_period.year)
self.assertEqual(1, conv_period.month)
# Temporary variable that stores the final period of conv legislative house
conv_period = retriever_periods[0].fim
self.assertEqual(1989, conv_period.year)
self.assertEqual(6, conv_period.month)
# Temporary variable that stores the initial period of conv legislative house
conv_period = retriever_periods[1].ini
self.assertEqual(1989, conv_period.year)
self.assertEqual(7, conv_period.month)
# Temporary variable that stores the final period of conv legislative house
conv_period = retriever_periods[1].fim
self.assertEqual(1989, conv_period.year)
self.assertEqual(12, conv_period.month)
self.assertEqual(retriever_periods[0].string, '1989 1o Semestre')
self.assertEqual(retriever_periods[1].string, '1989 2o Semestre')
def test_casa_legislativa_periodos_anuais(self):
retriever = utils.PeriodosRetriever(self.conv, models.ANO)
periodos = retriever.get_periodos()
self.assertEquals(len(periodos), 2)
self.assertEqual(periodos[0].string, '1989')
self.assertEqual(periodos[1].string, '1990')
self.assertEqual(periodos[0].quantidade_votacoes, 8)
self.assertEqual(periodos[1].quantidade_votacoes, 1)
def test_casa_legislativa_periodos_mensais(self):
retriever = utils.PeriodosRetriever(self.conv, models.MES)
periodos = retriever.get_periodos()
self.assertEquals(len(periodos), 2)
self.assertEqual(periodos[0].string, '1989 Fev')
self.assertEqual(periodos[0].quantidade_votacoes, 4)
self.assertEqual(periodos[1].string, '1989 Out')
self.assertEqual(periodos[1].quantidade_votacoes, 4)
def test_legislative_house_annual_periods(self):
# Stores the objects of conv and the year from models
retriever = utils.PeriodosRetriever(self.conv, models.ANO)
# Stores the retriever periods
retriever_periods = retriever.get_periods()
self.assertEquals(len(retriever_periods), 1)
self.assertEqual(retriever_periods[0].string, '1989')
self.assertEqual(retriever_periods[0].quantidade_votacoes, 8)
def test_legislative_house_periods_with_no_votes_list(self):
# Receives objects of legislative house where name is equal to casa_nova to
# perform home test without legislative voting list
new_house = models.CasaLegislativa(nome="Casa Nova")
# Stores the objects of new house and the year from models
retriever = utils.PeriodosRetriever(new_house, models.ANO)
# Stores the retriever periods
retriever_periods = retriever.get_periods()
self.assertEquals(len(retriever_periods), 0)
def test_legislative_house_monthly_periods(self):
# Stores the objects of conv and the month from models
retriever = utils.PeriodosRetriever(self.conv, models.MES)
# Stores the retriever periods
retriever_periods = retriever.get_periods()
self.assertEquals(len(retriever_periods), 2)
self.assertEqual(retriever_periods[0].string, '1989 Fev')
self.assertEqual(retriever_periods[0].quantidade_votacoes, 4)
self.assertEqual(retriever_periods[1].string, '1989 Out')
self.assertEqual(retriever_periods[1].quantidade_votacoes, 4)
def _test_periods_in_two_dates(self, ano_ini, ano_fim, esfera,
periodicidade, expected_periodos_len):
# Receives the start date for testing
A_DATE = datetime.date(ano_ini, 02, 02)
# Receives the final date for testing
OTHER_DATE = datetime.date(ano_fim, 10, 02)
# Receives all objects of the Voting from models
votings = models.Votacao.objects.all()
# Receives half of the number of votes
half_of_votings_amount = len(votings) / 2
# Receives original dates of periods
original_dates = {}
# Receives original sphere of conv legislative house
original_sphere = self.conv.esfera
self.conv.esfera = esfera
for i in range(0, half_of_votings_amount):
voting_list = votings[i]
original_dates[voting_list.id] = voting_list.data
voting_list.data = A_DATE
voting_list.save()
for i in range(half_of_votings_amount, len(votings)):
voting_list = votings[i]
original_dates[voting_list.id] = voting_list.data
voting_list.data = OTHER_DATE
voting_list.save()
# Stores the objects of conv and the periodicity
retriever = utils.PeriodosRetriever(self.conv, periodicidade)
# Stores the retriever periods
retriever_periods = retriever.get_periods()
self.assertEquals(len(retriever_periods), expected_periodos_len)
self._restore(original_sphere, votings, original_dates)
def _test_periodos_em_duas_datas(self, ano_ini, ano_fim, esfera, periodicidade, expected_periodos_len):
UMA_DATA = datetime.date(ano_ini, 02, 02)
OUTRA_DATA = datetime.date(ano_fim, 10, 02)
votacoes = models.Votacao.objects.all()
half = len(votacoes) / 2
datas_originais = {} # votacao.id => data
esfera_original = self.conv.esfera
self.conv.esfera = esfera
for i in range(0, half):
v = votacoes[i]
datas_originais[v.id] = v.data
v.data = UMA_DATA
v.save()
for i in range(half, len(votacoes)):
v = votacoes[i]
datas_originais[v.id] = v.data
v.data = OUTRA_DATA
v.save()
retriever = utils.PeriodosRetriever(self.conv, periodicidade)
periodos = retriever.get_periodos()
self.assertEquals(len(periodos), expected_periodos_len)
self._restore(esfera_original, votacoes, datas_originais)
def test_casa_legislativa_periodos_sem_lista_votacoes(self):
casa_nova = models.CasaLegislativa(nome="Casa Nova")
retriever = utils.PeriodosRetriever(casa_nova, models.ANO)
periodos = retriever.get_periodos()
self.assertEquals(len(periodos), 0)