IndexQueryRestrictionProcessorTest.java

package rocks.inspectit.shared.cs.indexing.restriction;

import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.Matchers.is;

import java.util.Collections;
import java.util.List;

import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.slf4j.Logger;
import org.testng.annotations.Test;

import rocks.inspectit.shared.all.communication.data.HttpTimerData;
import rocks.inspectit.shared.all.communication.data.TimerData;
import rocks.inspectit.shared.all.indexing.restriction.IIndexQueryRestriction;
import rocks.inspectit.shared.all.testbase.TestBase;
import rocks.inspectit.shared.cs.indexing.restriction.impl.CachingIndexQueryRestrictionProcessor;
import rocks.inspectit.shared.cs.indexing.restriction.impl.IndexQueryRestrictionFactory;

/**
 * Tests the indexing restriction used with index queries.
 *
 * @author Ivan Senic
 *
 */
@SuppressWarnings("PMD")
public class IndexQueryRestrictionProcessorTest extends TestBase {

	/**
	 * The processor under test.
	 */
	@InjectMocks
	protected CachingIndexQueryRestrictionProcessor processor;

	@Mock
	protected Logger log;

	public static class AreAllRestrictionsFulfilled extends IndexQueryRestrictionProcessorTest {

		/**
		 * Test equal restriction.
		 */
		@Test
		public void equalsTrue() {
			TimerData timerData = new TimerData();
			timerData.setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.equal("id", 1L));

			boolean result = processor.areAllRestrictionsFulfilled(timerData, restrictions);

			assertThat(result, is(true));
		}

		/**
		 * Test equal restriction.
		 */
		@Test
		public void equalsFalse() {
			TimerData timerData = new TimerData();
			timerData.setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.equal("id", 0L));

			boolean result = processor.areAllRestrictionsFulfilled(timerData, restrictions);

			assertThat(result, is(false));
		}

		/**
		 * Test not equal restriction.
		 */
		@Test
		public void notEqualsFalse() {
			TimerData timerData = new TimerData();
			timerData.setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.notEqual("id", 1L));

			boolean result = processor.areAllRestrictionsFulfilled(timerData, restrictions);

			assertThat(result, is(false));
		}

		/**
		 * Test not equal restriction.
		 */
		@Test
		public void notEqualsTrue() {
			TimerData timerData = new TimerData();
			timerData.setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.notEqual("id", 0L));

			boolean result = processor.areAllRestrictionsFulfilled(timerData, restrictions);

			assertThat(result, is(true));
		}

		/**
		 * Test is null restriction.
		 */
		@Test
		public void isNull() {
			TimerData timerData = new TimerData();
			timerData.setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.isNull("id"));

			boolean result = processor.areAllRestrictionsFulfilled(timerData, restrictions);

			assertThat(result, is(false));
		}

		/**
		 * Test is not null restriction.
		 */
		@Test
		public void isNotNull() {
			TimerData timerData = new TimerData();
			timerData.setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.isNotNull("id"));

			boolean result = processor.areAllRestrictionsFulfilled(timerData, restrictions);

			assertThat(result, is(true));
		}

		/**
		 * Test greater than restriction.
		 */
		@Test
		public void greaterThanOne() {
			TimerData timerData = new TimerData();
			timerData.setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.greaterThan("id", 1L));

			boolean result = processor.areAllRestrictionsFulfilled(timerData, restrictions);

			assertThat(result, is(false));
		}

		/**
		 * Test greater than restriction.
		 */
		@Test
		public void greaterThanZero() {
			TimerData timerData = new TimerData();
			timerData.setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.greaterThan("id", 0L));

			boolean result = processor.areAllRestrictionsFulfilled(timerData, restrictions);

			assertThat(result, is(true));
		}

		/**
		 * Test greater than restriction.
		 */
		@Test
		public void greaterThanTwo() {
			TimerData timerData = new TimerData();
			timerData.setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.greaterThan("id", 2L));

			boolean result = processor.areAllRestrictionsFulfilled(timerData, restrictions);

			assertThat(result, is(false));
		}

		/**
		 * Test greater or equal restriction.
		 */
		@Test
		public void greaterEqualOne() {
			TimerData timerData = new TimerData();
			timerData.setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.greaterEqual("id", 1L));

			boolean result = processor.areAllRestrictionsFulfilled(timerData, restrictions);

			assertThat(result, is(true));
		}

		/**
		 * Test greater or equal restriction.
		 */
		@Test
		public void greaterEqualZero() {
			TimerData timerData = new TimerData();
			timerData.setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.greaterEqual("id", 0L));

			boolean result = processor.areAllRestrictionsFulfilled(timerData, restrictions);

			assertThat(result, is(true));
		}

		/**
		 * Test greater or equal restriction.
		 */
		@Test
		public void greaterEqualTwo() {
			TimerData timerData = new TimerData();
			timerData.setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.greaterEqual("id", 2L));

			boolean result = processor.areAllRestrictionsFulfilled(timerData, restrictions);

			assertThat(result, is(false));
		}

		/**
		 * Test less than restriction.
		 */
		@Test
		public void lessThanOne() {
			TimerData timerData = new TimerData();
			timerData.setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.lessThan("id", 1L));

			boolean result = processor.areAllRestrictionsFulfilled(timerData, restrictions);

			assertThat(result, is(false));
		}

		/**
		 * Test less than restriction.
		 */
		@Test
		public void lessThanZero() {
			TimerData timerData = new TimerData();
			timerData.setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.lessThan("id", 0L));

			boolean result = processor.areAllRestrictionsFulfilled(timerData, restrictions);

			assertThat(result, is(false));
		}

		/**
		 * Test less than restriction.
		 */
		@Test
		public void lessThanTwo() {
			TimerData timerData = new TimerData();
			timerData.setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.lessThan("id", 2L));

			boolean result = processor.areAllRestrictionsFulfilled(timerData, restrictions);

			assertThat(result, is(true));
		}

		/**
		 * Test less or equal restriction.
		 */
		@Test
		public void lessEqualOne() {
			TimerData timerData = new TimerData();
			timerData.setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.lessEqual("id", 1L));

			boolean result = processor.areAllRestrictionsFulfilled(timerData, restrictions);

			assertThat(result, is(true));
		}

		/**
		 * Test less or equal restriction.
		 */
		@Test
		public void lessEqualZero() {
			TimerData timerData = new TimerData();
			timerData.setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.lessEqual("id", 0L));

			boolean result = processor.areAllRestrictionsFulfilled(timerData, restrictions);

			assertThat(result, is(false));
		}

		/**
		 * Test less or equal restriction.
		 */
		@Test
		public void lessEqualTwo() {
			TimerData timerData = new TimerData();
			timerData.setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.lessEqual("id", 2L));

			boolean result = processor.areAllRestrictionsFulfilled(timerData, restrictions);

			assertThat(result, is(true));
		}

		/**
		 * Tests is in collection restrictions.
		 */
		@Test
		public void isInCollectionTrue() {
			TimerData timerData = new TimerData();
			timerData.setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.isInCollection("id", Collections.singletonList(1L)));

			boolean result = processor.areAllRestrictionsFulfilled(timerData, restrictions);

			assertThat(result, is(true));
		}

		/**
		 * Tests is in collection restrictions.
		 */
		@Test
		public void isInCollectionFalse() {
			TimerData timerData = new TimerData();
			timerData.setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.isInCollection("id", Collections.singletonList(2L)));

			boolean result = processor.areAllRestrictionsFulfilled(timerData, restrictions);

			assertThat(result, is(false));
		}

		/**
		 * Test equal restriction with field navigation.
		 */
		@Test
		public void equalsTrueNavigation() {
			HttpTimerData httpData = new HttpTimerData();
			httpData.getHttpInfo().setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.equal("httpInfo.id", 1L));

			boolean result = processor.areAllRestrictionsFulfilled(httpData, restrictions);

			assertThat(result, is(true));
		}

		/**
		 * Test equal restriction with field navigation.
		 */
		@Test
		public void equalsFalseNavigation() {
			HttpTimerData httpData = new HttpTimerData();
			httpData.getHttpInfo().setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.equal("httpInfo.id", 0L));

			boolean result = processor.areAllRestrictionsFulfilled(httpData, restrictions);

			assertThat(result, is(false));
		}

		/**
		 * Test not equal restriction with field navigation.
		 */
		@Test
		public void notEqualsFalseNavigation() {
			HttpTimerData httpData = new HttpTimerData();
			httpData.getHttpInfo().setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.notEqual("httpInfo.id", 1L));

			boolean result = processor.areAllRestrictionsFulfilled(httpData, restrictions);

			assertThat(result, is(false));
		}

		/**
		 * Test not equal restriction with field navigation.
		 */
		@Test
		public void notEqualsTrueRestrictionNavigation() {
			HttpTimerData httpData = new HttpTimerData();
			httpData.getHttpInfo().setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.notEqual("httpInfo.id", 0L));

			boolean result = processor.areAllRestrictionsFulfilled(httpData, restrictions);

			assertThat(result, is(true));
		}

		/**
		 * Test is null restriction with field navigation.
		 */
		@Test
		public void isNullNavigation() {
			HttpTimerData httpData = new HttpTimerData();
			httpData.getHttpInfo().setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.isNull("httpInfo.id"));

			boolean result = processor.areAllRestrictionsFulfilled(httpData, restrictions);

			assertThat(result, is(false));
		}

		/**
		 * Test is not null restriction with field navigation.
		 */
		@Test
		public void isNotNullNavigation() {
			HttpTimerData httpData = new HttpTimerData();
			httpData.getHttpInfo().setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.isNotNull("httpInfo.id"));

			boolean result = processor.areAllRestrictionsFulfilled(httpData, restrictions);

			assertThat(result, is(true));
		}

		/**
		 * Test greater than restriction with field navigation.
		 */
		@Test
		public void greaterThanOneNavigation() {
			HttpTimerData httpData = new HttpTimerData();
			httpData.getHttpInfo().setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.greaterThan("httpInfo.id", 1L));

			boolean result = processor.areAllRestrictionsFulfilled(httpData, restrictions);

			assertThat(result, is(false));
		}

		/**
		 * Test greater than restriction with field navigation.
		 */
		@Test
		public void greaterThanZeroNavigation() {
			HttpTimerData httpData = new HttpTimerData();
			httpData.getHttpInfo().setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.greaterThan("httpInfo.id", 0L));

			boolean result = processor.areAllRestrictionsFulfilled(httpData, restrictions);

			assertThat(result, is(true));
		}

		/**
		 * Test greater than restriction with field navigation.
		 */
		@Test
		public void greaterThanTwoNavigation() {
			HttpTimerData httpData = new HttpTimerData();
			httpData.getHttpInfo().setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.greaterThan("httpInfo.id", 2L));

			boolean result = processor.areAllRestrictionsFulfilled(httpData, restrictions);

			assertThat(result, is(false));
		}

		/**
		 * Test greater or equal restriction with field navigation.
		 */
		@Test
		public void greaterEqualOneNavigation() {
			HttpTimerData httpData = new HttpTimerData();
			httpData.getHttpInfo().setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.greaterEqual("httpInfo.id", 1L));

			boolean result = processor.areAllRestrictionsFulfilled(httpData, restrictions);

			assertThat(result, is(true));
		}

		/**
		 * Test greater or equal restriction with field navigation.
		 */
		@Test
		public void greaterEqualZeroNavigation() {
			HttpTimerData httpData = new HttpTimerData();
			httpData.getHttpInfo().setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.greaterEqual("httpInfo.id", 0L));

			boolean result = processor.areAllRestrictionsFulfilled(httpData, restrictions);

			assertThat(result, is(true));
		}

		/**
		 * Test greater or equal restriction with field navigation.
		 */
		@Test
		public void greaterEqualTwoNavigation() {
			HttpTimerData httpData = new HttpTimerData();
			httpData.getHttpInfo().setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.greaterEqual("httpInfo.id", 2L));

			boolean result = processor.areAllRestrictionsFulfilled(httpData, restrictions);

			assertThat(result, is(false));
		}

		/**
		 * Test less than restriction with field navigation.
		 */
		@Test
		public void lessThanOneNavigation() {
			HttpTimerData httpData = new HttpTimerData();
			httpData.getHttpInfo().setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.lessThan("httpInfo.id", 1L));

			boolean result = processor.areAllRestrictionsFulfilled(httpData, restrictions);

			assertThat(result, is(false));
		}

		/**
		 * Test less than restriction with field navigation.
		 */
		@Test
		public void lessThanZeroNavigation() {
			HttpTimerData httpData = new HttpTimerData();
			httpData.getHttpInfo().setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.lessThan("httpInfo.id", 0L));

			boolean result = processor.areAllRestrictionsFulfilled(httpData, restrictions);

			assertThat(result, is(false));
		}

		/**
		 * Test less than restriction with field navigation.
		 */
		@Test
		public void lessThanTwoNavigation() {
			HttpTimerData httpData = new HttpTimerData();
			httpData.getHttpInfo().setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.lessThan("httpInfo.id", 2L));

			boolean result = processor.areAllRestrictionsFulfilled(httpData, restrictions);

			assertThat(result, is(true));
		}

		/**
		 * Test less or equal restriction with field navigation.
		 */
		@Test
		public void lessEqualOneNavigation() {
			HttpTimerData httpData = new HttpTimerData();
			httpData.getHttpInfo().setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.lessEqual("httpInfo.id", 1L));

			boolean result = processor.areAllRestrictionsFulfilled(httpData, restrictions);

			assertThat(result, is(true));
		}

		/**
		 * Test less or equal restriction with field navigation.
		 */
		@Test
		public void lessEqualZeroNavigation() {
			HttpTimerData httpData = new HttpTimerData();
			httpData.getHttpInfo().setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.lessEqual("httpInfo.id", 0L));

			boolean result = processor.areAllRestrictionsFulfilled(httpData, restrictions);

			assertThat(result, is(false));
		}

		/**
		 * Test less or equal restriction with field navigation.
		 */
		@Test
		public void lessEqualTwoNavigation() {
			HttpTimerData httpData = new HttpTimerData();
			httpData.getHttpInfo().setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.lessEqual("httpInfo.id", 2L));

			boolean result = processor.areAllRestrictionsFulfilled(httpData, restrictions);

			assertThat(result, is(true));
		}

		/**
		 * Tests is in collection restrictions with field navigation.
		 */
		@Test
		public void isInCollectionTrueNavigation() {
			HttpTimerData httpData = new HttpTimerData();
			httpData.getHttpInfo().setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.isInCollection("httpInfo.id", Collections.singletonList(1L)));

			boolean result = processor.areAllRestrictionsFulfilled(httpData, restrictions);

			assertThat(result, is(true));
		}

		/**
		 * Tests is in collection restrictions with field navigation.
		 */
		@Test
		public void isInCollectionFalseNavigation() {
			HttpTimerData httpData = new HttpTimerData();
			httpData.getHttpInfo().setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.isInCollection("httpInfo.id", Collections.singletonList(2L)));

			boolean result = processor.areAllRestrictionsFulfilled(httpData, restrictions);

			assertThat(result, is(false));
		}

		/**
		 * Tests that the field to restrict on does not exists.
		 */
		@Test
		public void fieldDoesNotExists() {
			TimerData timerData = new TimerData();
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.equal("somefield", 1L));

			boolean result = processor.areAllRestrictionsFulfilled(timerData, restrictions);

			assertThat(result, is(false));
		}

		/**
		 * Tests that the same field is invoked twice so we pull method from the cache.
		 */
		@Test
		public void twiceExistingField() {
			TimerData timerData = new TimerData();
			timerData.setId(1L);
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.equal("id", 1L));

			processor.areAllRestrictionsFulfilled(timerData, restrictions);
			boolean result = processor.areAllRestrictionsFulfilled(timerData, restrictions);

			assertThat(result, is(true));
		}

		/**
		 * Tests that the same non-existing field is invoked twice so we pull method marker from the
		 * cache.
		 */
		@Test
		public void twiceNonExistingField() {
			TimerData timerData = new TimerData();
			List<IIndexQueryRestriction> restrictions = Collections.singletonList(IndexQueryRestrictionFactory.equal("somefield", 1L));

			processor.areAllRestrictionsFulfilled(timerData, restrictions);
			boolean result = processor.areAllRestrictionsFulfilled(timerData, restrictions);

			assertThat(result, is(false));
		}
	}

}