D-Net

package eu.dnetlib.data.oai.store.parser;

import java.time.LocalDateTime;

import java.time.ZoneId;

import java.time.format.DateTimeFormatter;

import java.util.Date;

import com.google.common.collect.Lists;

import com.mongodb.BasicDBObject;

import com.mongodb.DBObject;

import com.mongodb.util.JSON;

import eu.dnetlib.oai.parser.MongoQueryParser;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.bson.conversions.Bson;

import org.bson.types.ObjectId;

import org.junit.Test;

import static org.junit.Assert.assertEquals;

public class MongoQueryParserTest {

	private static final Log log = LogFactory.getLog(MongoQueryParserTest.class); // NOPMD by marko on 11/24/08 5:02 PM

	private final MongoQueryParser mongoParser = new MongoQueryParser();

	@Test

	public void testParseEq() {

		final Bson expected = new BasicDBObject("set", "CEDIASManuscripts");

		final Bson o = this.mongoParser.parse("set = \"CEDIASManuscripts\"");

		assertEquals(expected, o);

	}

	@Test

	public void testParseNeq() {

		final Bson expected = new BasicDBObject("set", new BasicDBObject("$ne", "CEDIASManuscripts"));

		final Bson o = this.mongoParser.parse("set <> \"CEDIASManuscripts\"");

		assertEquals(expected, o);

	}

	@Test

	public void testParseAnd() {

		final BasicDBObject expected = new BasicDBObject("$and", Lists.newArrayList(new BasicDBObject("set", new BasicDBObject("$ne", "CEDIASManuscripts")),

				new BasicDBObject("pippo", new BasicDBObject("$gt", "x"))));

		final Bson o = this.mongoParser.parse("set <> \"CEDIASManuscripts\" AND pippo > x");

		log.info(o);

		assertEquals(expected, o);

	}

	@Test

	public void testParseOr() {

		final BasicDBObject expected = new BasicDBObject("$or", Lists.newArrayList(new BasicDBObject("set", new BasicDBObject("$ne", "CEDIASManuscripts")),

				new BasicDBObject("pippo", new BasicDBObject("$gt", "x"))));

		final Bson o = this.mongoParser.parse("set <> \"CEDIASManuscripts\" OR pippo > x");

		log.info(o);

		assertEquals(expected, o);

	}

	@Test

	public void testParseNot() {

		final BasicDBObject expected = new BasicDBObject("$and", Lists.newArrayList(new BasicDBObject("set", "CEDIASManuscripts"), new BasicDBObject("$not",

				new BasicDBObject("pippo", new BasicDBObject("$gt", "x")))));

		final Bson o = this.mongoParser.parse("set = \"CEDIASManuscripts\" NOT pippo > x");

		// log.info(o)

		assertEquals(expected, o);

	}

	@Test

	public void testParseStar() {

		final BasicDBObject expected = new BasicDBObject();

		final Bson o = this.mongoParser.parse("*");

		final Bson o2 = this.mongoParser.parse("*=*");

		assertEquals(expected, o);

		assertEquals(expected, o2);

	}

	@Test

	public void testParseStarAnd() {

		final BasicDBObject expected =

				new BasicDBObject("$and", Lists.newArrayList(new BasicDBObject(), new BasicDBObject("pippo", new BasicDBObject("$gt", "x"))));

		final Bson o = this.mongoParser.parse("* AND pippo > x");

		final Bson o2 = this.mongoParser.parse("*=* AND pippo > x");

		assertEquals(expected, o);

		assertEquals(expected, o2);

	}

	@Test

	public void testParseIdQuery() {

		final BasicDBObject expected = new BasicDBObject("_id", new BasicDBObject("$gt", new ObjectId("5225e093aabf055637bf2c65")));

		final Bson o = this.mongoParser.parse("_id > 5225e093aabf055637bf2c65");

		assertEquals(expected, o);

	}

	@Test

	public void testParseUntilDatestamp() {

		final Date dateTime = Date.from(this.parseDate("2014-04-03T00:00:00.000Z").atZone(ZoneId.systemDefault()).toInstant());

		final BasicDBObject expected =

				new BasicDBObject("$and", Lists.newArrayList(new BasicDBObject("oaftype", new BasicDBObject("$ne", "person")), new BasicDBObject(

						"datestamp", new BasicDBObject("$lt", dateTime))));

		final Bson o = this.mongoParser.parse("(oaftype <> \"person\") AND datestamp <= 2014-04-02");

		System.out.println("o.toString() = " + expected.toString());

		System.out.println("o.toString() = " + o.toString());

		assertEquals(expected, o);

	}

	@Test

	public void testJsonParsing() {

		final String jsonQuery = "{licenses:{$elemMatch:{\"license\" : \"OPEN\", \"hostedby\" : {$ne:\"Europe PubMed Central\"}}}}";

		final DBObject query = (DBObject) JSON.parse(jsonQuery);

		System.out.println(query);

	}

	private LocalDateTime parseDate(final String date) {

		final DateTimeFormatter dateNoTimeFormatter = DateTimeFormatter.ofPattern("yyyy-MM-dd");

		final DateTimeFormatter iso8601NoMsTimeFormatter = DateTimeFormatter.ofPattern("yyyy-MM-dd'T'HH:mm:ssZ");

		final DateTimeFormatter iso8601Formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd'T'HH:mm:ss'.'SSS'Z'");

		try {

			System.err.println("Using default " + iso8601Formatter.getClass());

			return LocalDateTime.parse(date, iso8601Formatter);

		} catch (final Exception e) {

			e.printStackTrace();

			try {

				System.err.println("Switching to ISO with no millisecond date formatter: yyyy-MM-dd'T'HH:mm:ssZ");

				return LocalDateTime.parse(date, iso8601NoMsTimeFormatter);

			} catch (final Exception ex) {

				System.err.println("Switching to simple date formatter: yyyy-MM-dd");

				return LocalDateTime.parse(date, dateNoTimeFormatter);

			}

		}

	}

}