add test cases for agg_keys