def test_get_throughput_returns_0_if_no_data(self, mock_boto3, mocker):
sqs_scaler = SqsScaler(app_name, min_instances, max_instances, **self.input_attrs)
mocker.patch.object(sqs_scaler, '_get_sqs_throughput', return_value=[])
statsd_mock = mocker.patch.object(sqs_scaler, 'statsd_client')
assert sqs_scaler._get_throughput('my-queue') == 0
statsd_mock.gauge.assert_called_once_with('testmy-queue.queue-throughput', 0)
def test_get_throughput_uses_max_value(self, mock_boto3, mocker):
sqs_scaler = SqsScaler(app_name, min_instances, max_instances, **self.input_attrs)
_get_sqs_throughput_mock = mocker.patch.object(sqs_scaler, '_get_sqs_throughput', return_value=[100, 200, 50])
statsd_mock = mocker.patch.object(sqs_scaler, 'statsd_client')
assert sqs_scaler._get_throughput('my-queue') == 200
# queue prefix "test" pulled from SQS_QUEUE_PREFIX env var
statsd_mock.gauge.assert_called_once_with('testmy-queue.queue-throughput', 50)
_get_sqs_throughput_mock.assert_called_once_with('testmy-queue')
