/jira-project/jira-components/jira-tests-parent/jira-tests-unit/src/test/java/com/atlassian/jira/jql/context/TestIssueTypeClauseContextFactory.java

package com.atlassian.jira.jql.context;

import com.atlassian.jira.config.ConstantsManager;
import com.atlassian.jira.issue.search.constants.SystemSearchConstants;
import com.atlassian.jira.jql.operand.JqlOperandResolver;
import com.atlassian.jira.jql.operand.QueryLiteral;
import com.atlassian.jira.jql.resolver.IssueTypeResolver;
import com.atlassian.jira.jql.validator.MockJqlOperandResolver;
import com.atlassian.jira.user.ApplicationUser;
import com.atlassian.jira.util.collect.CollectionBuilder;
import com.atlassian.query.clause.TerminalClause;
import com.atlassian.query.clause.TerminalClauseImpl;
import com.atlassian.query.operand.EmptyOperand;
import com.atlassian.query.operand.MultiValueOperand;
import com.atlassian.query.operand.Operand;
import com.atlassian.query.operand.SingleValueOperand;
import com.atlassian.query.operator.Operator;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnit;
import org.mockito.junit.MockitoRule;

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

import static com.atlassian.jira.jql.operand.SimpleLiteralFactory.createLiteral;
import static org.junit.Assert.assertEquals;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

/**
 * @since v4.0
 */
public class TestIssueTypeClauseContextFactory {
    @Rule
    public MockitoRule mockito = MockitoJUnit.rule();

    @Mock
    private IssueTypeResolver issueTypeResolver;
    @Mock
    private ConstantsManager constantsManager;

    private MockJqlOperandResolver jqlOperandResolver;
    private IssueTypeClauseContextFactory factory;


    @Before
    public void setUp() throws Exception {
        jqlOperandResolver = MockJqlOperandResolver.createSimpleSupport();
        factory = new IssueTypeClauseContextFactory(issueTypeResolver, jqlOperandResolver, constantsManager);

    }

    @Test
    public void testGetClauseContextSingleIssueTypeEquals() throws Exception {
        final SingleValueOperand operand = new SingleValueOperand("it1");
        final TerminalClause clause = new TerminalClauseImpl(SystemSearchConstants.forIssueType().getJqlClauseNames().getPrimaryName(), Operator.EQUALS, operand);
        final IssueTypeClauseContextFactory factory = createNoopIssueTypeClauseContextFactory();

        final ClauseContext result = factory.getClauseContext(null, clause);
        ClauseContext expectedResult = new ClauseContextImpl(CollectionBuilder.<ProjectIssueTypeContext>newBuilder(
                new ProjectIssueTypeContextImpl(AllProjectsContext.INSTANCE, new IssueTypeContextImpl("it1"))
        ).asSet());

        assertEquals(expectedResult, result);
    }

    @Test
    public void testGetClauseContextMultiIssueTypeEquals() throws Exception {
        final MultiValueOperand operand = new MultiValueOperand("it2", "it1");
        TerminalClause clause = new TerminalClauseImpl(SystemSearchConstants.forIssueType().getJqlClauseNames().getPrimaryName(), Operator.EQUALS, operand);

        final IssueTypeClauseContextFactory factory = createNoopIssueTypeClauseContextFactory();

        final ClauseContext result = factory.getClauseContext(null, clause);
        ClauseContext expectedResult = new ClauseContextImpl(CollectionBuilder.<ProjectIssueTypeContext>newBuilder(
                new ProjectIssueTypeContextImpl(AllProjectsContext.INSTANCE, new IssueTypeContextImpl("it1")),
                new ProjectIssueTypeContextImpl(AllProjectsContext.INSTANCE, new IssueTypeContextImpl("it2"))
        ).asSet());

        assertEquals(expectedResult, result);
    }

    @Test
    public void testGetClauseContextMultiIssueTypeWithEmptyEquals() throws Exception {
        final MultiValueOperand operand = new MultiValueOperand("it1", "it2");
        TerminalClause clause = new TerminalClauseImpl(SystemSearchConstants.forIssueType().getJqlClauseNames().getPrimaryName(), Operator.EQUALS, operand);

        final IssueTypeClauseContextFactory factory = createNoopIssueTypeClauseContextFactory();

        final ClauseContext result = factory.getClauseContext(null, clause);
        ClauseContext expectedResult = new ClauseContextImpl(CollectionBuilder.<ProjectIssueTypeContext>newBuilder(
                new ProjectIssueTypeContextImpl(AllProjectsContext.INSTANCE, new IssueTypeContextImpl("it1")),
                new ProjectIssueTypeContextImpl(AllProjectsContext.INSTANCE, new IssueTypeContextImpl("it2"))
        ).asSet());

        assertEquals(expectedResult, result);

    }

    @Test
    public void testGetClauseContextSingleIssueTypeNotEquals() throws Exception {
        final SingleValueOperand operand = new SingleValueOperand("it1");
        TerminalClause clause = new TerminalClauseImpl(SystemSearchConstants.forIssueType().getJqlClauseNames().getPrimaryName(), Operator.NOT_EQUALS, operand);

        when(constantsManager.getAllIssueTypeIds()).thenReturn(CollectionBuilder.newBuilder("it1", "it2").asList());
        final IssueTypeClauseContextFactory factory = createNoopIssueTypeClauseContextFactory();

        final ClauseContext result = factory.getClauseContext(null, clause);
        ClauseContext expectedResult = new ClauseContextImpl(CollectionBuilder.<ProjectIssueTypeContext>newBuilder(
                new ProjectIssueTypeContextImpl(AllProjectsContext.INSTANCE, new IssueTypeContextImpl("it2"))
        ).asSet());

        assertEquals(expectedResult, result);
    }

    @Test
    public void testGetClauseContextMultiIssueTypeNotEquals() throws Exception {
        final Operand operand = new MultiValueOperand("it1");
        TerminalClause clause = new TerminalClauseImpl(SystemSearchConstants.forIssueType().getJqlClauseNames().getPrimaryName(), Operator.NOT_EQUALS, operand);

        when(constantsManager.getAllIssueTypeIds()).thenReturn(CollectionBuilder.newBuilder("it1", "it2", "it4").asList());

        IssueTypeClauseContextFactory factory = createNoopIssueTypeClauseContextFactory();

        final ClauseContext result = factory.getClauseContext(null, clause);
        ClauseContext expectedResult = new ClauseContextImpl(CollectionBuilder.<ProjectIssueTypeContext>newBuilder(
                new ProjectIssueTypeContextImpl(AllProjectsContext.INSTANCE, new IssueTypeContextImpl("it2")),
                new ProjectIssueTypeContextImpl(AllProjectsContext.INSTANCE, new IssueTypeContextImpl("it4"))
        ).asSet());

        assertEquals(expectedResult, result);
    }

    @Test
    public void testGetClauseContextForNullLiterals() throws Exception {
        final EmptyOperand operand = EmptyOperand.EMPTY;
        TerminalClause clause = new TerminalClauseImpl(SystemSearchConstants.forIssueType().getJqlClauseNames().getPrimaryName(), Operator.IS, operand);
        JqlOperandResolver mockJqlOperandResolver = mock(JqlOperandResolver.class);
        when(mockJqlOperandResolver.getValues((ApplicationUser) null, operand, clause)).thenReturn(null);


        factory = new IssueTypeClauseContextFactory(issueTypeResolver, mockJqlOperandResolver, constantsManager);

        final ClauseContext result = factory.getClauseContext(null, clause);
        ClauseContext expectedResult = ClauseContextImpl.createGlobalClauseContext();

        assertEquals(expectedResult, result);
    }

    @Test
    public void testGetClauseContextForEmpty() throws Exception {
        final EmptyOperand operand = EmptyOperand.EMPTY;
        TerminalClause clause = new TerminalClauseImpl(SystemSearchConstants.forIssueType().getJqlClauseNames().getPrimaryName(), Operator.IS, operand);

        final IssueTypeClauseContextFactory factory = createNoopIssueTypeClauseContextFactory();

        final ClauseContext result = factory.getClauseContext(null, clause);
        ClauseContext expectedResult = ClauseContextImpl.createGlobalClauseContext();

        assertEquals(expectedResult, result);

    }

    @Test
    public void testGetClauseContextForNotEmpty() throws Exception {
        final TerminalClause clause = new TerminalClauseImpl(SystemSearchConstants.forIssueType().getJqlClauseNames().getPrimaryName(), Operator.IS_NOT, EmptyOperand.EMPTY);
        final IssueTypeClauseContextFactory factory = createNoopIssueTypeClauseContextFactory();

        final ClauseContext result = factory.getClauseContext(null, clause);
        ClauseContext expectedResult = ClauseContextImpl.createGlobalClauseContext();
        assertEquals(expectedResult, result);

    }

    @Test
    public void testGetClauseContextForNotEmptyOnlyInList() throws Exception {
        final TerminalClause clause = new TerminalClauseImpl(SystemSearchConstants.forIssueType().getJqlClauseNames().getPrimaryName(), Operator.IS_NOT, new MultiValueOperand(EmptyOperand.EMPTY));
        final IssueTypeClauseContextFactory factory = createNoopIssueTypeClauseContextFactory();

        final ClauseContext result = factory.getClauseContext(null, clause);
        ClauseContext expectedResult = ClauseContextImpl.createGlobalClauseContext();
        assertEquals(expectedResult, result);

    }

    @Test
    public void testGetClauseContextForEmptyInList() throws Exception {
        final TerminalClause clause = new TerminalClauseImpl(SystemSearchConstants.forIssueType().getJqlClauseNames().getPrimaryName(), Operator.IN, new MultiValueOperand(EmptyOperand.EMPTY, new SingleValueOperand("it2")));
        final IssueTypeClauseContextFactory factory = createNoopIssueTypeClauseContextFactory();


        final ClauseContext actualResult = factory.getClauseContext(null, clause);
        final ClauseContext expectedResult = new ClauseContextImpl(
                Collections.<ProjectIssueTypeContext>singleton(
                        new ProjectIssueTypeContextImpl(AllProjectsContext.INSTANCE, new IssueTypeContextImpl("it2")
                        )));

        assertEquals(expectedResult, actualResult);

    }

    @Test
    public void testGetClauseContextForNotEmptyInList() throws Exception {
        final TerminalClause clause = new TerminalClauseImpl(SystemSearchConstants.forIssueType().getJqlClauseNames().getPrimaryName(), Operator.NOT_IN, new MultiValueOperand(EmptyOperand.EMPTY, new SingleValueOperand("it2")));
        final IssueTypeClauseContextFactory factory = createNoopIssueTypeClauseContextFactory();


        when(constantsManager.getAllIssueTypeIds()).thenReturn(CollectionBuilder.newBuilder("it1", "it2", "it4").asList());

        final ClauseContext actualResult = factory.getClauseContext(null, clause);
        final ClauseContext expectedResult = new ClauseContextImpl(
                CollectionBuilder.<ProjectIssueTypeContext>newBuilder(
                        new ProjectIssueTypeContextImpl(AllProjectsContext.INSTANCE, new IssueTypeContextImpl("it1")),
                        new ProjectIssueTypeContextImpl(AllProjectsContext.INSTANCE, new IssueTypeContextImpl("it4"))
                ).asSet());

        assertEquals(expectedResult, actualResult);

    }

    @Test
    public void testGetClauseContextForNotInAll() throws Exception {
        final TerminalClause clause = new TerminalClauseImpl(SystemSearchConstants.forIssueType().getJqlClauseNames().getPrimaryName(), Operator.NOT_IN, new MultiValueOperand("it1", "it2", "it4"));
        final IssueTypeClauseContextFactory factory = createNoopIssueTypeClauseContextFactory();

        when(constantsManager.getAllIssueTypeIds()).thenReturn(CollectionBuilder.newBuilder("it1", "it2", "it4").asList());

        final ClauseContext actualResult = factory.getClauseContext(null, clause);
        final ClauseContext expectedResult = ClauseContextImpl.createGlobalClauseContext();

        assertEquals(expectedResult, actualResult);

    }

    @Test
    public void testGetClauseContextBadOperator() throws Exception {
        final EmptyOperand operand = EmptyOperand.EMPTY;
        TerminalClause clause = new TerminalClauseImpl(SystemSearchConstants.forIssueType().getJqlClauseNames().getPrimaryName(), Operator.LESS_THAN, operand);

        final IssueTypeClauseContextFactory factory = createNoopIssueTypeClauseContextFactory();

        final ClauseContext result = factory.getClauseContext(null, clause);
        ClauseContext expectedResult = ClauseContextImpl.createGlobalClauseContext();

        assertEquals(expectedResult, result);

    }

    @Test
    public void testGetIdStringValue() throws Exception {
        final List<String> ids = CollectionBuilder.newBuilder("name").asList();
        when(issueTypeResolver.getIdsFromName("name")).thenReturn(ids);


        QueryLiteral literal = createLiteral("name");

        assertEquals(ids, factory.getIds(literal));
    }

    @Test
    public void testGetIdLongValue() throws Exception {
        final List<String> ids = CollectionBuilder.newBuilder("10").asList();
        when(issueTypeResolver.idExists(10L)).thenReturn(true);


        QueryLiteral literal = createLiteral(10L);

        assertEquals(ids, factory.getIds(literal));
    }

    private IssueTypeClauseContextFactory createNoopIssueTypeClauseContextFactory() {
        return new IssueTypeClauseContextFactory(issueTypeResolver, jqlOperandResolver, constantsManager) {
            @Override
            List<String> getIds(final QueryLiteral value) {
                return CollectionBuilder.newBuilder(value.getStringValue()).asList();
            }
        };
    }
}