def abstract_test_chargeback_cost(
rate_key, obj_type, interval, chargeback_report_data, compute_rate, soft_assert):
"""This is an abstract test function for testing rate costs.
It's comparing the expected value that calculated by the rate
to the value in the chargeback report
Args:
:py:type:`str` rate_key: The rate key as it appear in the CHARGEBACK_HEADER_NAMES keys.
:py:type:`str` obj_type: Object being tested; only 'Project' and 'Image' are supported
:py:type:`str` interval: The rate interval, (Hourly/Daily/Weekly/Monthly)
:py:class:`Report` chargeback_report_data: The chargeback report data.
:py:class:`ComputeRate` compute_rate: The compute rate object.
:var soft_assert: soft_assert fixture.
"""
report_headers = CHARGEBACK_HEADER_NAMES[rate_key]
found_something_to_test = False
for row in chargeback_report_data.rows:
if row['Chargeback Rates'].lower() != compute_rate.description.lower():
continue
found_something_to_test = True
fixed_rate = float(compute_rate.fields[report_headers.rate_name]['fixed_rate'])
variable_rate = float(compute_rate.fields[report_headers.rate_name].get('variable_rate', 0))
# Calculate numerical metric
if rate_key == 'Memory':
size_, unit_ = tokenize(row[report_headers.metric_name].upper())
metric = round(parse_size(str(size_) + unit_, binary=True) / 1048576.0, 2)
else:
metric = parse_number(row[report_headers.metric_name])
# Calculate fixed product and cost
num_hours = parse_number(row[CHARGEBACK_HEADER_NAMES['Fixed1'].metric_name])
num_intervals = num_hours / hours_count_lut[interval]
fixed_cost = num_intervals * fixed_rate
variable_cost = num_intervals * metric * variable_rate
# Calculate expected cost
expected_cost = round(variable_cost + fixed_cost, 2)
found_cost = round(parse_number(row[report_headers.cost_name]), 2)
match_threshold = TEST_MATCH_ACCURACY * expected_cost
soft_assert(
abs(found_cost - expected_cost) <= match_threshold,
'Unexpected Chargeback: {}'.format(dump_args(
charge_for=obj_type, rate_key=rate_key, metric=metric, num_hours=num_hours,
num_intervals=num_intervals, fixed_rate=fixed_rate, variable_rate=variable_rate,
fixed_cost=fixed_cost, variable_cost=variable_cost,
expected_full_cost=expected_cost, found_full_cost=found_cost
))
)
assert found_something_to_test, \
'Could not find {} with the assigned rate: {}'.format(obj_type, compute_rate.description)
def abstract_test_chargeback_cost(rate_key, obj_type, interval,
chargeback_report_data, compute_rate,
soft_assert):
"""This is an abstract test function for testing rate costs.
It's comparing the expected value that calculated by the rate
to the value in the chargeback report
Args:
:py:type:`str` rate_key: The rate key as it appear in the CHARGEBACK_HEADER_NAMES keys.
:py:type:`str` obj_type: Object being tested; only 'Project' and 'Image' are supported
:py:type:`str` interval: The rate interval, (Hourly/Daily/Weekly/Monthly)
:py:class:`Report` chargeback_report_data: The chargeback report data.
:py:class:`ComputeRate` compute_rate: The compute rate object.
:var soft_assert: soft_assert fixture.
"""
report_headers = CHARGEBACK_HEADER_NAMES[rate_key]
found_something_to_test = False
for row in chargeback_report_data.rows:
if row['Chargeback Rates'].lower() != compute_rate.description.lower():
continue
found_something_to_test = True
fixed_rate = float(
compute_rate.fields[report_headers.rate_name]['fixed_rate'])
variable_rate = float(
compute_rate.fields[report_headers.rate_name].get(
'variable_rate', 0))
# Calculate numerical metric
if rate_key == 'Memory':
size_, unit_ = tokenize(row[report_headers.metric_name].upper())
metric = round(
parse_size(str(size_) + unit_, binary=True) / 1048576.0, 2)
else:
metric = parse_number(row[report_headers.metric_name])
# Calculate fixed product and cost
num_hours = parse_number(
row[CHARGEBACK_HEADER_NAMES['Fixed1'].metric_name])
num_intervals = num_hours / hours_count_lut[interval]
fixed_cost = num_intervals * fixed_rate
variable_cost = num_intervals * metric * variable_rate
# Calculate expected cost
expected_cost = round(variable_cost + fixed_cost, 2)
found_cost = round(parse_number(row[report_headers.cost_name]), 2)
match_threshold = TEST_MATCH_ACCURACY * expected_cost
soft_assert(
abs(found_cost - expected_cost) <= match_threshold,
'Unexpected Chargeback: {}'.format(
dump_args(charge_for=obj_type,
rate_key=rate_key,
metric=metric,
num_hours=num_hours,
num_intervals=num_intervals,
fixed_rate=fixed_rate,
variable_rate=variable_rate,
fixed_cost=fixed_cost,
variable_cost=variable_cost,
expected_full_cost=expected_cost,
found_full_cost=found_cost)))
assert found_something_to_test, \
'Could not find {} with the assigned rate: {}'.format(obj_type, compute_rate.description)