def test_mask_card_data_from_xml_with_existing_card_number_and_cvv(self): xml = '<CardNumber>1234567890123456</CardNumber><CardSecurityCode>123</CardSecurityCode>' assert (mask_card_data_from_xml(xml) == '<CardNumber>123456******3456</CardNumber><CardSecurityCode>***</CardSecurityCode>')
def test_mask_card_data_from_xml_with_mixed_xml_with_cvv(self): xml = '<html><head></head><CardSecurityCode>1234</CardSecurityCode><body></body></html>' assert (mask_card_data_from_xml(xml) == '<html><head></head><CardSecurityCode>****</CardSecurityCode><body></body></html>')
def test_mask_card_data_from_xml_with_non_existing_cvv(self): xml = '<html><head></head><body></body></html>' assert mask_card_data_from_xml(xml) == xml
def test_mask_card_data_from_xml_with_existing_cvv(self): xml = '<CardSecurityCode>123</CardSecurityCode>' assert (mask_card_data_from_xml(xml) == '<CardSecurityCode>***</CardSecurityCode>')
def test_mask_card_data_from_xml_with_mixed_xml(self): xml = '<html><head></head><CardNumber>1234567890123456</CardNumber><body></body></html>' assert (mask_card_data_from_xml(xml) == '<html><head></head><CardNumber>123456******3456</CardNumber><body></body></html>')
def test_mask_card_data_from_xml_with_existing_card_number(self): xml = '<CardNumber>1234567890123456</CardNumber>' assert (mask_card_data_from_xml(xml) == '<CardNumber>123456******3456</CardNumber>')