1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92
|
# Copyright 2013 Amazon.com, Inc. or its affiliates. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License"). You
# may not use this file except in compliance with the License. A copy of
# the License is located at
#
# http://aws.amazon.com/apache2.0/
#
# or in the "license" file accompanying this file. This file is
# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
# ANY KIND, either express or implied. See the License for the specific
# language governing permissions and limitations under the License.
import decimal
from awscli.customizations import putmetricdata
from awscli.testutils import unittest
class TestPutMetricArgument(unittest.TestCase):
def test_build_metric_name_arg(self):
arg = putmetricdata.PutMetricArgument(
'metric-name', help_text='metric-name'
)
parameters = {}
arg.add_to_params(parameters, 'MyMetricName')
self.assertEqual(
parameters['MetricData'][0]['MetricName'], 'MyMetricName'
)
def test_build_unit_arg(self):
arg = putmetricdata.PutMetricArgument('unit', help_text='unit')
parameters = {}
arg.add_to_params(parameters, 'Percent')
self.assertEqual(parameters['MetricData'][0]['Unit'], 'Percent')
def test_value_arg(self):
arg = putmetricdata.PutMetricArgument('value', help_text='value')
parameters = {}
arg.add_to_params(parameters, '123.1')
self.assertEqual(
parameters['MetricData'][0]['Value'], decimal.Decimal('123.1')
)
def test_timestamp_arg(self):
arg = putmetricdata.PutMetricArgument(
'timestamp', help_text='timestamp'
)
parameters = {}
arg.add_to_params(parameters, '2013-09-01')
self.assertEqual(
parameters['MetricData'][0]['Timestamp'], '2013-09-01'
)
def test_dimensions_arg(self):
arg = putmetricdata.PutMetricArgument(
'dimensions', help_text='dimensions'
)
parameters = {}
arg.add_to_params(parameters, 'User=someuser,Stack=test')
self.assertEqual(
parameters['MetricData'][0]['Dimensions'],
[
{"Name": "User", "Value": "someuser"},
{"Name": "Stack", "Value": "test"},
],
)
def test_statistics_arg(self):
arg = putmetricdata.PutMetricArgument(
'statistic-values', help_text='statistic-values'
)
parameters = {}
arg.add_to_params(
parameters, 'Sum=250,Minimum=30,Maximum=70,SampleCount=5'
)
self.assertEqual(
parameters['MetricData'][0]['StatisticValues'],
{
'Maximum': decimal.Decimal('70'),
'Minimum': decimal.Decimal('30'),
'SampleCount': decimal.Decimal('5'),
'Sum': decimal.Decimal('250'),
},
)
def test_parse_empty_value(self):
arg = putmetricdata.PutMetricArgument(
'dimensions', help_text='dimensions'
)
parameters = {}
arg.add_to_params(parameters, None)
self.assertEqual(parameters, {})
|