integration: add count test