/*
 *
 *  * Copyright 2010-2016 OrientDB LTD (http://orientdb.com)
 *  *
 *  * 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.orientechnologies.lucene.test;

import com.orientechnologies.orient.core.command.OCommandOutputListener;
import com.orientechnologies.orient.core.db.tool.ODatabaseExport;
import com.orientechnologies.orient.core.db.tool.ODatabaseImport;
import com.orientechnologies.orient.core.index.OIndex;
import com.orientechnologies.orient.core.metadata.schema.OClass;
import com.orientechnologies.orient.core.metadata.schema.OSchema;
import com.orientechnologies.orient.core.metadata.schema.OType;
import com.orientechnologies.orient.core.record.impl.ODocument;
import com.orientechnologies.orient.core.sql.OCommandSQL;
import com.orientechnologies.orient.core.sql.query.OSQLSynchQuery;
import org.assertj.core.api.Assertions;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;

import java.io.FileInputStream;
import java.io.IOException;
import java.util.List;
import java.util.zip.GZIPInputStream;

import static org.assertj.core.api.Assertions.*;

/**
 * Created by Enrico Risa on 07/07/15.
 */
public class LuceneExportImportTest extends BaseLuceneTest {

  @Before
  public void init() {

    OSchema schema = db.getMetadata().getSchema();
    OClass oClass = schema.createClass("City");

    oClass.createProperty("name", OType.STRING);
    db.command(new OCommandSQL("create index City.name on City (name) FULLTEXT ENGINE LUCENE")).execute();

    ODocument doc = new ODocument("City");
    doc.field("name", "Rome");
    db.save(doc);
  }

  @Test
  public void testExportImport() {

    String file = "./target/exportTest.json";

    List<?> query = db.query(new OSQLSynchQuery<Object>("select from City where name lucene 'Rome'"));

    assertThat(query).hasSize(1);

    try {

      //export the DB
      new ODatabaseExport(db, file, new OCommandOutputListener() {
        @Override
        public void onMessage(String s) {
        }
      }).exportDatabase();

      db.drop();
      db.create();
      GZIPInputStream stream = new GZIPInputStream(new FileInputStream(file + ".gz"));

      //import
      new ODatabaseImport(db, stream, new OCommandOutputListener() {
        @Override
        public void onMessage(String s) {
          System.out.println(s);
        }
      }).importDatabase();
    } catch (IOException e) {
      Assert.fail(e.getMessage());
    }

    db.commit();
    long city = db.countClass("City");

    Assert.assertEquals(city, 1);

    OIndex<?> index = db.getMetadata().getIndexManager().getIndex("City.name");

    Assert.assertNotNull(index);
    assertThat(index.getSize()).isEqualTo(1L);
    Assert.assertEquals(index.getType(), "FULLTEXT");
    Assert.assertEquals(index.getAlgorithm(), "LUCENE");

    query = db.query(new OSQLSynchQuery<Object>("select from City "));

    assertThat(query.size()).isEqualTo(1);
    query = db.query(new OSQLSynchQuery<Object>("select from City where name lucene 'rome'"));
    assertThat(query).hasSize(1);
  }

}