MavenDepenenciesUtilsTest.java

/*
 * Copyright 2014 Bernd Vogt and others.
 * 
 * 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 org.sourcepit.b2.internal.maven.util;

import static org.junit.Assert.assertEquals;

import java.util.List;

import org.apache.maven.model.Dependency;
import org.apache.maven.model.Model;
import org.apache.maven.model.Profile;
import org.junit.Test;

import com.google.common.base.Predicate;

public class MavenDepenenciesUtilsTest {

   @Test
   public void testRemoveDependencies() {
      final Model model = new Model();
      model.getDependencies().add(newDependency("foo", "bar", "1", "module"));
      model.getDependencies().add(newDependency("foo", "bar", "1", "jar"));

      final Profile profile = new Profile();
      profile.getDependencies().add(newDependency("foo", "bar", "2", "module"));
      profile.getDependencies().add(newDependency("foo", "bar", "2", "jar"));

      model.getProfiles().add(profile);

      MavenDepenenciesUtils.removeDependencies(model, new Predicate<Dependency>() {
         public boolean apply(Dependency input) {
            return "module".equals(input.getType());
         }
      });

      List<Dependency> dependencies = model.getDependencies();
      assertEquals(1, dependencies.size());
      assertEquals("jar", dependencies.get(0).getType());

      dependencies = profile.getDependencies();
      assertEquals(1, dependencies.size());
      assertEquals("jar", dependencies.get(0).getType());
   }

   private static Dependency newDependency(String groupId, String artifactId, String version, String type) {
      final Dependency dependency = new Dependency();
      dependency.setGroupId(groupId);
      dependency.setArtifactId(artifactId);
      dependency.setVersion(version);
      if (type != null) {
         dependency.setType(type);
      }
      return dependency;
   }

}