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>')