Add regression tests for token buckets on their own