...
 
Commits (3)
......@@ -28,7 +28,7 @@ from parlamento.models import Comissao
class EventoManager(models.Manager):
def na_mesma_semana_que_o_dia(self, dia):
semana = dia.isocalendar()[1]
return self.filter(data__week=semana)
return self.filter(data__week=semana, data__year=dia.year)
@python_2_unicode_compatible
......
......@@ -48,3 +48,12 @@ class TesteAgendaManager(TestCase):
Evento.objects.na_mesma_semana_que_o_dia(
datetime.date(2018, 4, 6)),
)
def teste_nao_encontra_evento_do_ano_passado_na_mesma_semana(self):
evento_do_ano_passado = mommy.make("Evento",
data=datetime.date(2017, 4, 6))
self.assertNotIn(
evento_do_ano_passado,
Evento.objects.na_mesma_semana_que_o_dia(
datetime.date(2018, 4, 6)),
)