def testAccountPrefix(self): ledger = self.parser.parse_ledger(fixture_path("simple.dat")) processor = LedgerProcessor(ledger, self.rules) collector = TransactionCollector() processor.add_listener(collector) processor.add_account_prefix("Business") processor.run() for transaction, entries in collector.transactions: for entry in entries: self.assertEqual("Business", entry.account.root().name)
def testRemoveAccountPrefix(self): ledger = self.parser.parse_ledger(fixture_path("simple.dat")) processor = LedgerProcessor(ledger, self.rules) collector = TransactionCollector() processor.add_listener(collector) self.assertEqual(2, len(ledger.transactions)) processor.add_account_prefix("Business") ledger.transactions[0].execute(processor) for transaction, entries in collector.transactions: for entry in entries: self.assertEqual("Business", entry.account.root().name) collector.transactions = [] processor.remove_account_prefix() ledger.transactions[0].execute(processor) for transaction, entries in collector.transactions: for entry in entries: self.assertNotEqual("Business", entry.account.root().name)