mdsfred/cashbook_planner
2
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages 1 Wiki Activity Actions

add booking with category + test, todo: booking of asset

Browse source
This commit is contained in:
Frederik Jaeckel 2024-03-03 21:54:16 +01:00
parent e51abf589b
commit b86e421298
6 changed files with 372 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

118
tests/planner.py
View file

@ -3,6 +3,7 @@
# The COPYRIGHT file at the top level of this repository contains the
# full copyright notices and license terms.
from decimal import Decimal
from unittest.mock import MagicMock
from trytond.tests.test_tryton import with_transaction
from trytond.pool import Pool
@ -19,6 +20,7 @@ class PlannerTestCase(object):
pool = Pool()
Book = pool.get('cashbook.book')
Planner = pool.get('cashbook.planner')
Party = pool.get('party.party')
types = self.prep_type()
company = self.prep_company()
@ -36,11 +38,18 @@ class PlannerTestCase(object):
}])
self.assertEqual(book.rec_name, 'Book 1 | 0.00 usd | Open')
category = self.prep_category()
party, = Party.create([{
'name': 'Party',
'addresses': [('create', [{}])]}])
job, = Planner.create([{
'cashbook': book.id,
'name': name,
'start_date': date(2022, 5, 1),
'bookingtype': 'out'}])
'bookingtype': 'out',
'category': category.id,
'party': party.id,
'subject': 'Booking text'}])
# check applied defaults
self.assertEqual(job.rec_name, 'Job 1')
self.assertEqual(job.start_date, date(2022, 5, 1))
@ -286,4 +295,111 @@ class PlannerTestCase(object):
IrDate.today = MagicMock(return_value=date.today())
@with_transaction()
def test_planner_cronjobs_booking_with_category(self):
""" create job, configure booking with category, run job
"""
pool = Pool()
Planner = pool.get('cashbook.planner')
IrDate = pool.get('ir.date')
Category = pool.get('cashbook.category')
Cashbook = pool.get('cashbook.book')
job = self.prep_create_job()
self.assertEqual(
job._compute_dates_by_rrule(
count=1, query_date=date(2022, 5, 1)), [
date(2022, 5, 1)])
IrDate.today = MagicMock(return_value=date(2022, 5, 24))
category, = Category.search([('name', '=', 'Cat1')])
Planner.write(*[
[job],
{
'name': 'Booking to category',
'amount': Decimal('10.0'),
'bookingtype': 'out',
'category': category.id,
'subject': 'booking ${month}/${year}, ${date}',
'wfcheck': True,
}])
self.assertEqual(job.rec_name, 'Booking to category')
self.assertEqual(job.cashbook.rec_name, 'Book 1 | 0.00 usd | Open')
self.assertEqual(len(job.cashbook.lines), 0)
job, = Planner.search([])
self.assertEqual(job.nextrun[0].date, date(2022, 6, 1))
IrDate.today = MagicMock(return_value=date(2022, 6, 1))
Planner.cronjob()
self.assertEqual(job.nextrun[0].date, date(2022, 7, 1))
# check cashbook
self.assertEqual(len(job.cashbook.lines), 1)
self.assertEqual(
job.cashbook.lines[0].rec_name,
"06/01/2022|Exp|-10.00 usd|booking " +
"${month}/${year}, ${date} [Cat1]")
self.assertEqual(job.cashbook.lines[0].state, 'check')
with Transaction().set_context({'date': date(2022, 6, 1)}):
cashbook, = Cashbook.browse([job.cashbook])
self.assertEqual(cashbook.rec_name, 'Book 1 | -10.00 usd | Open')
IrDate.today = MagicMock(return_value=date.today())
@with_transaction()
def test_planner_cronjobs_booking_transfer_nonasset_eur(self):
""" create job, configure transfer-booking to non-asset-cashbook,
run job
"""
pool = Pool()
Planner = pool.get('cashbook.planner')
IrDate = pool.get('ir.date')
Category = pool.get('cashbook.category')
Cashbook = pool.get('cashbook.book')
job = self.prep_create_job()
self.assertEqual(
job._compute_dates_by_rrule(
count=1, query_date=date(2022, 5, 1)), [
date(2022, 5, 1)])
IrDate.today = MagicMock(return_value=date(2022, 5, 24))
category, = Category.search([('name', '=', 'Cat1')])
Planner.write(*[
[job],
{
'name': 'Booking to category',
'amount': Decimal('10.0'),
'bookingtype': 'out',
'category': category.id,
'subject': 'booking ${month}/${year}, ${date}',
'wfcheck': True,
}])
self.assertEqual(job.rec_name, 'Booking to category')
self.assertEqual(job.cashbook.rec_name, 'Book 1 | 0.00 usd | Open')
self.assertEqual(len(job.cashbook.lines), 0)
job, = Planner.search([])
self.assertEqual(job.nextrun[0].date, date(2022, 6, 1))
IrDate.today = MagicMock(return_value=date(2022, 6, 1))
Planner.cronjob()
self.assertEqual(job.nextrun[0].date, date(2022, 7, 1))
# check cashbook
self.assertEqual(len(job.cashbook.lines), 1)
self.assertEqual(
job.cashbook.lines[0].rec_name,
"06/01/2022|Exp|-10.00 usd|booking " +
"${month}/${year}, ${date} [Cat1]")
self.assertEqual(job.cashbook.lines[0].state, 'check')
with Transaction().set_context({'date': date(2022, 6, 1)}):
cashbook, = Cashbook.browse([job.cashbook])
self.assertEqual(cashbook.rec_name, 'Book 1 | -10.00 usd | Open')
IrDate.today = MagicMock(return_value=date.today())
# end PlannerTestCase