693 lines
34 KiB
Python
693 lines
34 KiB
Python
|
# The Mia! Accounting Project.
|
||
|
|
# Author: imacat@mail.imacat.idv.tw (imacat), 2023/3/11
|
||
|
|
|
||
|
|
# Copyright (c) 2023 imacat.
|
||
|
|
#
|
||
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
# you may not use this file except in compliance with the License.
|
||
|
|
# You may obtain a copy of the License at
|
||
|
|
#
|
||
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
#
|
||
|
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
# See the License for the specific language governing permissions and
|
||
|
|
# limitations under the License.
|
||
|
|
"""The test for the offset matcher.
|
||
|
|
|
||
|
|
"""
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
import httpx
|
||
|
|
from click.testing import Result
|
||
|
|
from flask import Flask
|
||
|
|
from flask.testing import FlaskCliRunner
|
||
|
|
|
||
|
|
from test_site import db
|
||
|
|
from testlib import create_test_app, get_client, Accounts
|
||
|
|
from testlib_journal_entry import match_journal_entry_detail
|
||
|
|
from testlib_offset import JournalEntryData, CurrencyData, \
|
||
|
|
JournalEntryLineItemData
|
||
|
|
|
||
|
|
|
||
|
|
class OffsetMatcherTestCase(unittest.TestCase):
|
||
|
|
"""The offset matcher test case."""
|
||
|
|
|
||
|
|
def setUp(self) -> None:
|
||
|
|
"""Sets up the test.
|
||
|
|
This is run once per test.
|
||
|
|
|
||
|
|
:return: None.
|
||
|
|
"""
|
||
|
|
self.app: Flask = create_test_app()
|
||
|
|
|
||
|
|
runner: FlaskCliRunner = self.app.test_cli_runner()
|
||
|
|
with self.app.app_context():
|
||
|
|
from accounting.models import BaseAccount, JournalEntry, \
|
||
|
|
JournalEntryLineItem
|
||
|
|
result: Result
|
||
|
|
result = runner.invoke(args="init-db")
|
||
|
|
self.assertEqual(result.exit_code, 0)
|
||
|
|
if BaseAccount.query.first() is None:
|
||
|
|
result = runner.invoke(args="accounting-init-base")
|
||
|
|
self.assertEqual(result.exit_code, 0)
|
||
|
|
result = runner.invoke(args=["accounting-init-currencies",
|
||
|
|
"-u", "editor"])
|
||
|
|
self.assertEqual(result.exit_code, 0)
|
||
|
|
result = runner.invoke(args=["accounting-init-accounts",
|
||
|
|
"-u", "editor"])
|
||
|
|
self.assertEqual(result.exit_code, 0)
|
||
|
|
JournalEntry.query.delete()
|
||
|
|
JournalEntryLineItem.query.delete()
|
||
|
|
|
||
|
|
self.client, self.csrf_token = get_client(self.app, "editor")
|
||
|
|
|
||
|
|
def test_different(self) -> None:
|
||
|
|
"""Tests to match against different descriptions and amounts.
|
||
|
|
|
||
|
|
:return: None.
|
||
|
|
"""
|
||
|
|
from accounting.models import Account, JournalEntryLineItem
|
||
|
|
from accounting.utils.offset_matcher import OffsetMatcher
|
||
|
|
data: DifferentTestData \
|
||
|
|
= DifferentTestData(self.app, self.client, self.csrf_token)
|
||
|
|
account: Account | None
|
||
|
|
line_item: JournalEntryLineItem | None
|
||
|
|
matcher: OffsetMatcher
|
||
|
|
list_uri: str
|
||
|
|
response: httpx.Response
|
||
|
|
|
||
|
|
# The receivables
|
||
|
|
with self.app.app_context():
|
||
|
|
account = Account.find_by_code(Accounts.RECEIVABLE)
|
||
|
|
assert account is not None
|
||
|
|
matcher = OffsetMatcher(account)
|
||
|
|
self.assertEqual({x.id for x in matcher.unapplied},
|
||
|
|
{data.l_r_or1d.id, data.l_r_or2d.id,
|
||
|
|
data.l_r_or3d.id, data.l_r_or4d.id})
|
||
|
|
self.assertEqual({x.id for x in matcher.unmatched_offsets},
|
||
|
|
{data.l_r_of1c.id, data.l_r_of2c.id,
|
||
|
|
data.l_r_of3c.id, data.l_r_of4c.id,
|
||
|
|
data.l_r_of5c.id})
|
||
|
|
self.assertEqual({(x.original_line_item.id, x.offset.id)
|
||
|
|
for x in matcher.matched_pairs},
|
||
|
|
{(data.l_r_or4d.id, data.l_r_of5c.id)})
|
||
|
|
for line_item_id in {data.l_r_of1c.id, data.l_r_of2c.id,
|
||
|
|
data.l_r_of3c.id, data.l_r_of4c.id,
|
||
|
|
data.l_r_of5c.id}:
|
||
|
|
line_item = db.session.get(JournalEntryLineItem, line_item_id)
|
||
|
|
self.assertIsNotNone(line_item)
|
||
|
|
self.assertIsNone(line_item.original_line_item_id)
|
||
|
|
|
||
|
|
list_uri = f"/accounting/unmatched-offsets/{Accounts.RECEIVABLE}"
|
||
|
|
response = self.client.post(list_uri,
|
||
|
|
data={"csrf_token": self.csrf_token})
|
||
|
|
self.assertEqual(response.status_code, 302)
|
||
|
|
self.assertEqual(response.headers["Location"], list_uri)
|
||
|
|
|
||
|
|
with self.app.app_context():
|
||
|
|
account = Account.find_by_code(Accounts.RECEIVABLE)
|
||
|
|
assert account is not None
|
||
|
|
matcher = OffsetMatcher(account)
|
||
|
|
self.assertEqual({x.id for x in matcher.unapplied},
|
||
|
|
{data.l_r_or1d.id, data.l_r_or2d.id,
|
||
|
|
data.l_r_or3d.id})
|
||
|
|
self.assertEqual({x.id for x in matcher.unmatched_offsets},
|
||
|
|
{data.l_r_of1c.id, data.l_r_of2c.id,
|
||
|
|
data.l_r_of3c.id, data.l_r_of4c.id})
|
||
|
|
self.assertEqual(matcher.matches, 0)
|
||
|
|
for line_item_id in {data.l_r_of1c.id, data.l_r_of2c.id,
|
||
|
|
data.l_r_of3c.id, data.l_r_of4c.id}:
|
||
|
|
line_item = db.session.get(JournalEntryLineItem, line_item_id)
|
||
|
|
self.assertIsNotNone(line_item)
|
||
|
|
self.assertIsNone(line_item.original_line_item_id)
|
||
|
|
line_item = db.session.get(JournalEntryLineItem, data.l_r_of5c.id)
|
||
|
|
self.assertIsNotNone(line_item)
|
||
|
|
self.assertIsNotNone(line_item.original_line_item_id)
|
||
|
|
self.assertEqual(line_item.original_line_item_id, data.l_r_or4d.id)
|
||
|
|
|
||
|
|
# The payables
|
||
|
|
with self.app.app_context():
|
||
|
|
account = Account.find_by_code(Accounts.PAYABLE)
|
||
|
|
assert account is not None
|
||
|
|
matcher = OffsetMatcher(account)
|
||
|
|
self.assertEqual({x.id for x in matcher.unapplied},
|
||
|
|
{data.l_p_or1c.id, data.l_p_or2c.id,
|
||
|
|
data.l_p_or3c.id, data.l_p_or4c.id})
|
||
|
|
self.assertEqual({x.id for x in matcher.unmatched_offsets},
|
||
|
|
{data.l_p_of1d.id, data.l_p_of2d.id,
|
||
|
|
data.l_p_of3d.id, data.l_p_of4d.id,
|
||
|
|
data.l_p_of5d.id})
|
||
|
|
self.assertEqual({(x.original_line_item.id, x.offset.id)
|
||
|
|
for x in matcher.matched_pairs},
|
||
|
|
{(data.l_p_or4c.id, data.l_p_of5d.id)})
|
||
|
|
for line_item_id in {data.l_p_of1d.id, data.l_p_of2d.id,
|
||
|
|
data.l_p_of3d.id, data.l_p_of4d.id,
|
||
|
|
data.l_p_of5d.id}:
|
||
|
|
line_item = db.session.get(JournalEntryLineItem, line_item_id)
|
||
|
|
self.assertIsNotNone(line_item)
|
||
|
|
self.assertIsNone(line_item.original_line_item_id)
|
||
|
|
|
||
|
|
list_uri = f"/accounting/unmatched-offsets/{Accounts.PAYABLE}"
|
||
|
|
response = self.client.post(list_uri,
|
||
|
|
data={"csrf_token": self.csrf_token})
|
||
|
|
self.assertEqual(response.status_code, 302)
|
||
|
|
self.assertEqual(response.headers["Location"], list_uri)
|
||
|
|
|
||
|
|
with self.app.app_context():
|
||
|
|
account = Account.find_by_code(Accounts.PAYABLE)
|
||
|
|
assert account is not None
|
||
|
|
matcher = OffsetMatcher(account)
|
||
|
|
self.assertEqual({x.id for x in matcher.unapplied},
|
||
|
|
{data.l_p_or1c.id, data.l_p_or2c.id,
|
||
|
|
data.l_p_or3c.id})
|
||
|
|
self.assertEqual({x.id for x in matcher.unmatched_offsets},
|
||
|
|
{data.l_p_of1d.id, data.l_p_of2d.id,
|
||
|
|
data.l_p_of3d.id, data.l_p_of4d.id})
|
||
|
|
self.assertEqual(matcher.matches, 0)
|
||
|
|
for line_item_id in {data.l_p_of1d.id, data.l_p_of2d.id,
|
||
|
|
data.l_p_of3d.id, data.l_p_of4d.id}:
|
||
|
|
line_item = db.session.get(JournalEntryLineItem, line_item_id)
|
||
|
|
self.assertIsNotNone(line_item)
|
||
|
|
self.assertIsNone(line_item.original_line_item_id)
|
||
|
|
line_item = db.session.get(JournalEntryLineItem, data.l_p_of5d.id)
|
||
|
|
self.assertIsNotNone(line_item)
|
||
|
|
self.assertIsNotNone(line_item.original_line_item_id)
|
||
|
|
self.assertEqual(line_item.original_line_item_id, data.l_p_or4c.id)
|
||
|
|
|
||
|
|
def test_same(self) -> None:
|
||
|
|
"""Tests to match against same descriptions and amounts.
|
||
|
|
|
||
|
|
:return: None.
|
||
|
|
"""
|
||
|
|
from accounting.models import Account, JournalEntryLineItem
|
||
|
|
from accounting.utils.offset_matcher import OffsetMatcher
|
||
|
|
data: SameTestData \
|
||
|
|
= SameTestData(self.app, self.client, self.csrf_token)
|
||
|
|
account: Account | None
|
||
|
|
line_item: JournalEntryLineItem | None
|
||
|
|
matcher: OffsetMatcher
|
||
|
|
list_uri: str
|
||
|
|
response: httpx.Response
|
||
|
|
|
||
|
|
# The receivables
|
||
|
|
with self.app.app_context():
|
||
|
|
account = Account.find_by_code(Accounts.RECEIVABLE)
|
||
|
|
assert account is not None
|
||
|
|
matcher = OffsetMatcher(account)
|
||
|
|
self.assertEqual({x.id for x in matcher.unapplied},
|
||
|
|
{data.l_r_or1d.id, data.l_r_or3d.id,
|
||
|
|
data.l_r_or4d.id, data.l_r_or5d.id,
|
||
|
|
data.l_r_or6d.id})
|
||
|
|
self.assertEqual({x.id for x in matcher.unmatched_offsets},
|
||
|
|
{data.l_r_of1c.id, data.l_r_of2c.id,
|
||
|
|
data.l_r_of4c.id, data.l_r_of5c.id,
|
||
|
|
data.l_r_of6c.id})
|
||
|
|
self.assertEqual({(x.original_line_item.id, x.offset.id)
|
||
|
|
for x in matcher.matched_pairs},
|
||
|
|
{(data.l_r_or1d.id, data.l_r_of2c.id),
|
||
|
|
(data.l_r_or3d.id, data.l_r_of4c.id),
|
||
|
|
(data.l_r_or4d.id, data.l_r_of6c.id)})
|
||
|
|
for line_item_id in {data.l_r_of1c.id, data.l_r_of2c.id,
|
||
|
|
data.l_r_of4c.id, data.l_r_of5c.id,
|
||
|
|
data.l_r_of6c.id}:
|
||
|
|
line_item = db.session.get(JournalEntryLineItem, line_item_id)
|
||
|
|
self.assertIsNotNone(line_item)
|
||
|
|
self.assertIsNone(line_item.original_line_item_id)
|
||
|
|
line_item = db.session.get(JournalEntryLineItem, data.l_r_of3c.id)
|
||
|
|
self.assertIsNotNone(line_item)
|
||
|
|
self.assertIsNotNone(line_item.original_line_item_id)
|
||
|
|
self.assertEqual(line_item.original_line_item_id, data.l_r_or2d.id)
|
||
|
|
|
||
|
|
list_uri = f"/accounting/unmatched-offsets/{Accounts.RECEIVABLE}"
|
||
|
|
response = self.client.post(list_uri,
|
||
|
|
data={"csrf_token": self.csrf_token})
|
||
|
|
self.assertEqual(response.status_code, 302)
|
||
|
|
self.assertEqual(response.headers["Location"], list_uri)
|
||
|
|
|
||
|
|
with self.app.app_context():
|
||
|
|
account = Account.find_by_code(Accounts.RECEIVABLE)
|
||
|
|
assert account is not None
|
||
|
|
matcher = OffsetMatcher(account)
|
||
|
|
self.assertEqual({x.id for x in matcher.unapplied},
|
||
|
|
{data.l_r_or5d.id, data.l_r_or6d.id})
|
||
|
|
self.assertEqual({x.id for x in matcher.unmatched_offsets},
|
||
|
|
{data.l_r_of1c.id, data.l_r_of5c.id})
|
||
|
|
self.assertEqual(matcher.matches, 0)
|
||
|
|
for line_item_id in {data.l_r_of1c.id, data.l_r_of5c.id}:
|
||
|
|
line_item = db.session.get(JournalEntryLineItem, line_item_id)
|
||
|
|
self.assertIsNotNone(line_item)
|
||
|
|
self.assertIsNone(line_item.original_line_item_id)
|
||
|
|
line_item = db.session.get(JournalEntryLineItem, data.l_r_of2c.id)
|
||
|
|
self.assertIsNotNone(line_item)
|
||
|
|
self.assertIsNotNone(line_item.original_line_item_id)
|
||
|
|
self.assertEqual(line_item.original_line_item_id, data.l_r_or1d.id)
|
||
|
|
line_item = db.session.get(JournalEntryLineItem, data.l_r_of3c.id)
|
||
|
|
self.assertIsNotNone(line_item)
|
||
|
|
self.assertIsNotNone(line_item.original_line_item_id)
|
||
|
|
self.assertEqual(line_item.original_line_item_id, data.l_r_or2d.id)
|
||
|
|
line_item = db.session.get(JournalEntryLineItem, data.l_r_of4c.id)
|
||
|
|
self.assertIsNotNone(line_item)
|
||
|
|
self.assertIsNotNone(line_item.original_line_item_id)
|
||
|
|
self.assertEqual(line_item.original_line_item_id, data.l_r_or3d.id)
|
||
|
|
line_item = db.session.get(JournalEntryLineItem, data.l_r_of6c.id)
|
||
|
|
self.assertIsNotNone(line_item)
|
||
|
|
self.assertIsNotNone(line_item.original_line_item_id)
|
||
|
|
self.assertEqual(line_item.original_line_item_id, data.l_r_or4d.id)
|
||
|
|
|
||
|
|
# The payables
|
||
|
|
with self.app.app_context():
|
||
|
|
account = Account.find_by_code(Accounts.PAYABLE)
|
||
|
|
assert account is not None
|
||
|
|
matcher = OffsetMatcher(account)
|
||
|
|
self.assertEqual({x.id for x in matcher.unapplied},
|
||
|
|
{data.l_p_or1c.id, data.l_p_or3c.id,
|
||
|
|
data.l_p_or4c.id, data.l_p_or5c.id,
|
||
|
|
data.l_p_or6c.id})
|
||
|
|
self.assertEqual({x.id for x in matcher.unmatched_offsets},
|
||
|
|
{data.l_p_of1d.id, data.l_p_of2d.id,
|
||
|
|
data.l_p_of4d.id, data.l_p_of5d.id,
|
||
|
|
data.l_p_of6d.id})
|
||
|
|
self.assertEqual({(x.original_line_item.id, x.offset.id)
|
||
|
|
for x in matcher.matched_pairs},
|
||
|
|
{(data.l_p_or1c.id, data.l_p_of2d.id),
|
||
|
|
(data.l_p_or3c.id, data.l_p_of4d.id),
|
||
|
|
(data.l_p_or4c.id, data.l_p_of6d.id)})
|
||
|
|
for line_item_id in {data.l_p_of1d.id, data.l_p_of2d.id,
|
||
|
|
data.l_p_of4d.id, data.l_p_of5d.id,
|
||
|
|
data.l_p_of6d.id}:
|
||
|
|
line_item = db.session.get(JournalEntryLineItem, line_item_id)
|
||
|
|
self.assertIsNotNone(line_item)
|
||
|
|
self.assertIsNone(line_item.original_line_item_id)
|
||
|
|
line_item = db.session.get(JournalEntryLineItem, data.l_p_of3d.id)
|
||
|
|
self.assertIsNotNone(line_item)
|
||
|
|
self.assertIsNotNone(line_item.original_line_item_id)
|
||
|
|
self.assertEqual(line_item.original_line_item_id, data.l_p_or2c.id)
|
||
|
|
|
||
|
|
list_uri = f"/accounting/unmatched-offsets/{Accounts.PAYABLE}"
|
||
|
|
response = self.client.post(list_uri,
|
||
|
|
data={"csrf_token": self.csrf_token})
|
||
|
|
self.assertEqual(response.status_code, 302)
|
||
|
|
self.assertEqual(response.headers["Location"], list_uri)
|
||
|
|
|
||
|
|
with self.app.app_context():
|
||
|
|
account = Account.find_by_code(Accounts.PAYABLE)
|
||
|
|
assert account is not None
|
||
|
|
matcher = OffsetMatcher(account)
|
||
|
|
self.assertEqual({x.id for x in matcher.unapplied},
|
||
|
|
{data.l_p_or5c.id, data.l_p_or6c.id})
|
||
|
|
self.assertEqual({x.id for x in matcher.unmatched_offsets},
|
||
|
|
{data.l_p_of1d.id, data.l_p_of5d.id})
|
||
|
|
self.assertEqual(matcher.matches, 0)
|
||
|
|
for line_item_id in {data.l_p_of1d.id, data.l_p_of5d.id}:
|
||
|
|
line_item = db.session.get(JournalEntryLineItem, line_item_id)
|
||
|
|
self.assertIsNotNone(line_item)
|
||
|
|
self.assertIsNone(line_item.original_line_item_id)
|
||
|
|
line_item = db.session.get(JournalEntryLineItem, data.l_p_of2d.id)
|
||
|
|
self.assertIsNotNone(line_item)
|
||
|
|
self.assertIsNotNone(line_item.original_line_item_id)
|
||
|
|
self.assertEqual(line_item.original_line_item_id, data.l_p_or1c.id)
|
||
|
|
line_item = db.session.get(JournalEntryLineItem, data.l_p_of3d.id)
|
||
|
|
self.assertIsNotNone(line_item)
|
||
|
|
self.assertIsNotNone(line_item.original_line_item_id)
|
||
|
|
self.assertEqual(line_item.original_line_item_id, data.l_p_or2c.id)
|
||
|
|
line_item = db.session.get(JournalEntryLineItem, data.l_p_of4d.id)
|
||
|
|
self.assertIsNotNone(line_item)
|
||
|
|
self.assertIsNotNone(line_item.original_line_item_id)
|
||
|
|
self.assertEqual(line_item.original_line_item_id, data.l_p_or3c.id)
|
||
|
|
line_item = db.session.get(JournalEntryLineItem, data.l_p_of6d.id)
|
||
|
|
self.assertIsNotNone(line_item)
|
||
|
|
self.assertIsNotNone(line_item.original_line_item_id)
|
||
|
|
self.assertEqual(line_item.original_line_item_id, data.l_p_or4c.id)
|
||
|
|
|
||
|
|
|
||
|
|
class DifferentTestData:
|
||
|
|
"""The test data for different descriptions and amounts."""
|
||
|
|
|
||
|
|
def __init__(self, app: Flask, client: httpx.Client, csrf_token: str):
|
||
|
|
"""Constructs the test data.
|
||
|
|
|
||
|
|
:param app: The Flask application.
|
||
|
|
:param client: The client.
|
||
|
|
:param csrf_token: The CSRF token.
|
||
|
|
"""
|
||
|
|
self.app: Flask = app
|
||
|
|
self.client: httpx.Client = client
|
||
|
|
self.csrf_token: str = csrf_token
|
||
|
|
|
||
|
|
def couple(description: str, amount: str, debit: str, credit: str) \
|
||
|
|
-> tuple[JournalEntryLineItemData, JournalEntryLineItemData]:
|
||
|
|
"""Returns a couple of debit-credit line items.
|
||
|
|
|
||
|
|
:param description: The description.
|
||
|
|
:param amount: The amount.
|
||
|
|
:param debit: The debit account code.
|
||
|
|
:param credit: The credit account code.
|
||
|
|
:return: The debit line item and credit line item.
|
||
|
|
"""
|
||
|
|
return JournalEntryLineItemData(debit, description, amount),\
|
||
|
|
JournalEntryLineItemData(credit, description, amount)
|
||
|
|
|
||
|
|
# Receivable original line items
|
||
|
|
self.l_r_or1d, self.l_r_or1c = couple(
|
||
|
|
"Accountant", "1200", Accounts.RECEIVABLE, Accounts.SERVICE)
|
||
|
|
self.l_r_or2d, self.l_r_or2c = couple(
|
||
|
|
"Toy", "600", Accounts.RECEIVABLE, Accounts.SALES)
|
||
|
|
self.l_r_or3d, self.l_r_or3c = couple(
|
||
|
|
"Noodles", "100", Accounts.RECEIVABLE, Accounts.SALES)
|
||
|
|
self.l_r_or4d, self.l_r_or4c = couple(
|
||
|
|
"Interest", "3.4", Accounts.RECEIVABLE, Accounts.INTEREST)
|
||
|
|
|
||
|
|
# Payable original line items
|
||
|
|
self.l_p_or1d, self.l_p_or1c = couple(
|
||
|
|
"Airplane", "2000", Accounts.TRAVEL, Accounts.PAYABLE)
|
||
|
|
self.l_p_or2d, self.l_p_or2c = couple(
|
||
|
|
"Phone", "900", Accounts.OFFICE, Accounts.PAYABLE)
|
||
|
|
self.l_p_or3d, self.l_p_or3c = couple(
|
||
|
|
"Steak", "120", Accounts.MEAL, Accounts.PAYABLE)
|
||
|
|
self.l_p_or4d, self.l_p_or4c = couple(
|
||
|
|
"Envelop", "0.9", Accounts.OFFICE, Accounts.PAYABLE)
|
||
|
|
|
||
|
|
# Original journal entries
|
||
|
|
self.j_r_or1: JournalEntryData = JournalEntryData(
|
||
|
|
50, [CurrencyData("USD", [self.l_r_or1d, self.l_r_or4d],
|
||
|
|
[self.l_r_or1c, self.l_r_or4c])])
|
||
|
|
self.j_r_or2: JournalEntryData = JournalEntryData(
|
||
|
|
30, [CurrencyData("USD", [self.l_r_or2d, self.l_r_or3d],
|
||
|
|
[self.l_r_or2c, self.l_r_or3c])])
|
||
|
|
self.j_p_or1: JournalEntryData = JournalEntryData(
|
||
|
|
40, [CurrencyData("USD", [self.l_p_or1d, self.l_p_or4d],
|
||
|
|
[self.l_p_or1c, self.l_p_or4c])])
|
||
|
|
self.j_p_or2: JournalEntryData = JournalEntryData(
|
||
|
|
20, [CurrencyData("USD", [self.l_p_or2d, self.l_p_or3d],
|
||
|
|
[self.l_p_or2c, self.l_p_or3c])])
|
||
|
|
|
||
|
|
self.__add_journal_entry(self.j_r_or1)
|
||
|
|
self.__add_journal_entry(self.j_r_or2)
|
||
|
|
self.__add_journal_entry(self.j_p_or1)
|
||
|
|
self.__add_journal_entry(self.j_p_or2)
|
||
|
|
|
||
|
|
# Receivable offset items
|
||
|
|
self.l_r_of1d, self.l_r_of1c = couple(
|
||
|
|
"Accountant", "500", Accounts.CASH, Accounts.RECEIVABLE)
|
||
|
|
self.l_r_of2d, self.l_r_of2c = couple(
|
||
|
|
"Accountant", "200", Accounts.CASH, Accounts.RECEIVABLE)
|
||
|
|
self.l_r_of3d, self.l_r_of3c = couple(
|
||
|
|
"Accountant", "100", Accounts.CASH, Accounts.RECEIVABLE)
|
||
|
|
self.l_r_of4d, self.l_r_of4c = couple(
|
||
|
|
"Toy", "240", Accounts.CASH, Accounts.RECEIVABLE)
|
||
|
|
self.l_r_of5d, self.l_r_of5c = couple(
|
||
|
|
"Interest", "3.4", Accounts.CASH, Accounts.RECEIVABLE)
|
||
|
|
|
||
|
|
# Payable offset items
|
||
|
|
self.l_p_of1d, self.l_p_of1c = couple(
|
||
|
|
"Airplane", "800", Accounts.PAYABLE, Accounts.CASH)
|
||
|
|
self.l_p_of2d, self.l_p_of2c = couple(
|
||
|
|
"Airplane", "300", Accounts.PAYABLE, Accounts.CASH)
|
||
|
|
self.l_p_of3d, self.l_p_of3c = couple(
|
||
|
|
"Airplane", "100", Accounts.PAYABLE, Accounts.CASH)
|
||
|
|
self.l_p_of4d, self.l_p_of4c = couple(
|
||
|
|
"Phone", "400", Accounts.PAYABLE, Accounts.CASH)
|
||
|
|
self.l_p_of5d, self.l_p_of5c = couple(
|
||
|
|
"Envelop", "0.9", Accounts.PAYABLE, Accounts.CASH)
|
||
|
|
|
||
|
|
# Offset journal entries
|
||
|
|
self.j_r_of1: JournalEntryData = JournalEntryData(
|
||
|
|
25, [CurrencyData("USD", [self.l_r_of1d], [self.l_r_of1c])])
|
||
|
|
self.j_r_of2: JournalEntryData = JournalEntryData(
|
||
|
|
20, [CurrencyData("USD",
|
||
|
|
[self.l_r_of2d, self.l_r_of3d, self.l_r_of4d],
|
||
|
|
[self.l_r_of2c, self.l_r_of3c, self.l_r_of4c])])
|
||
|
|
self.j_r_of3: JournalEntryData = JournalEntryData(
|
||
|
|
15, [CurrencyData("USD", [self.l_r_of5d], [self.l_r_of5c])])
|
||
|
|
self.j_p_of1: JournalEntryData = JournalEntryData(
|
||
|
|
15, [CurrencyData("USD", [self.l_p_of1d], [self.l_p_of1c])])
|
||
|
|
self.j_p_of2: JournalEntryData = JournalEntryData(
|
||
|
|
10, [CurrencyData("USD",
|
||
|
|
[self.l_p_of2d, self.l_p_of3d, self.l_p_of4d],
|
||
|
|
[self.l_p_of2c, self.l_p_of3c, self.l_p_of4c])])
|
||
|
|
self.j_p_of3: JournalEntryData = JournalEntryData(
|
||
|
|
5, [CurrencyData("USD", [self.l_p_of5d], [self.l_p_of5c])])
|
||
|
|
|
||
|
|
self.__set_is_need_offset(False)
|
||
|
|
self.__add_journal_entry(self.j_r_of1)
|
||
|
|
self.__add_journal_entry(self.j_r_of2)
|
||
|
|
self.__add_journal_entry(self.j_r_of3)
|
||
|
|
self.__add_journal_entry(self.j_p_of1)
|
||
|
|
self.__add_journal_entry(self.j_p_of2)
|
||
|
|
self.__add_journal_entry(self.j_p_of3)
|
||
|
|
self.__set_is_need_offset(True)
|
||
|
|
|
||
|
|
def __set_is_need_offset(self, is_need_offset: bool) -> None:
|
||
|
|
"""Sets whether the payables and receivables need offset.
|
||
|
|
|
||
|
|
:param is_need_offset: True if payables and receivables need offset, or
|
||
|
|
False otherwise.
|
||
|
|
:return:
|
||
|
|
"""
|
||
|
|
from accounting.models import Account
|
||
|
|
with self.app.app_context():
|
||
|
|
for code in {Accounts.RECEIVABLE, Accounts.PAYABLE}:
|
||
|
|
account: Account | None = Account.find_by_code(code)
|
||
|
|
assert account is not None
|
||
|
|
account.is_need_offset = is_need_offset
|
||
|
|
db.session.commit()
|
||
|
|
|
||
|
|
def __add_journal_entry(self, journal_entry_data: JournalEntryData) \
|
||
|
|
-> None:
|
||
|
|
"""Adds a journal entry.
|
||
|
|
|
||
|
|
:param journal_entry_data: The journal entry data.
|
||
|
|
:return: None.
|
||
|
|
"""
|
||
|
|
from accounting.models import JournalEntry
|
||
|
|
store_uri: str = "/accounting/journal-entries/store/transfer"
|
||
|
|
|
||
|
|
response: httpx.Response = self.client.post(
|
||
|
|
store_uri, data=journal_entry_data.new_form(self.csrf_token))
|
||
|
|
assert response.status_code == 302
|
||
|
|
journal_entry_id: int \
|
||
|
|
= match_journal_entry_detail(response.headers["Location"])
|
||
|
|
journal_entry_data.id = journal_entry_id
|
||
|
|
with self.app.app_context():
|
||
|
|
journal_entry: JournalEntry | None \
|
||
|
|
= db.session.get(JournalEntry, journal_entry_id)
|
||
|
|
assert journal_entry is not None
|
||
|
|
for i in range(len(journal_entry.currencies)):
|
||
|
|
for j in range(len(journal_entry.currencies[i].debit)):
|
||
|
|
journal_entry_data.currencies[i].debit[j].id \
|
||
|
|
= journal_entry.currencies[i].debit[j].id
|
||
|
|
for j in range(len(journal_entry.currencies[i].credit)):
|
||
|
|
journal_entry_data.currencies[i].credit[j].id \
|
||
|
|
= journal_entry.currencies[i].credit[j].id
|
||
|
|
|
||
|
|
|
||
|
|
class SameTestData:
|
||
|
|
"""The test data with same descriptions and amounts."""
|
||
|
|
|
||
|
|
def __init__(self, app: Flask, client: httpx.Client, csrf_token: str):
|
||
|
|
"""Constructs the test data.
|
||
|
|
|
||
|
|
:param app: The Flask application.
|
||
|
|
:param client: The client.
|
||
|
|
:param csrf_token: The CSRF token.
|
||
|
|
"""
|
||
|
|
self.app: Flask = app
|
||
|
|
self.client: httpx.Client = client
|
||
|
|
self.csrf_token: str = csrf_token
|
||
|
|
|
||
|
|
def couple(description: str, amount: str, debit: str, credit: str) \
|
||
|
|
-> tuple[JournalEntryLineItemData, JournalEntryLineItemData]:
|
||
|
|
"""Returns a couple of debit-credit line items.
|
||
|
|
|
||
|
|
:param description: The description.
|
||
|
|
:param amount: The amount.
|
||
|
|
:param debit: The debit account code.
|
||
|
|
:param credit: The credit account code.
|
||
|
|
:return: The debit line item and credit line item.
|
||
|
|
"""
|
||
|
|
return JournalEntryLineItemData(debit, description, amount),\
|
||
|
|
JournalEntryLineItemData(credit, description, amount)
|
||
|
|
|
||
|
|
# Receivable original line items
|
||
|
|
self.l_r_or1d, self.l_r_or1c = couple(
|
||
|
|
"Noodles", "100", Accounts.RECEIVABLE, Accounts.SALES)
|
||
|
|
self.l_r_or2d, self.l_r_or2c = couple(
|
||
|
|
"Noodles", "100", Accounts.RECEIVABLE, Accounts.SALES)
|
||
|
|
self.l_r_or3d, self.l_r_or3c = couple(
|
||
|
|
"Noodles", "100", Accounts.RECEIVABLE, Accounts.SALES)
|
||
|
|
self.l_r_or4d, self.l_r_or4c = couple(
|
||
|
|
"Noodles", "100", Accounts.RECEIVABLE, Accounts.SALES)
|
||
|
|
self.l_r_or5d, self.l_r_or5c = couple(
|
||
|
|
"Noodles", "100", Accounts.RECEIVABLE, Accounts.SALES)
|
||
|
|
self.l_r_or6d, self.l_r_or6c = couple(
|
||
|
|
"Noodles", "100", Accounts.RECEIVABLE, Accounts.SALES)
|
||
|
|
|
||
|
|
# Payable original line items
|
||
|
|
self.l_p_or1d, self.l_p_or1c = couple(
|
||
|
|
"Steak", "120", Accounts.MEAL, Accounts.PAYABLE)
|
||
|
|
self.l_p_or2d, self.l_p_or2c = couple(
|
||
|
|
"Steak", "120", Accounts.MEAL, Accounts.PAYABLE)
|
||
|
|
self.l_p_or3d, self.l_p_or3c = couple(
|
||
|
|
"Steak", "120", Accounts.MEAL, Accounts.PAYABLE)
|
||
|
|
self.l_p_or4d, self.l_p_or4c = couple(
|
||
|
|
"Steak", "120", Accounts.MEAL, Accounts.PAYABLE)
|
||
|
|
self.l_p_or5d, self.l_p_or5c = couple(
|
||
|
|
"Steak", "120", Accounts.MEAL, Accounts.PAYABLE)
|
||
|
|
self.l_p_or6d, self.l_p_or6c = couple(
|
||
|
|
"Steak", "120", Accounts.MEAL, Accounts.PAYABLE)
|
||
|
|
|
||
|
|
# Original journal entries
|
||
|
|
self.j_r_or1: JournalEntryData = JournalEntryData(
|
||
|
|
60, [CurrencyData("USD", [self.l_r_or1d], [self.l_r_or1c])])
|
||
|
|
self.j_r_or2: JournalEntryData = JournalEntryData(
|
||
|
|
50, [CurrencyData("USD", [self.l_r_or2d], [self.l_r_or2c])])
|
||
|
|
self.j_r_or3: JournalEntryData = JournalEntryData(
|
||
|
|
40, [CurrencyData("USD", [self.l_r_or3d], [self.l_r_or3c])])
|
||
|
|
self.j_r_or4: JournalEntryData = JournalEntryData(
|
||
|
|
30, [CurrencyData("USD", [self.l_r_or4d], [self.l_r_or4c])])
|
||
|
|
self.j_r_or5: JournalEntryData = JournalEntryData(
|
||
|
|
20, [CurrencyData("USD", [self.l_r_or5d], [self.l_r_or5c])])
|
||
|
|
self.j_r_or6: JournalEntryData = JournalEntryData(
|
||
|
|
10, [CurrencyData("USD", [self.l_r_or6d], [self.l_r_or6c])])
|
||
|
|
self.j_p_or1: JournalEntryData = JournalEntryData(
|
||
|
|
60, [CurrencyData("USD", [self.l_p_or1d], [self.l_p_or1c])])
|
||
|
|
self.j_p_or2: JournalEntryData = JournalEntryData(
|
||
|
|
50, [CurrencyData("USD", [self.l_p_or2d], [self.l_p_or2c])])
|
||
|
|
self.j_p_or3: JournalEntryData = JournalEntryData(
|
||
|
|
40, [CurrencyData("USD", [self.l_p_or3d], [self.l_p_or3c])])
|
||
|
|
self.j_p_or4: JournalEntryData = JournalEntryData(
|
||
|
|
30, [CurrencyData("USD", [self.l_p_or4d], [self.l_p_or4c])])
|
||
|
|
self.j_p_or5: JournalEntryData = JournalEntryData(
|
||
|
|
20, [CurrencyData("USD", [self.l_p_or5d], [self.l_p_or5c])])
|
||
|
|
self.j_p_or6: JournalEntryData = JournalEntryData(
|
||
|
|
10, [CurrencyData("USD", [self.l_p_or6d], [self.l_p_or6c])])
|
||
|
|
|
||
|
|
self.__add_journal_entry(self.j_r_or1)
|
||
|
|
self.__add_journal_entry(self.j_r_or2)
|
||
|
|
self.__add_journal_entry(self.j_r_or3)
|
||
|
|
self.__add_journal_entry(self.j_r_or4)
|
||
|
|
self.__add_journal_entry(self.j_r_or5)
|
||
|
|
self.__add_journal_entry(self.j_r_or6)
|
||
|
|
self.__add_journal_entry(self.j_p_or1)
|
||
|
|
self.__add_journal_entry(self.j_p_or2)
|
||
|
|
self.__add_journal_entry(self.j_p_or3)
|
||
|
|
self.__add_journal_entry(self.j_p_or4)
|
||
|
|
self.__add_journal_entry(self.j_p_or5)
|
||
|
|
self.__add_journal_entry(self.j_p_or6)
|
||
|
|
|
||
|
|
# Receivable offset items
|
||
|
|
self.l_r_of1d, self.l_r_of1c = couple(
|
||
|
|
"Noodles", "100", Accounts.CASH, Accounts.RECEIVABLE)
|
||
|
|
self.l_r_of2d, self.l_r_of2c = couple(
|
||
|
|
"Noodles", "100", Accounts.CASH, Accounts.RECEIVABLE)
|
||
|
|
self.l_r_of3d, self.l_r_of3c = couple(
|
||
|
|
"Noodles", "100", Accounts.CASH, Accounts.RECEIVABLE)
|
||
|
|
self.l_r_of3c.original_line_item = self.l_r_or2d
|
||
|
|
self.l_r_of4d, self.l_r_of4c = couple(
|
||
|
|
"Noodles", "100", Accounts.CASH, Accounts.RECEIVABLE)
|
||
|
|
self.l_r_of5d, self.l_r_of5c = couple(
|
||
|
|
"Noodles", "100", Accounts.CASH, Accounts.RECEIVABLE)
|
||
|
|
self.l_r_of6d, self.l_r_of6c = couple(
|
||
|
|
"Noodles", "100", Accounts.CASH, Accounts.RECEIVABLE)
|
||
|
|
|
||
|
|
# Payable offset items
|
||
|
|
self.l_p_of1d, self.l_p_of1c = couple(
|
||
|
|
"Steak", "120", Accounts.PAYABLE, Accounts.CASH)
|
||
|
|
self.l_p_of2d, self.l_p_of2c = couple(
|
||
|
|
"Steak", "120", Accounts.PAYABLE, Accounts.CASH)
|
||
|
|
self.l_p_of3d, self.l_p_of3c = couple(
|
||
|
|
"Steak", "120", Accounts.PAYABLE, Accounts.CASH)
|
||
|
|
self.l_p_of3d.original_line_item = self.l_p_or2c
|
||
|
|
self.l_p_of4d, self.l_p_of4c = couple(
|
||
|
|
"Steak", "120", Accounts.PAYABLE, Accounts.CASH)
|
||
|
|
self.l_p_of5d, self.l_p_of5c = couple(
|
||
|
|
"Steak", "120", Accounts.PAYABLE, Accounts.CASH)
|
||
|
|
self.l_p_of6d, self.l_p_of6c = couple(
|
||
|
|
"Steak", "120", Accounts.PAYABLE, Accounts.CASH)
|
||
|
|
|
||
|
|
# Offset journal entries
|
||
|
|
self.j_r_of1: JournalEntryData = JournalEntryData(
|
||
|
|
65, [CurrencyData("USD", [self.l_r_of1d], [self.l_r_of1c])])
|
||
|
|
self.j_r_of2: JournalEntryData = JournalEntryData(
|
||
|
|
35, [CurrencyData("USD", [self.l_r_of2d], [self.l_r_of2c])])
|
||
|
|
self.j_r_of3: JournalEntryData = JournalEntryData(
|
||
|
|
35, [CurrencyData("USD", [self.l_r_of3d], [self.l_r_of3c])])
|
||
|
|
self.j_r_of4: JournalEntryData = JournalEntryData(
|
||
|
|
35, [CurrencyData("USD", [self.l_r_of4d], [self.l_r_of4c])])
|
||
|
|
self.j_r_of5: JournalEntryData = JournalEntryData(
|
||
|
|
35, [CurrencyData("USD", [self.l_r_of5d], [self.l_r_of5c])])
|
||
|
|
self.j_r_of6: JournalEntryData = JournalEntryData(
|
||
|
|
15, [CurrencyData("USD", [self.l_r_of6d], [self.l_r_of6c])])
|
||
|
|
self.j_p_of1: JournalEntryData = JournalEntryData(
|
||
|
|
65, [CurrencyData("USD", [self.l_p_of1d], [self.l_p_of1c])])
|
||
|
|
self.j_p_of2: JournalEntryData = JournalEntryData(
|
||
|
|
35, [CurrencyData("USD", [self.l_p_of2d], [self.l_p_of2c])])
|
||
|
|
self.j_p_of3: JournalEntryData = JournalEntryData(
|
||
|
|
35, [CurrencyData("USD", [self.l_p_of3d], [self.l_p_of3c])])
|
||
|
|
self.j_p_of4: JournalEntryData = JournalEntryData(
|
||
|
|
35, [CurrencyData("USD", [self.l_p_of4d], [self.l_p_of4c])])
|
||
|
|
self.j_p_of5: JournalEntryData = JournalEntryData(
|
||
|
|
35, [CurrencyData("USD", [self.l_p_of5d], [self.l_p_of5c])])
|
||
|
|
self.j_p_of6: JournalEntryData = JournalEntryData(
|
||
|
|
15, [CurrencyData("USD", [self.l_p_of6d], [self.l_p_of6c])])
|
||
|
|
|
||
|
|
self.__set_is_need_offset(False)
|
||
|
|
self.__add_journal_entry(self.j_r_of1)
|
||
|
|
self.__add_journal_entry(self.j_r_of2)
|
||
|
|
self.__add_journal_entry(self.j_r_of4)
|
||
|
|
self.__add_journal_entry(self.j_r_of5)
|
||
|
|
self.__add_journal_entry(self.j_r_of6)
|
||
|
|
self.__add_journal_entry(self.j_p_of1)
|
||
|
|
self.__add_journal_entry(self.j_p_of2)
|
||
|
|
self.__add_journal_entry(self.j_p_of4)
|
||
|
|
self.__add_journal_entry(self.j_p_of5)
|
||
|
|
self.__add_journal_entry(self.j_p_of6)
|
||
|
|
self.__set_is_need_offset(True)
|
||
|
|
self.__add_journal_entry(self.j_r_of3)
|
||
|
|
self.__add_journal_entry(self.j_p_of3)
|
||
|
|
|
||
|
|
def __set_is_need_offset(self, is_need_offset: bool) -> None:
|
||
|
|
"""Sets whether the payables and receivables need offset.
|
||
|
|
|
||
|
|
:param is_need_offset: True if payables and receivables need offset, or
|
||
|
|
False otherwise.
|
||
|
|
:return:
|
||
|
|
"""
|
||
|
|
from accounting.models import Account
|
||
|
|
with self.app.app_context():
|
||
|
|
for code in {Accounts.RECEIVABLE, Accounts.PAYABLE}:
|
||
|
|
account: Account | None = Account.find_by_code(code)
|
||
|
|
assert account is not None
|
||
|
|
account.is_need_offset = is_need_offset
|
||
|
|
db.session.commit()
|
||
|
|
|
||
|
|
def __add_journal_entry(self, journal_entry_data: JournalEntryData) \
|
||
|
|
-> None:
|
||
|
|
"""Adds a journal entry.
|
||
|
|
|
||
|
|
:param journal_entry_data: The journal entry data.
|
||
|
|
:return: None.
|
||
|
|
"""
|
||
|
|
from accounting.models import JournalEntry
|
||
|
|
store_uri: str = "/accounting/journal-entries/store/transfer"
|
||
|
|
|
||
|
|
response: httpx.Response = self.client.post(
|
||
|
|
store_uri, data=journal_entry_data.new_form(self.csrf_token))
|
||
|
|
assert response.status_code == 302
|
||
|
|
journal_entry_id: int \
|
||
|
|
= match_journal_entry_detail(response.headers["Location"])
|
||
|
|
journal_entry_data.id = journal_entry_id
|
||
|
|
with self.app.app_context():
|
||
|
|
journal_entry: JournalEntry | None \
|
||
|
|
= db.session.get(JournalEntry, journal_entry_id)
|
||
|
|
assert journal_entry is not None
|
||
|
|
for i in range(len(journal_entry.currencies)):
|
||
|
|
for j in range(len(journal_entry.currencies[i].debit)):
|
||
|
|
journal_entry_data.currencies[i].debit[j].id \
|
||
|
|
= journal_entry.currencies[i].debit[j].id
|
||
|
|
for j in range(len(journal_entry.currencies[i].credit)):
|
||
|
|
journal_entry_data.currencies[i].credit[j].id \
|
||
|
|
= journal_entry.currencies[i].credit[j].id
|