Add tests to grunt