def METHOD_NAME(aggregator, tags, is_enabled=True):
    expected_metrics = set()
    for raw_metric_name, metric_name in JMX_METRICS_MAP.items():
        if raw_metric_name.endswith('_total') and raw_metric_name not in JMX_METRICS_OVERRIDES:
            expected_metrics.add('{}.count'.format(metric_name[:-6]))
        else:
            expected_metrics.add(metric_name)
    expected_metrics.update(METRICS_AS_GAUGE_AND_COUNT + [m + ".count" for m in METRICS_AS_GAUGE_AND_COUNT])
    expected_metrics.update(METRICS_FROM_LABELS + [m + ".count" for m in METRICS_FROM_LABELS])
    for metric in sorted(expected_metrics):
        metric = 'aws.msk.{}'.format(metric)
        if is_enabled:
            for tag in tags:
                aggregator.assert_metric_has_tag(metric, tag)
        else:
            aggregator.assert_metric(metric, count=0)