def check_terror(self):
if self.transfer.type == TYPE_RECEIVE or self.transfer.type == TYPE_SEND:
is_sender_in_terror = find_name_in_terror(self.transfer.sender_name)
is_receiver_in_terror = find_name_in_terror(self.transfer.receiver_name)
if is_sender_in_terror or is_receiver_in_terror:
if is_sender_in_terror and not is_receiver_in_terror:
reason = "SENDER"
elif is_receiver_in_terror and not is_sender_in_terror:
reason = "RECEIVER"
else:
reason = "BOTH SIDES"
insert_name_to_terror_list(self.transfer, reason)
def test_find_not_terror_name_in_terror(self):
row_number = find_name_in_terror("INNOCENT NAME", mdb=self.test_db_file)
self.assertEqual(row_number, -1)
def test_find_terror_name_in_terror(self):
row_number = find_name_in_terror(self.test_transfer.sender_name, mdb=self.test_db_file)
self.assertGreater(row_number, 0)
