def test_prepare_camadas_data(self, mock_orcado, mock_empenhado): mock_orcado.return_value = 'mock_o' mock_empenhado.return_value = 'mock_e' execs_2017 = mommy.make( Execucao, year=date(2017, 1, 1), _quantity=2) execs_2018 = mommy.make( Execucao, year=date(2018, 1, 1), _quantity=2) execucoes = Execucao.objects.all() serializer = GeologiaSerializer(execucoes) ret = serializer.prepare_data() expected = { 'orcado': ['mock_o', 'mock_o'], 'empenhado': ['mock_e', 'mock_e'], } assert expected == ret execs = [execs_2017, execs_2018] for exec_year, call in zip(execs, mock_orcado.mock_calls): assert set(exec_year) == set(call[1][0]) for exec_year, call in zip(execs, mock_empenhado.mock_calls): assert set(exec_year) == set(call[1][0])
def test_prepare_subfuncao_data(self, mock_orcado, mock_empenhado): mock_orcado.return_value = 'mock_o' mock_empenhado.return_value = 'mock_e' subfuncao_id = 1 execs_2017_p1 = mommy.make(Execucao, year=date(2017, 1, 1), subfuncao__id=subfuncao_id, _quantity=2) mommy.make(Execucao, year=date(2017, 1, 1), subfuncao__id=2, _quantity=2) execs_2018_p1 = mommy.make(Execucao, year=date(2018, 1, 1), subfuncao__id=subfuncao_id, _quantity=2) mommy.make(Execucao, year=date(2018, 1, 1), subfuncao__id=2, _quantity=2) execucoes = Execucao.objects.all() serializer = GeologiaSerializer(execucoes, subfuncao_id=subfuncao_id) ret = serializer.prepare_data(subfuncao_id=subfuncao_id) expected = { 'subfuncao_id': subfuncao_id, 'orcado': ['mock_o', 'mock_o'], 'empenhado': ['mock_e', 'mock_e'], } assert expected == ret execs = [execs_2017_p1, execs_2018_p1] for exec_year, call in zip(execs, mock_orcado.mock_calls): assert set(exec_year) == set(call[1][0]) for exec_year, call in zip(execs, mock_empenhado.mock_calls): assert set(exec_year) == set(call[1][0])