def initialize(self): self.initialized = True self.next_invoice_process_time = datetime.datetime.now() InvoicePrototype.reset_all() self.logger.info('BANK PROCESSOR INITIALIZED')
def test_reset_all(self): for state in INVOICE_STATE.records: invoice = self.create_invoice() invoice.state = state invoice.save() InvoicePrototype.reset_all() self.assertEqual(InvoicePrototype._model_class.objects.filter(state=INVOICE_STATE.RESETED).count(), len(INVOICE_STATE.records) - 4) self.assertEqual(InvoicePrototype._model_class.objects.filter(state=INVOICE_STATE.CONFIRMED).count(), 1) self.assertEqual(InvoicePrototype._model_class.objects.filter(state=INVOICE_STATE.CANCELED).count(), 1) self.assertEqual(InvoicePrototype._model_class.objects.filter(state=INVOICE_STATE.REJECTED).count(), 1) self.assertEqual(InvoicePrototype._model_class.objects.filter(state=INVOICE_STATE.FORCED).count(), 1)
def test_reset_all(self): for state in INVOICE_STATE.records: invoice = self.create_invoice() invoice.state = state invoice.save() InvoicePrototype.reset_all() self.assertEqual( InvoicePrototype._model_class.objects.filter( state=INVOICE_STATE.RESETED).count(), len(INVOICE_STATE.records) - 4) self.assertEqual( InvoicePrototype._model_class.objects.filter( state=INVOICE_STATE.CONFIRMED).count(), 1) self.assertEqual( InvoicePrototype._model_class.objects.filter( state=INVOICE_STATE.CANCELED).count(), 1) self.assertEqual( InvoicePrototype._model_class.objects.filter( state=INVOICE_STATE.REJECTED).count(), 1) self.assertEqual( InvoicePrototype._model_class.objects.filter( state=INVOICE_STATE.FORCED).count(), 1)