def test_get_subgrupo_year_orcado_data(self, mock_orcado):
mock_orcado.return_value = 'mock_o'
year = date(2018, 1, 1)
execs_sub1 = mommy.make(
Execucao,
year=year,
subgrupo__id=1,
_quantity=2)
execs_sub2 = mommy.make(
Execucao,
year=year,
subgrupo__id=2,
_quantity=2)
execucoes = Execucao.objects.all()
serializer = GeologiaSerializer([])
ret = serializer.get_subgrupo_year_orcado_data(execucoes)
expected = {
'year': year.strftime('%Y'),
'subgrupos': ['mock_o', 'mock_o'],
}
assert expected == ret
execs = [execs_sub1, execs_sub2]
for exec_subgrupo, call in zip(execs, mock_orcado.mock_calls):
assert set(exec_subgrupo) == set(call[1][0])
def test_get_subgrupo_year_orcado_data(self, mock_orcado):
mock_orcado.return_value = 'mock_o'
subgrupo1 = mommy.make(Subgrupo, desc="Uniceu")
subgrupo2 = mommy.make(Subgrupo, desc="Alimentação Escolar")
year = date(2018, 1, 1)
execs_sub1 = mommy.make(Execucao,
year=year,
subgrupo=subgrupo1,
_quantity=2)
execs_sub2 = mommy.make(Execucao,
year=year,
subgrupo=subgrupo2,
_quantity=2)
execucoes = Execucao.objects.all()
serializer = GeologiaSerializer([])
ret = serializer.get_subgrupo_year_orcado_data(execucoes)
expected = {
'year': year.strftime('%Y'),
'subgrupos': ['mock_o', 'mock_o'],
}
assert expected == ret
execs = [execs_sub2, execs_sub1] # Alphabetical order
for exec_subgrupo, call in zip(execs, mock_orcado.mock_calls):
assert set(exec_subgrupo) == set(call[1][0])