bitwarden-server/test/Core.Test/Models/Business/TaxInfoTests.cs

using Bit.Core.Models.Business;
using NSubstitute;
using Xunit;

namespace Bit.Core.Test.Models.Business
{
    public class TaxInfoTests
    {
        // PH = Placeholder
        [Theory]
        [InlineData(null, null, null, null)]
        [InlineData("", "", null, null)]
        [InlineData("PH", "", null, null)]
        [InlineData("", "PH", null, null)]
        [InlineData("AE", "PH", null, "ae_trn")]
        [InlineData("AU", "PH", null, "au_abn")]
        [InlineData("BR", "PH", null, "br_cnpj")]
        [InlineData("CA", "PH", "bec", "ca_qst")]
        [InlineData("CA", "PH", null, "ca_bn")]
        [InlineData("CL", "PH", null, "cl_tin")]
        [InlineData("AT", "PH", null, "eu_vat")]
        [InlineData("BE", "PH", null, "eu_vat")]
        [InlineData("BG", "PH", null, "eu_vat")]
        [InlineData("CY", "PH", null, "eu_vat")]
        [InlineData("CZ", "PH", null, "eu_vat")]
        [InlineData("DE", "PH", null, "eu_vat")]
        [InlineData("DK", "PH", null, "eu_vat")]
        [InlineData("EE", "PH", null, "eu_vat")]
        [InlineData("ES", "PH", null, "eu_vat")]
        [InlineData("FI", "PH", null, "eu_vat")]
        [InlineData("FR", "PH", null, "eu_vat")]
        [InlineData("GB", "PH", null, "eu_vat")]
        [InlineData("GR", "PH", null, "eu_vat")]
        [InlineData("HR", "PH", null, "eu_vat")]
        [InlineData("HU", "PH", null, "eu_vat")]
        [InlineData("IE", "PH", null, "eu_vat")]
        [InlineData("IT", "PH", null, "eu_vat")]
        [InlineData("LT", "PH", null, "eu_vat")]
        [InlineData("LU", "PH", null, "eu_vat")]
        [InlineData("LV", "PH", null, "eu_vat")]
        [InlineData("MT", "PH", null, "eu_vat")]
        [InlineData("NL", "PH", null, "eu_vat")]
        [InlineData("PL", "PH", null, "eu_vat")]
        [InlineData("PT", "PH", null, "eu_vat")]
        [InlineData("RO", "PH", null, "eu_vat")]
        [InlineData("SE", "PH", null, "eu_vat")]
        [InlineData("SI", "PH", null, "eu_vat")]
        [InlineData("SK", "PH", null, "eu_vat")]
        [InlineData("HK", "PH", null, "hk_br")]
        [InlineData("IN", "PH", null, "in_gst")]
        [InlineData("JP", "PH", null, "jp_cn")]
        [InlineData("KR", "PH", null, "kr_brn")]
        [InlineData("LI", "PH", null, "li_uid")]
        [InlineData("MX", "PH", null, "mx_rfc")]
        [InlineData("MY", "PH", null, "my_sst")]
        [InlineData("NO", "PH", null, "no_vat")]
        [InlineData("NZ", "PH", null, "nz_gst")]
        [InlineData("RU", "PH", null, "ru_inn")]
        [InlineData("SA", "PH", null, "sa_vat")]
        [InlineData("SG", "PH", null, "sg_gst")]
        [InlineData("TH", "PH", null, "th_vat")]
        [InlineData("TW", "PH", null, "tw_vat")]
        [InlineData("US", "PH", null, "us_ein")]
        [InlineData("ZA", "PH", null, "za_vat")]
        [InlineData("ABCDEF", "PH", null, null)]
        public void GetTaxIdType_Success(string billingAddressCountry,
            string taxIdNumber,
            string billingAddressState,
            string expectedTaxIdType)
        {
            var taxInfo = new TaxInfo
            {
                BillingAddressCountry = billingAddressCountry,
                TaxIdNumber = taxIdNumber,
                BillingAddressState = billingAddressState,
            };

            Assert.Equal(expectedTaxIdType, taxInfo.TaxIdType);
        }

        [Fact]
        public void GetTaxIdType_CreateOnce_ReturnCacheSecondTime()
        {
            var taxInfo = new TaxInfo
            {
                BillingAddressCountry = "US",
                TaxIdNumber = "PH",
                BillingAddressState = null,
            };

            Assert.Equal("us_ein", taxInfo.TaxIdType);

            // Per the current spec even if the values change to something other than null it
            // will return the cached version of TaxIdType.
            taxInfo.BillingAddressCountry = "ZA";

            Assert.Equal("us_ein", taxInfo.TaxIdType);
        }

        [Theory]
        [InlineData(null, null, false)]
        [InlineData("123", "US", true)]
        [InlineData("123", "ZQ12", false)]
        [InlineData("    ", "US", false)]
        public void HasTaxId_ReturnsExpected(string taxIdNumber, string billingAddressCountry, bool expected)
        {
            var taxInfo = new TaxInfo
            {
                TaxIdNumber = taxIdNumber,
                BillingAddressCountry = billingAddressCountry,
            };

            Assert.Equal(expected, taxInfo.HasTaxId);
        }
    }
}