/**
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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 org.apache.solr.handler.dataimport;

import org.apache.solr.SolrTestCaseJ4;
import org.junit.Assert;
import org.junit.Ignore;
import org.junit.Test;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.xml.sax.InputSource;

import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import java.io.StringReader;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * <p>
 * Test for ScriptTransformer
 * </p>
 * <p/>
 * All tests in this have been ignored because script support is only available
 * in Java 1.6+
 *
 * @version $Id: TestScriptTransformer.java 963873 2010-07-13 21:45:31Z rmuir $
 * @since solr 1.3
 */
public class TestScriptTransformer extends SolrTestCaseJ4 {

  @Test
  @Ignore
  public void basic() {
    String script = "function f1(row,context){"
            + "row.put('name','Hello ' + row.get('name'));" + "return row;\n" + "}";
    Context context = getContext("f1", script);
    Map<String, Object> map = new HashMap<String, Object>();
    map.put("name", "Scott");
    EntityProcessorWrapper sep = new EntityProcessorWrapper(new SqlEntityProcessor(), null);
    sep.init(context);
    sep.applyTransformer(map);
    Assert.assertEquals(map.get("name"), "Hello Scott");

  }

  private Context getContext(String funcName, String script) {
    List<Map<String, String>> fields = new ArrayList<Map<String, String>>();
    Map<String, String> entity = new HashMap<String, String>();
    entity.put("name", "hello");
    entity.put("transformer", "script:" + funcName);

    AbstractDataImportHandlerTestCase.TestContext context = AbstractDataImportHandlerTestCase.getContext(null, null, null,
            Context.FULL_DUMP, fields, entity);
    context.script = script;
    context.scriptlang = "JavaScript";
    return context;
  }

  @Test
  @Ignore
  public void oneparam() {

    String script = "function f1(row){"
            + "row.put('name','Hello ' + row.get('name'));" + "return row;\n" + "}";

    Context context = getContext("f1", script);
    Map<String, Object> map = new HashMap<String, Object>();
    map.put("name", "Scott");
    EntityProcessorWrapper sep = new EntityProcessorWrapper(new SqlEntityProcessor(), null);
    sep.init(context);
    sep.applyTransformer(map);
    Assert.assertEquals(map.get("name"), "Hello Scott");

  }

  @Test
  @Ignore
  public void readScriptTag() throws Exception {
    DocumentBuilder builder = DocumentBuilderFactory.newInstance()
            .newDocumentBuilder();
    Document document = builder.parse(new InputSource(new StringReader(xml)));
    DataConfig config = new DataConfig();
    config.readFromXml((Element) document.getElementsByTagName("dataConfig")
            .item(0));
    Assert.assertTrue(config.script.text.indexOf("checkNextToken") > -1);
  }

  @Test
  @Ignore
  public void checkScript() throws Exception {
    DocumentBuilder builder = DocumentBuilderFactory.newInstance()
            .newDocumentBuilder();
    Document document = builder.parse(new InputSource(new StringReader(xml)));
    DataConfig config = new DataConfig();
    config.readFromXml((Element) document.getElementsByTagName("dataConfig")
            .item(0));

    Context c = getContext("checkNextToken", config.script.text);

    Map map = new HashMap();
    map.put("nextToken", "hello");
    EntityProcessorWrapper sep = new EntityProcessorWrapper(new SqlEntityProcessor(), null);
    sep.init(c);
    sep.applyTransformer(map);
    Assert.assertEquals("true", map.get("$hasMore"));
    map = new HashMap();
    map.put("nextToken", "");
    sep.applyTransformer(map);
    Assert.assertNull(map.get("$hasMore"));

  }

  static String xml = "<dataConfig>\n"
          + "<script><![CDATA[\n"
          + "function checkNextToken(row)\t{\n"
          + " var nt = row.get('nextToken');"
          + " if (nt && nt !='' ){ "
          + "    row.put('$hasMore', 'true');}\n"
          + "    return row;\n"
          + "}]]></script>\t<document>\n"
          + "\t\t<entity name=\"mbx\" pk=\"articleNumber\" processor=\"XPathEntityProcessor\"\n"
          + "\t\t\turl=\"?boardId=${dataimporter.defaults.boardId}&maxRecords=20&includeBody=true&startDate=${dataimporter.defaults.startDate}&guid=:autosearch001&reqId=1&transactionId=stringfortracing&listPos=${mbx.nextToken}\"\n"
          + "\t\t\tforEach=\"/mbmessage/articles/navigation | /mbmessage/articles/article\" transformer=\"script:checkNextToken\">\n"
          + "\n" + "\t\t\t<field column=\"nextToken\"\n"
          + "\t\t\t\txpath=\"/mbmessage/articles/navigation/nextToken\" />\n"
          + "\n" + "\t\t</entity>\n" + "\t</document>\n" + "</dataConfig>";
}