def test_init(self):
base = BaseDispatcher('test_subject',
'send_lots',
'send_one',
max_batch_size=1)
base._initialise_aws_client()
self.assertEqual('test_subject_client', base._aws_service.client_name)
self.assertEqual('send_lots', base._batch_dispatch_method.__name__)
self.assertEqual('send_one', base._individual_dispatch_method.__name__)
def test_boto3_overrides(self):
base = BaseDispatcher('test_subject',
'send_lots',
'send_one',
max_batch_size=1,
endpoint_url='https://dummy_endpoint:54321/',
aws_session_token='session_token')
base._initialise_aws_client()
self.assertEqual('https://dummy_endpoint:54321/',
base._aws_service.kwargs['endpoint_url'])
self.assertEqual('session_token',
base._aws_service.kwargs['aws_session_token'])
def test_payload_equal_max_batch_size(self, mock_chunks):
base = BaseDispatcher('test_subject',
'send_lots',
'send_one',
max_batch_size=3)
base._batch_payload = [1, 2, 3]
base._initialise_aws_client = Mock()
base._batch_send_payloads = Mock()
mock_chunks.return_value = [[1, 2, 3]]
base.flush_payloads()
base._initialise_aws_client.assert_called_once()
base._batch_send_payloads.assert_called_once_with([1, 2, 3])
self.assertEqual([], base._batch_payload)
def test_empty_payload_list(self, mock_chunks):
base = BaseDispatcher('test_subject',
'send_lots',
'send_one',
max_batch_size=3)
base._batch_payload = []
base._initialise_aws_client = Mock()
base._batch_send_payloads = Mock()
mock_chunks.return_value = [[]]
base.flush_payloads()
base._initialise_aws_client.assert_called_once()
base._batch_send_payloads.assert_not_called()
self.assertEqual([], base._batch_payload)
def test_payload_multiple_batches(self, mock_chunks):
base = BaseDispatcher('test_subject',
'send_lots',
'send_one',
max_batch_size=3)
base._batch_payload = [1, 2, 3, 4]
base._initialise_aws_client = Mock()
base._batch_send_payloads = Mock()
mock_chunks.return_value = [[1, 2, 3], [4]]
base.flush_payloads()
base._initialise_aws_client.assert_called_once()
base._batch_send_payloads.assert_has_calls(
[call([1, 2, 3]), call([4])])
self.assertEqual([], base._batch_payload)
def test_unprocessed_items_are_returned(self, mock_chunks):
test_unprocessed_items = ['abc1', 'cde2', 'efg3']
base = BaseDispatcher('test_subject',
'send_lots',
'send_one',
max_batch_size=3)
base.unprocessed_items = test_unprocessed_items
base._batch_payload = [1, 2, 3, 4]
base._initialise_aws_client = Mock()
base._batch_send_payloads = Mock()
mock_chunks.return_value = [[1, 2, 3], [4]]
response = base.flush_payloads()
base._initialise_aws_client.assert_called_once()
base._batch_send_payloads.assert_has_calls(
[call([1, 2, 3]), call([4])])
self.assertEqual([], base._batch_payload)
self.assertEqual(test_unprocessed_items, response)