def test_cloudwatch_get_metric_statistics_bad_response(aws_client,
mock_datetime):
client = MagicMock()
aws_client.return_value = client
mock_datetime.utcnow.return_value = datetime(2015,
1,
1,
15,
15,
tzinfo=timezone.utc)
namespace = "AWS/Lambda"
metric_name = "Invocations"
dimension_name = "FunctionName"
dimension_value = "MyFunction"
duration = 120
offset = 60
statistic = "Sum"
extended_statistic = None
unit = "Count"
client.get_metric_statistics.return_value = {
"Datapoints": [{
"some": "value"
}]
}
with pytest.raises(FailedActivity):
get_metric_statistics(
namespace=namespace,
metric_name=metric_name,
dimension_name=dimension_name,
dimension_value=dimension_value,
duration=duration,
offset=offset,
statistic=statistic,
extended_statistic=extended_statistic,
unit=unit,
)
client.get_metric_statistics.assert_called_with(
Namespace=namespace,
MetricName=metric_name,
Dimensions=[{
"Name": dimension_name,
"Value": dimension_value
}],
Period=duration,
StartTime=datetime(2015, 1, 1, 15, 12, tzinfo=timezone.utc),
EndTime=datetime(2015, 1, 1, 15, 14, tzinfo=timezone.utc),
Statistics=[statistic],
Unit=unit,
)
def test_cloudwatch_get_metric_statistics_bad_response(aws_client,
mock_datetime):
client = MagicMock()
aws_client.return_value = client
mock_datetime.utcnow.return_value = datetime(2015,
1,
1,
15,
15,
tzinfo=timezone.utc)
namespace = 'AWS/Lambda'
metric_name = 'Invocations'
dimension_name = 'FunctionName'
dimension_value = 'MyFunction'
duration = 120
offset = 60
statistic = 'Sum'
extended_statistic = None
unit = 'Count'
client.get_metric_statistics.return_value = {
'Datapoints': [{
'some': 'value'
}]
}
with pytest.raises(FailedActivity):
get_metric_statistics(namespace=namespace,
metric_name=metric_name,
dimension_name=dimension_name,
dimension_value=dimension_value,
duration=duration,
offset=offset,
statistic=statistic,
extended_statistic=extended_statistic,
unit=unit)
client.get_metric_statistics.assert_called_with(
Namespace=namespace,
MetricName=metric_name,
Dimensions=[{
'Name': dimension_name,
'Value': dimension_value
}],
Period=duration,
StartTime=datetime(2015, 1, 1, 15, 12, tzinfo=timezone.utc),
EndTime=datetime(2015, 1, 1, 15, 14, tzinfo=timezone.utc),
Statistics=[statistic],
Unit=unit)
def test_cloudwatch_get_metric_statistics_extended_ok(aws_client,
mock_datetime):
client = MagicMock()
aws_client.return_value = client
mock_datetime.utcnow.return_value = datetime(2015,
1,
1,
15,
15,
tzinfo=timezone.utc)
namespace = "AWS/Lambda"
metric_name = "Invocations"
dimension_name = "FunctionName"
dimension_value = "MyFunction"
duration = 120
offset = 60
statistic = None
extended_statistic = "p99"
client.get_metric_statistics.return_value = {
"Datapoints": [{
"ExtendedStatistics": {
extended_statistic: 4753.0
}
}]
}
unit = None
result = get_metric_statistics(
namespace=namespace,
metric_name=metric_name,
dimension_name=dimension_name,
dimension_value=dimension_value,
duration=duration,
offset=offset,
statistic=statistic,
extended_statistic=extended_statistic,
unit=unit,
)
assert result == 4753.0
client.get_metric_statistics.assert_called_with(
Namespace=namespace,
MetricName=metric_name,
Dimensions=[{
"Name": dimension_name,
"Value": dimension_value
}],
Period=duration,
StartTime=datetime(2015, 1, 1, 15, 12, tzinfo=timezone.utc),
EndTime=datetime(2015, 1, 1, 15, 14, tzinfo=timezone.utc),
ExtendedStatistics=[extended_statistic],
)
def test_cloudwatch_get_metric_statistics_extended_ok(aws_client,
mock_datetime):
client = MagicMock()
aws_client.return_value = client
mock_datetime.utcnow.return_value = datetime(2015,
1,
1,
15,
15,
tzinfo=timezone.utc)
namespace = 'AWS/Lambda'
metric_name = 'Invocations'
dimension_name = 'FunctionName'
dimension_value = 'MyFunction'
duration = 120
offset = 60
statistic = None
extended_statistic = 'p99'
client.get_metric_statistics.return_value = {
'Datapoints': [{
'ExtendedStatistics': {
extended_statistic: 4753.0
}
}]
}
unit = None
result = get_metric_statistics(namespace=namespace,
metric_name=metric_name,
dimension_name=dimension_name,
dimension_value=dimension_value,
duration=duration,
offset=offset,
statistic=statistic,
extended_statistic=extended_statistic,
unit=unit)
assert result == 4753.0
client.get_metric_statistics.assert_called_with(
Namespace=namespace,
MetricName=metric_name,
Dimensions=[{
'Name': dimension_name,
'Value': dimension_value
}],
Period=duration,
StartTime=datetime(2015, 1, 1, 15, 12, tzinfo=timezone.utc),
EndTime=datetime(2015, 1, 1, 15, 14, tzinfo=timezone.utc),
ExtendedStatistics=[extended_statistic],
)
