def test_serialize_list_of_subfuncoes(self): subfuncao_1 = mommy.make('Subfuncao', desc='Turismo') subfuncao_2 = mommy.make('Subfuncao', desc='Educação Infantil') subfuncoes = [subfuncao_1, subfuncao_2] expected = [ dict(id=subfuncao.id, desc=subfuncao.desc, selecionado=False) for subfuncao in subfuncoes ] # selected funcao subfuncao_3 = mommy.make('Subfuncao', desc='Selected Function') expected.append( dict(id=subfuncao_3.id, desc=subfuncao_3.desc, selecionado=True)) expected.sort(key=lambda s: s['desc']) mommy.make(Execucao, subfuncao=cycle([subfuncao_1, subfuncao_2, subfuncao_3]), _quantity=3) queryset = Execucao.objects.all() # not expected mommy.make('Subfuncao') serialized = GeologiaSerializer(queryset, subfuncao_id=subfuncao_3.id).data assert expected == serialized.get('subfuncoes')
def test_serialize_list_of_gnds(self): gnd_1 = mommy.make('GndGeologia', desc='Consultoria', slug='consulting') gnd_2 = mommy.make('GndGeologia', desc='Custeio operacional', slug='operational') gnds = [gnd_1, gnd_2] expected = [dict(slug=gnd.slug, desc=gnd.desc) for gnd in gnds] queryset = Execucao.objects.all() serialized = GeologiaSerializer(queryset).data assert expected == serialized.get('gnds')