/*
 * 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.maven.shared.artifact.filter;

import org.apache.maven.artifact.Artifact;
import org.apache.maven.artifact.resolver.filter.ArtifactFilter;
import org.apache.maven.artifact.versioning.VersionRange;
import org.apache.maven.plugin.testing.ArtifactStubFactory;
import org.junit.Test;

import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

public class ScopeArtifactFilterTest {

    @Test
    public void testExcludedArtifactWithRangeShouldNotCauseNPE() throws Exception {
        ArtifactStubFactory factory = new ArtifactStubFactory();

        Artifact excluded = factory.createArtifact(
                "group",
                "artifact",
                VersionRange.createFromVersionSpec("[1.2.3]"),
                Artifact.SCOPE_PROVIDED,
                "jar",
                null,
                false);

        ArtifactFilter filter = new ScopeArtifactFilter(Artifact.SCOPE_RUNTIME);

        assertFalse(filter.include(excluded));
    }

    @Test
    public void testNullScopeDisabled() {
        ScopeArtifactFilter filter = new ScopeArtifactFilter();
        filter.setIncludeNullScope(false);

        verifyExcluded(filter, null);
    }

    @Test
    public void testFineGrained_IncludeOnlyScopesThatWereEnabled_TestScope() {
        ScopeArtifactFilter filter = new ScopeArtifactFilter();
        filter.setIncludeTestScope(true);

        verifyExcluded(filter, Artifact.SCOPE_COMPILE);
        verifyExcluded(filter, Artifact.SCOPE_PROVIDED);
        verifyExcluded(filter, Artifact.SCOPE_RUNTIME);
        verifyExcluded(filter, Artifact.SCOPE_SYSTEM);
        verifyIncluded(filter, Artifact.SCOPE_TEST);
        verifyIncluded(filter, null);
    }

    @Test
    public void testFineGrained_IncludeOnlyScopesThatWereEnabled_CompileScope() {
        ScopeArtifactFilter filter = new ScopeArtifactFilter();
        filter.setIncludeCompileScope(true);

        verifyIncluded(filter, Artifact.SCOPE_COMPILE);
        verifyExcluded(filter, Artifact.SCOPE_PROVIDED);
        verifyExcluded(filter, Artifact.SCOPE_RUNTIME);
        verifyExcluded(filter, Artifact.SCOPE_SYSTEM);
        verifyExcluded(filter, Artifact.SCOPE_TEST);
        verifyIncluded(filter, null);
    }

    @Test
    public void testFineGrained_IncludeOnlyScopesThatWereEnabled_RuntimeScope() {
        ScopeArtifactFilter filter = new ScopeArtifactFilter();
        filter.setIncludeRuntimeScope(true);

        verifyExcluded(filter, Artifact.SCOPE_COMPILE);
        verifyExcluded(filter, Artifact.SCOPE_PROVIDED);
        verifyIncluded(filter, Artifact.SCOPE_RUNTIME);
        verifyExcluded(filter, Artifact.SCOPE_SYSTEM);
        verifyExcluded(filter, Artifact.SCOPE_TEST);
        verifyIncluded(filter, null);
    }

    @Test
    public void testFineGrained_IncludeOnlyScopesThatWereEnabled_ProvidedScope() {
        ScopeArtifactFilter filter = new ScopeArtifactFilter();
        filter.setIncludeProvidedScope(true);

        verifyExcluded(filter, Artifact.SCOPE_COMPILE);
        verifyIncluded(filter, Artifact.SCOPE_PROVIDED);
        verifyExcluded(filter, Artifact.SCOPE_RUNTIME);
        verifyExcluded(filter, Artifact.SCOPE_SYSTEM);
        verifyExcluded(filter, Artifact.SCOPE_TEST);
        verifyIncluded(filter, null);
    }

    @Test
    public void testFineGrained_IncludeOnlyScopesThatWereEnabled_SystemScope() {
        ScopeArtifactFilter filter = new ScopeArtifactFilter();
        filter.setIncludeSystemScope(true);

        verifyExcluded(filter, Artifact.SCOPE_COMPILE);
        verifyExcluded(filter, Artifact.SCOPE_PROVIDED);
        verifyExcluded(filter, Artifact.SCOPE_RUNTIME);
        verifyIncluded(filter, Artifact.SCOPE_SYSTEM);
        verifyExcluded(filter, Artifact.SCOPE_TEST);
        verifyIncluded(filter, null);
    }

    @Test
    public void testFineGrained_IncludeOnlyScopesThatWereEnabled_ProvidedAndRuntimeScopes() {
        ScopeArtifactFilter filter = new ScopeArtifactFilter();
        filter.setIncludeRuntimeScope(true);
        filter.setIncludeProvidedScope(true);

        verifyExcluded(filter, Artifact.SCOPE_COMPILE);
        verifyIncluded(filter, Artifact.SCOPE_PROVIDED);
        verifyIncluded(filter, Artifact.SCOPE_RUNTIME);
        verifyExcluded(filter, Artifact.SCOPE_SYSTEM);
        verifyExcluded(filter, Artifact.SCOPE_TEST);
        verifyIncluded(filter, null);
    }

    @Test
    public void testFineGrained_IncludeOnlyScopesThatWereEnabled_SystemAndRuntimeScopes() {
        ScopeArtifactFilter filter = new ScopeArtifactFilter();
        filter.setIncludeRuntimeScope(true);
        filter.setIncludeSystemScope(true);

        verifyExcluded(filter, Artifact.SCOPE_COMPILE);
        verifyExcluded(filter, Artifact.SCOPE_PROVIDED);
        verifyIncluded(filter, Artifact.SCOPE_RUNTIME);
        verifyIncluded(filter, Artifact.SCOPE_SYSTEM);
        verifyExcluded(filter, Artifact.SCOPE_TEST);
        verifyIncluded(filter, null);
    }

    @Test
    public void
            testFineGrainedWithImplications_CompileScopeShouldIncludeOnlyArtifactsWithNullSystemProvidedOrCompileScopes() {
        ScopeArtifactFilter filter = new ScopeArtifactFilter();
        filter.setIncludeCompileScopeWithImplications(true);

        verifyIncluded(filter, null);
        verifyIncluded(filter, Artifact.SCOPE_COMPILE);
        verifyIncluded(filter, Artifact.SCOPE_PROVIDED);
        verifyIncluded(filter, Artifact.SCOPE_SYSTEM);

        verifyExcluded(filter, Artifact.SCOPE_RUNTIME);
        verifyExcluded(filter, Artifact.SCOPE_TEST);
    }

    @Test
    public void testFineGrainedWithImplications_RuntimeScopeShouldIncludeOnlyArtifactsWithNullRuntimeOrCompileScopes() {
        ScopeArtifactFilter filter = new ScopeArtifactFilter();
        filter.setIncludeRuntimeScopeWithImplications(true);

        verifyIncluded(filter, null);
        verifyIncluded(filter, Artifact.SCOPE_COMPILE);
        verifyIncluded(filter, Artifact.SCOPE_RUNTIME);

        verifyExcluded(filter, Artifact.SCOPE_PROVIDED);
        verifyExcluded(filter, Artifact.SCOPE_SYSTEM);
        verifyExcluded(filter, Artifact.SCOPE_TEST);
    }

    @Test
    public void testFineGrainedWithImplications_TestScopeShouldIncludeAllScopes() {
        ScopeArtifactFilter filter = new ScopeArtifactFilter();
        filter.setIncludeTestScopeWithImplications(true);

        verifyIncluded(filter, null);
        verifyIncluded(filter, Artifact.SCOPE_COMPILE);
        verifyIncluded(filter, Artifact.SCOPE_RUNTIME);

        verifyIncluded(filter, Artifact.SCOPE_PROVIDED);
        verifyIncluded(filter, Artifact.SCOPE_SYSTEM);
        verifyIncluded(filter, Artifact.SCOPE_TEST);
    }

    @Test
    public void testScopesShouldIncludeArtifactWithSameScope() {
        verifyIncluded(Artifact.SCOPE_COMPILE, Artifact.SCOPE_COMPILE);
        verifyIncluded(Artifact.SCOPE_PROVIDED, Artifact.SCOPE_PROVIDED);
        verifyIncluded(Artifact.SCOPE_RUNTIME, Artifact.SCOPE_RUNTIME);
        verifyIncluded(Artifact.SCOPE_SYSTEM, Artifact.SCOPE_SYSTEM);
        verifyIncluded(Artifact.SCOPE_TEST, Artifact.SCOPE_TEST);
        verifyIncluded((String) null, null);
    }

    @Test
    public void testCompileScopeShouldIncludeOnlyArtifactsWithNullSystemProvidedOrCompileScopes() {
        String scope = Artifact.SCOPE_COMPILE;

        verifyIncluded(scope, null);
        verifyIncluded(scope, Artifact.SCOPE_COMPILE);
        verifyIncluded(scope, Artifact.SCOPE_PROVIDED);
        verifyIncluded(scope, Artifact.SCOPE_SYSTEM);

        verifyExcluded(scope, Artifact.SCOPE_RUNTIME);
        verifyExcluded(scope, Artifact.SCOPE_TEST);
    }

    @Test
    public void testRuntimeScopeShouldIncludeOnlyArtifactsWithNullRuntimeOrCompileScopes() {
        String scope = Artifact.SCOPE_RUNTIME;

        verifyIncluded(scope, null);
        verifyIncluded(scope, Artifact.SCOPE_COMPILE);
        verifyIncluded(scope, Artifact.SCOPE_RUNTIME);

        verifyExcluded(scope, Artifact.SCOPE_PROVIDED);
        verifyExcluded(scope, Artifact.SCOPE_SYSTEM);
        verifyExcluded(scope, Artifact.SCOPE_TEST);
    }

    @Test
    public void testTestScopeShouldIncludeAllScopes() {
        String scope = Artifact.SCOPE_TEST;

        verifyIncluded(scope, null);
        verifyIncluded(scope, Artifact.SCOPE_COMPILE);
        verifyIncluded(scope, Artifact.SCOPE_RUNTIME);

        verifyIncluded(scope, Artifact.SCOPE_PROVIDED);
        verifyIncluded(scope, Artifact.SCOPE_SYSTEM);
        verifyIncluded(scope, Artifact.SCOPE_TEST);
    }

    @Test
    public void testProvidedScopeShouldIncludeOnlyArtifactsWithNullOrProvidedScopes() {
        String scope = Artifact.SCOPE_PROVIDED;

        verifyIncluded(scope, null);
        verifyExcluded(scope, Artifact.SCOPE_COMPILE);
        verifyExcluded(scope, Artifact.SCOPE_RUNTIME);

        verifyIncluded(scope, Artifact.SCOPE_PROVIDED);

        verifyExcluded(scope, Artifact.SCOPE_SYSTEM);
        verifyExcluded(scope, Artifact.SCOPE_TEST);
    }

    @Test
    public void testSystemScopeShouldIncludeOnlyArtifactsWithNullOrSystemScopes() {
        String scope = Artifact.SCOPE_SYSTEM;

        verifyIncluded(scope, null);
        verifyExcluded(scope, Artifact.SCOPE_COMPILE);
        verifyExcluded(scope, Artifact.SCOPE_RUNTIME);
        verifyExcluded(scope, Artifact.SCOPE_PROVIDED);

        verifyIncluded(scope, Artifact.SCOPE_SYSTEM);

        verifyExcluded(scope, Artifact.SCOPE_TEST);
    }

    private void verifyIncluded(String filterScope, String artifactScope) {
        Artifact artifact = createMockArtifact(artifactScope);

        ArtifactFilter filter = new ScopeArtifactFilter(filterScope);

        assertTrue(
                "Artifact scope: " + artifactScope + " NOT included using filter scope: " + filterScope,
                filter.include(artifact));
    }

    private void verifyExcluded(String filterScope, String artifactScope) {
        Artifact artifact = createMockArtifact(artifactScope);

        ArtifactFilter filter = new ScopeArtifactFilter(filterScope);

        assertFalse(
                "Artifact scope: " + artifactScope + " NOT excluded using filter scope: " + filterScope,
                filter.include(artifact));
    }

    private void verifyIncluded(ScopeArtifactFilter filter, String artifactScope) {
        Artifact artifact = createMockArtifact(artifactScope);

        assertTrue("Artifact scope: " + artifactScope + " SHOULD BE included", filter.include(artifact));
    }

    private void verifyExcluded(ScopeArtifactFilter filter, String artifactScope) {
        Artifact artifact = createMockArtifact(artifactScope);

        assertFalse("Artifact scope: " + artifactScope + " SHOULD BE excluded", filter.include(artifact));
    }

    private Artifact createMockArtifact(String scope) {
        Artifact artifact = mock(Artifact.class);

        when(artifact.getScope()).thenReturn(scope);
        when(artifact.getId()).thenReturn("group:artifact:type:version");

        return artifact;
    }
}