PageTreeProviderTest.java

/*
 * #%L
 * wcm.io
 * %%
 * Copyright (C) 2014 wcm.io
 * %%
 * 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.
 * #L%
 */
package io.wcm.wcm.ui.extjs.provider.impl.servlets;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNull;

import javax.servlet.http.HttpServletResponse;

import org.apache.sling.commons.json.JSONArray;
import org.apache.sling.commons.json.JSONException;
import org.apache.sling.commons.json.JSONObject;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;

import com.day.cq.commons.predicate.PredicateProvider;

import io.wcm.sling.commons.resource.ImmutableValueMap;
import io.wcm.testing.mock.aem.junit.AemContext;
import io.wcm.wcm.ui.extjs.provider.AbstractPageProvider;

public class PageTreeProviderTest {

  private static final String TEMPLATE = "/apps/app1/templates/template1";

  @Rule
  public AemContext context = new AemContext();

  @Before
  public void setUp() throws Exception {
    context.create().page("/content/sample/en");
    context.create().page("/content/sample/en/page1", TEMPLATE, "title1");
    context.create().page("/content/sample/en/page1/page11", TEMPLATE, "title11");
    context.create().page("/content/sample/en/page1/page11/page111", TEMPLATE, "title111");
    context.create().page("/content/sample/en/page1/page11/page112", TEMPLATE, "title111");
    context.create().page("/content/sample/en/page1/page12", TEMPLATE, "title12");
    context.create().page("/content/sample/en/page1/page12/page121", TEMPLATE, "title121");
    context.create().page("/content/sample/en/page1/page12/page122", TEMPLATE, "title122");
    context.create().page("/content/sample/en/page2", TEMPLATE, "title2");
  }

  @Test
  public void testWithPath() throws Exception {
    context.request().setParameterMap(ImmutableValueMap.of(AbstractPageProvider.RP_PATH, "/content/sample/en"));

    JSONArray result = getJsonResult();

    assertEquals(2, result.length());
    assertItem(result.getJSONObject(0), "page1", "title1", TEMPLATE, false);
    assertItem(result.getJSONObject(1), "page2", "title2", TEMPLATE, true);

    JSONArray page1children = result.getJSONObject(0).getJSONArray("children");
    assertEquals(2, page1children.length());

    assertItem(page1children.getJSONObject(0), "page11", "title11", TEMPLATE, false);
    assertNull(page1children.getJSONObject(0).optJSONArray("children"));
    assertItem(page1children.getJSONObject(1), "page12", "title12", TEMPLATE, false);
    assertNull(page1children.getJSONObject(1).optJSONArray("children"));
  }

  @Test
  public void testWithPredicate() throws Exception {
    context.registerService(PredicateProvider.class, new DummyPredicateProvider());

    context.request().setParameterMap(ImmutableValueMap.of(AbstractPageProvider.RP_PATH, "/content/sample/en",
        AbstractPageProvider.RP_PREDICATE, DummyPredicateProvider.PREDICATE_PAGENAME_PAGE1));

    JSONArray result = getJsonResult();

    assertEquals(1, result.length());
    assertItem(result.getJSONObject(0), "page1", "title1", TEMPLATE, true);
  }

  private JSONArray getJsonResult() throws Exception {
    PageTreeProvider underTest = context.registerInjectActivateService(new PageTreeProvider());
    underTest.service(context.request(), context.response());
    if (context.response().getStatus() == HttpServletResponse.SC_OK) {
      return new JSONArray(context.response().getOutputAsString());
    }
    else {
      return null;
    }
  }

  private void assertItem(JSONObject jsonObject, String name, String title, String template,
      boolean leaf) throws JSONException {
    assertEquals(name, jsonObject.get("name"));
    assertEquals(title, jsonObject.get("text"));
    assertEquals("cq:Page", jsonObject.get("type"));
    assertEquals(template, jsonObject.get("template"));
    assertEquals("page", jsonObject.get("cls"));
    assertEquals(leaf, jsonObject.optBoolean("leaf"));
  }

}