[expectations] Add expectation for categories