from django.test import TestCase

from general.soft_delete_helper import _determine_transaction_types


class SoftDeleteTransactionTypeTests(TestCase):
    def test_share_purchase_reference_is_treated_as_shares_and_savings(self):
        self.assertEqual(
            _determine_transaction_types("p-sh-2509-00001"),
            ["savings", "shares"],
        )

    def test_share_withdraw_reference_is_treated_as_shares_and_savings(self):
        self.assertEqual(
            _determine_transaction_types("w-sh-2509-00001"),
            ["savings", "shares"],
        )

    def test_share_transfer_reference_is_treated_as_shares_and_savings(self):
        self.assertEqual(
            _determine_transaction_types("t-sh-2509-00001"),
            ["savings", "shares"],
        )