HelpersTest.java

/**
 * Copyright (c) 2012 Edgar Espina
 *
 * This file is part of Handlebars.java.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.github.jknack.handlebars;

import java.io.IOException;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;

import mustache.specs.Spec;
import mustache.specs.SpecTest;

import org.junit.runners.Parameterized.Parameters;


public class HelpersTest extends SpecTest {

  public HelpersTest(final Spec spec) {
    super(spec);
  }

  @Override
  protected HelperRegistry configure(final Handlebars handlebars) {
    handlebars.registerHelper("list", new Helper<List<Object>>() {
      @Override
      public Object apply(final List<Object> list, final Options options)
          throws IOException {
        String text = "";
        if (options.isFalsy(list)) {
          return text;
        }

        text += "<ul>\n";
        for (Object element : list) {
          text += "  <li>\n    ";
          text += options.fn(element);
          text += "  </li>\n";
        }
        text += "</ul>\n";
        return text;
      }
    });
    handlebars.registerHelper("fullName", new Helper<Map<String, Object>>() {
      @Override
      public Object apply(final Map<String, Object> context,
          final Options options) throws IOException {
        return context.get("firstName") + " " + context.get("lastName");
      }
    });
    handlebars.registerHelper("agree_button",
        new Helper<Map<String, Object>>() {
          @Override
          public Object apply(final Map<String, Object> context,
              final Options options)
              throws IOException {
            String text = "<button>I agree. I " + context.get("emotion") + " "
                + context.get("name") + "</button>";
            return new Handlebars.SafeString(text);
          }
        });
    handlebars.registerHelper("link",
        new Helper<Object>() {
          @Override
          public Object apply(final Object context,
              final Options options)
              throws IOException {
            return "<a href='" + options.param(0) + "'>"
                + context + "</a>";
          }
        });
    handlebars.registerHelper("link-hash",
        new Helper<String>() {
          @Override
          public Object apply(final String text,
              final Options options)
              throws IOException {
            StringBuilder classes = new StringBuilder();
            String sep = " ";
            for (Entry<String, Object> entry : options.hash.entrySet()) {
              classes.append(entry.getKey()).append("=\"")
                  .append(entry.getValue()).append("\"").append(sep);
            }
            classes.setLength(classes.length() - sep.length());
            return new Handlebars.SafeString("<a " + classes + ">" + text
                + "</a>");
          }
        });
    return super.configure(handlebars);
  }

  @Parameters
  public static Collection<Object[]> data() throws IOException {
    return data(HelpersTest.class, "helpers.yml");
  }
}