def test_get_aggregators(self):
"""User passed --aggregate-by args"""
cmd = Command()
cmd.init_args([
"--aggregate-by", "asn_v4", "--aggregate-by", "country",
"--aggregate-by", "prefix_v4"
])
expected_output = [
ValueKeyAggregator(key="asn_v4"),
ValueKeyAggregator(key="country_code"),
ValueKeyAggregator(key="prefix_v4")
]
cmd.set_aggregators()
for index, v in enumerate(cmd.aggregators):
self.assertTrue(isinstance(v, ValueKeyAggregator))
self.assertEquals(v.aggregation_keys,
expected_output[index].aggregation_keys)
def test_get_aggregators(self):
"""User passed --aggregate-by args"""
cmd = Command()
cmd.init_args([
"--aggregate-by", "asn_v4",
"--aggregate-by", "country",
"--aggregate-by", "prefix_v4"
])
expected_output = [
ValueKeyAggregator(key="asn_v4"),
ValueKeyAggregator(key="country_code"),
ValueKeyAggregator(key="prefix_v4")
]
cmd.set_aggregators()
for index, v in enumerate(cmd.aggregators):
self.assertTrue(isinstance(v, ValueKeyAggregator))
self.assertEquals(
v.aggregation_keys,
expected_output[index].aggregation_keys
)
