Add tests for new rake task