def test_get_subgrupo_year_empenhado_data(self, mock_empenhado): mock_empenhado.return_value = 'mock_e' 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_empenhado_data(execucoes) expected = { 'year': year.strftime('%Y'), 'subgrupos': ['mock_e', 'mock_e'], } assert expected == ret execs = [execs_sub1, execs_sub2] for exec_subgrupo, call in zip(execs, mock_empenhado.mock_calls): assert set(exec_subgrupo) == set(call[1][0])
def test_get_subgrupo_year_empenhado_data(self, mock_empenhado): mock_empenhado.return_value = 'mock_e' 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_empenhado_data(execucoes) expected = { 'year': year.strftime('%Y'), 'subgrupos': ['mock_e', 'mock_e'], } assert expected == ret execs = [execs_sub2, execs_sub1] # Alphabetical order for exec_subgrupo, call in zip(execs, mock_empenhado.mock_calls): assert set(exec_subgrupo) == set(call[1][0])