def test_cria_entrada_extrato_cc(self):
exf = ExtratoFinanceiro.objects.get(pk=1)
self.assertIsNone(exf.entrada_extrato_cc)
retorno = ExtratoFinanceiro._insere_extrato_cc(exf)
self.assertEqual(retorno, 1)
self.assertIsNotNone(exf.entrada_extrato_cc)
def tool_save_and_insert_cc(self, request, obj, button):
"""
Tratamento do novo botão de "Salvar e inserir extrato c/c"
"""
retorno = ExtratoFinanceiro._insere_extrato_cc(obj)
if retorno == 1:
messages.add_message(request, messages.SUCCESS, u"Extrato de conta corrente inserido com sucesso.")
# como o objeto teve um extratoCC inserido, salvamos novamente o objeto.
obj.save()
elif retorno == 2:
messages.add_message(request, messages.WARNING, u"Extrato de conta corrente já existente.")
else:
messages.add_message(request, messages.WARNING, u"Extrato de conta corrente não inserido.")
return HttpResponseRedirect(reverse('admin:financeiro_extratofinanceiro_changelist'))
def test_delete_cascade_entrada_extrato_cc(self):
exf = ExtratoFinanceiro.objects.get(pk=1)
self.assertIsNone(exf.entrada_extrato_cc)
# verificando se foi inserido no ExtratoCC
retorno = ExtratoFinanceiro._insere_extrato_cc(exf)
self.assertEqual(retorno, 1)
self.assertIsNotNone(exf.entrada_extrato_cc)
# removendo o objeto do BD
exf.entrada_extrato_cc.delete()
# verificando se o extratoCC foi removido do BD
extratoccs = ExtratoCC.objects.filter(pk=exf.entrada_extrato_cc_id)
self.assertEqual(len(extratoccs), 0)
# verificando se o ExtratoFinanceiro não foi removido do BD com CASCADE
extratofinanceiro = ExtratoCC.objects.filter(pk=1)
self.assertIsNotNone(extratofinanceiro)
def tool_save_and_insert_cc(self, request, obj, button):
"""
Tratamento do novo botão de "Salvar e inserir extrato c/c"
"""
retorno = ExtratoFinanceiro._insere_extrato_cc(obj)
if retorno == 1:
messages.add_message(
request, messages.SUCCESS,
u"Extrato de conta corrente inserido com sucesso.")
# como o objeto teve um extratoCC inserido, salvamos novamente o objeto.
obj.save()
elif retorno == 2:
messages.add_message(request, messages.WARNING,
u"Extrato de conta corrente já existente.")
else:
messages.add_message(request, messages.WARNING,
u"Extrato de conta corrente não inserido.")
return HttpResponseRedirect(
reverse('admin:financeiro_extratofinanceiro_changelist'))
