def _test_successful_match(self, mock_get_conn, record, expected_date): conn = mock_get_conn() conn.batches.get.return_value = {"count": 1, "results": [{"id": 10}]} batch_id = upload.get_matching_batch_id_for_settlement(record) self.assertEqual(batch_id, 10) conn.batches.get.assert_called_with(date=expected_date.isoformat())
def _test_successful_match(self, mock_get_conn, record, expected_date): conn = mock_get_conn() conn.batches.get.return_value = {'count': 1, 'results': [{'id': 10}]} batch_id = upload.get_matching_batch_id_for_settlement(record) self.assertEqual(batch_id, 10) conn.batches.get.assert_called_with(date=expected_date.isoformat())
def test_invalid_date_is_ignored(self, mock_get_conn): record = DataRecord( "1234566717531509324543278990056000000000009802" "TT- GGGGGGGG -9934WORLDPAY " " 04036 " ) batch_id = upload.get_matching_batch_id_for_settlement(record) self.assertIsNone(batch_id)
def test_invalid_date_is_ignored(self, mock_get_conn): # record is for 36th date of 2004, i.e. 2004-02-05 (last 5 digits in record) # but settlement date cannot be parsed record = DataRecord('1234566717531509324543278990056000000000009802' 'TT- GGGGGGGG -9934WORLDPAY ' ' 04036 ') batch_id = upload.get_matching_batch_id_for_settlement(record) self.assertIsNone(batch_id)