def test_next_invoice_number_used(self):
from seantisinvoice.models import DBSession
from seantisinvoice.models import Company
from seantisinvoice.models import Invoice
from seantisinvoice.models import next_invoice_number
session = DBSession()
company = session.query(Company).one()
company.invoice_start_number = 20
invoice = Invoice(invoice_number=20)
session.add(invoice)
session.flush()
result = next_invoice_number()
self.assertEqual(21, result)
self.assertEqual(22, company.invoice_start_number)
def test_next_invoice_number_used(self):
from seantisinvoice.models import DBSession
from seantisinvoice.models import Company
from seantisinvoice.models import Invoice
from seantisinvoice.models import next_invoice_number
session = DBSession()
company = session.query(Company).one()
company.invoice_start_number = 20
invoice = Invoice(invoice_number=20)
session.add(invoice)
session.flush()
result = next_invoice_number()
self.assertEqual(21, result)
self.assertEqual(22, company.invoice_start_number)
def _add_customer(self):
from seantisinvoice.models import Customer
from seantisinvoice.models import CustomerContact
from seantisinvoice.models import DBSession
session = DBSession()
customer = Customer()
customer.name = u'Customers Inc.'
customer.address1 = u'Street'
customer.postal_code = u'12234'
customer.city = u'Dublin'
session.add(customer)
# Each customer needs at least one contact
contact = CustomerContact()
contact.first_name = u'Buck'
contact.last_name = u'Mulligan'
contact.title = u'Mr.'
contact.customer = customer
session.add(contact)
session.flush()
return customer
def _add_customer(self):
from seantisinvoice.models import Customer
from seantisinvoice.models import CustomerContact
from seantisinvoice.models import DBSession
session = DBSession()
customer = Customer()
customer.name = u"Customers Inc."
customer.address1 = u"Street"
customer.postal_code = u"12234"
customer.city = u"Dublin"
session.add(customer)
# Each customer needs at least one contact
contact = CustomerContact()
contact.first_name = u"Buck"
contact.last_name = u"Mulligan"
contact.title = u"Mr."
contact.customer = customer
session.add(contact)
session.flush()
return customer
def _add_invoice(self):
from seantisinvoice.models import Invoice
from seantisinvoice.models import InvoiceItem
from seantisinvoice.models import DBSession
session = DBSession()
customer = self._add_customer()
invoice = Invoice()
invoice.project_description = u'Project'
invoice.date = datetime.date.today()
invoice.due_date = invoice.date + datetime.timedelta(days=30)
invoice.currency = u'CHF'
invoice.contact = customer.contacts[0]
session.add(invoice)
# Add invoice item to the invoice
item = InvoiceItem()
item.item_number = 0
item.service_title = u'Testing'
item.service_description = u'A lot of work!'
item.amount = 1000.0
item.invoice = invoice
session.add(item)
session.flush()
return invoice
def _add_invoice(self):
from seantisinvoice.models import Invoice
from seantisinvoice.models import InvoiceItem
from seantisinvoice.models import DBSession
session = DBSession()
customer = self._add_customer()
invoice = Invoice()
invoice.project_description = u"Project"
invoice.date = datetime.date.today()
invoice.due_date = invoice.date + datetime.timedelta(days=30)
invoice.currency = u"CHF"
invoice.contact = customer.contacts[0]
session.add(invoice)
# Add invoice item to the invoice
item = InvoiceItem()
item.item_number = 0
item.service_title = u"Testing"
item.service_description = u"A lot of work!"
item.amount = 1000.0
item.invoice = invoice
session.add(item)
session.flush()
return invoice
