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'))