Add category integration test