/*
* 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 npanday.dao.impl;
import junit.framework.TestCase;
import npanday.dao.*;
import org.apache.maven.artifact.Artifact;
import org.apache.maven.artifact.repository.ArtifactRepository;
import org.apache.maven.artifact.resolver.ArtifactResolver;
import org.openrdf.repository.RepositoryConnection;
import org.openrdf.repository.RepositoryException;
import org.openrdf.repository.sail.SailRepository;
import org.openrdf.rio.RDFHandler;
import org.openrdf.rio.RDFHandlerException;
import org.openrdf.rio.rdfxml.RDFXMLWriter;
import org.openrdf.sail.memory.MemoryStore;
import org.openrdf.sail.memory.MemoryStoreRDFSInferencer;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.net.URI;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.Set;
import static org.mockito.Mockito.mock;
public class ProjectDaoImplTest
extends TestCase
{
private static File basedir = new File( System.getProperty( "basedir" ) );
private static File localRepository = new File( System.getProperty( "basedir" ), "/target/local-test-repo" );
private org.openrdf.repository.Repository rdfRepository;
private File dataDir;
public void testBogus()
{
}
private ProjectDao createProjectDao()
{
dataDir = new File( basedir, ( "/target/rdf-repos/rdf-repo-" + System.currentTimeMillis() ) );
rdfRepository = new SailRepository( new MemoryStoreRDFSInferencer( new MemoryStore( dataDir ) ) );
try
{
rdfRepository.initialize();
}
catch ( RepositoryException e )
{
return null;
}
ProjectDaoImpl dao = new ProjectDaoImpl();
ArtifactResolver stub = mock(ArtifactResolver.class);
dao.initForUnitTest( rdfRepository, "", "", stub, new ArtifactFactoryTestStub() );
dao.openConnection();
return dao;
}
public void testStore_WithGacDependencies()
{
ProjectDao dao = this.createProjectDao();
Project project = new Project();
project.setGroupId( "NPanday" );
project.setArtifactId( "NPanday.Test5" );
project.setVersion( "1.0.0" );
project.setArtifactType( "library" );
ProjectDependency test2 =
createProjectDependency( "Microsoft.Build.Conversion", "Microsoft.Build.Conversion", "2.0.0.0", "gac_msil",
"31bf3856ad364e35" );
project.addProjectDependency( test2 );
try
{
dao.storeProjectAndResolveDependencies( project, localRepository, new ArrayList<ArtifactRepository>() );
}
catch ( Exception e )
{
e.printStackTrace();
fail( "Could not store the project: " + e.getMessage() );
}
Set<Project> projects = null;
try
{
projects = dao.getAllProjects();
}
catch ( Exception e )
{
e.printStackTrace();
fail( "Could not retrieve the project: " + e.getMessage() );
}
dao.closeConnection();
}
public void testGetAllProjects()
{
ProjectDao dao = this.createProjectDao();
Project project = new Project();
project.setGroupId( "NPanday" );
project.setArtifactId( "NPanday.Test5" );
project.setVersion( "1.0.0" );
project.setArtifactType( "library" );
ProjectDependency test2 =
createProjectDependency( "Microsoft.VisualBasic", "Microsoft.VisualBasic", "8.0.0.0", "gac_msil",
"31bf3856ad364e35" );
project.addProjectDependency( test2 );
try
{
dao.storeProjectAndResolveDependencies( project, localRepository, new ArrayList<ArtifactRepository>() );
}
catch ( Exception e )
{
e.printStackTrace();
fail( "Could not store the project: " + e.getMessage() );
}
Set<Project> projects = null;
try
{
projects = dao.getAllProjects();
}
catch ( Exception e )
{
e.printStackTrace();
fail( "Could not retrieve the project: " + e.getMessage() );
}
dao.closeConnection();
}
public void testStore_WithRequirements()
{
ProjectDao dao = this.createProjectDao();
Project project = new Project();
project.setGroupId( "NPanday.Model" );
project.setArtifactId( "NPanday.Model.Pom" );
project.setVersion( "0.14.0.0" );
project.setPublicKeyTokenId( "b03f5f7f11d50a3a" );
project.setArtifactType( "library" );
Set<Requirement> requirements = new HashSet<Requirement>();
try
{
requirements.add( Requirement.Factory.createDefaultRequirement(
new URI( ProjectUri.VENDOR.getPredicate() ),
"MICROSOFT" ) );
requirements.add( Requirement.Factory.createDefaultRequirement(
new URI(
ProjectUri.FRAMEWORK_VERSION.getPredicate() ),
"2.0" ) );
}
catch ( Exception e )
{
e.printStackTrace();
}
project.setRequirements( requirements );
Set<Artifact> artifacts = null;
try
{
artifacts =
dao.storeProjectAndResolveDependencies( project, localRepository, new ArrayList<ArtifactRepository>() );
}
catch ( Exception e )
{
e.printStackTrace();
fail( "Could not store the project: " + e.getMessage() );
}
Project proj = null;
try
{
proj = dao.getProjectFor( "NPanday.Model", "NPanday.Model.Pom", "0.14.0.0", "library", null );
}
catch ( Exception e )
{
e.printStackTrace();
fail( "Could not query the project: " + e.getMessage() );
}
requirements = proj.getRequirements();
assertEquals( "Incorrect number of requirements.", 2, requirements.size() );
assertTrue( "Could not find framework requirement",
hasRequirement( ProjectUri.FRAMEWORK_VERSION.getPredicate(), "2.0", requirements ) );
assertTrue( "Could not find vendor requirement", hasRequirement( ProjectUri.VENDOR.getPredicate(), "MICROSOFT",
requirements ) );
dao.closeConnection();
}
private boolean hasRequirement( String uri, String value, Set<Requirement> requirements )
{
for ( Requirement requirement : requirements )
{
if ( uri.equals( requirement.getUri().toString() ) && value.equals( requirement.getValue() ) )
{
return true;
}
}
return false;
}
private boolean hasDependency( String groupId, String artifactId, String version,
Set<ProjectDependency> projectDependencies )
{
for ( ProjectDependency projectDependency : projectDependencies )
{
if ( projectDependency.getGroupId().equals( groupId )
&& projectDependency.getArtifactId().equals( artifactId )
&& projectDependency.getVersion().equals( version ) )
{
return true;
}
}
return false;
}
public void testStoreDependency()
{
ProjectDao dao = this.createProjectDao();
Project project1 = new Project();
project1.setGroupId( "NPanday" );
project1.setArtifactId( "NPanday.Plugin" );
project1.setVersion( "0.14.0.0" );
project1.setArtifactType( "library" );
project1.addProjectDependency( this.createProjectDependency( "NPanday", "NPanday.Test4", "1.0.0" ) );
Set<Artifact> artifacts = null;
try
{
artifacts =
dao.storeProjectAndResolveDependencies( project1, localRepository, new ArrayList<ArtifactRepository>() );
}
catch ( Exception e )
{
e.printStackTrace();
fail( "Could not store the project: " + e.getMessage() );
}
assertEquals( "Incorrect number of returned artifacts", 1, artifacts.size() );
Artifact artifact = (Artifact) artifacts.toArray()[0];
assertEquals( "Group Id is incorrect", "NPanday", artifact.getGroupId() );
assertEquals( "Artifact Id is incorrect", "NPanday.Test4", artifact.getArtifactId() );
assertEquals( "Version is incorrect", "1.0.0", artifact.getVersion() );
Project testProject = null;
try
{
testProject = dao.getProjectFor( "NPanday", "NPanday.Plugin", "0.14.0.0", "library", null );
}
catch ( Exception e )
{
e.printStackTrace();
fail( "Could not retrieve the project: " + e.getMessage() );
}
this.exportRepositoryToRdf( "testDependency-rdf.xml" );
Set<ProjectDependency> projectDependencies = testProject.getProjectDependencies();
assertEquals( "Incorrect number of dependencies", 1, projectDependencies.size() );
ProjectDependency projectDependency = (ProjectDependency) projectDependencies.toArray()[0];
assertTrue( "Could not find required dependency. Found Dependency: GroupId = " + projectDependency.getGroupId()
+ ", Artifact Id = " + projectDependency.getArtifactId(), this.hasDependency( "NPanday", "NPanday.Test4",
"1.0.0", projectDependencies ) );
dao.closeConnection();
}
public void testSingleStore()
{
ProjectDao dao = this.createProjectDao();
Project project = new Project();
project.setGroupId( "NPanday.Model" );
project.setArtifactId( "NPanday.Model.Pom" );
project.setVersion( "0.14.0.0" );
project.setArtifactType( "library" );
Set<Artifact> artifacts = null;
try
{
artifacts =
dao.storeProjectAndResolveDependencies( project, localRepository, new ArrayList<ArtifactRepository>() );
}
catch ( Exception e )
{
e.printStackTrace();
fail( "Could not store the project: " + e.getMessage() );
}
Project proj = null;
try
{
proj = dao.getProjectFor( "NPanday.Model", "NPanday.Model.Pom", "0.14.0.0", "library", null );
}
catch ( Exception e )
{
e.printStackTrace();
fail( "Could not query the project: " + e.getMessage() );
}
assertEquals( "Group IDs do not match", project.getGroupId(), proj.getGroupId() );
assertEquals( "Artifact IDs do not match", project.getArtifactId(), proj.getArtifactId() );
assertEquals( "Versions do not match", project.getVersion(), proj.getVersion() );
assertEquals( "Found dependency when there should be none", 0, artifacts.size() );
dao.closeConnection();
}
public void testSingleStore_WithIncorrectVersion()
{
ProjectDao dao = this.createProjectDao();
Project project = new Project();
project.setGroupId( "NPanday.Model" );
project.setArtifactId( "NPanday.Model.Pom" );
project.setVersion( "0.14.0.0" );
project.setArtifactType( "library" );
try
{
dao.storeProjectAndResolveDependencies( project, localRepository, new ArrayList<ArtifactRepository>() );
}
catch ( Exception e )
{
e.printStackTrace();
fail( e.getMessage() );
}
try
{
dao.getProjectFor( "NPanday.Model", "NPanday.Model.Pom", "0.15.0.0", "library", null );
}
catch ( Exception e )
{
return;
}
dao.closeConnection();
fail( "Found project when none should exist." );
}
private ProjectDependency createProjectDependency( String groupId, String artifactId, String version )
{
ProjectDependency projectDependency = new ProjectDependency();
projectDependency.setGroupId( groupId );
projectDependency.setArtifactId( artifactId );
projectDependency.setVersion( version );
projectDependency.setArtifactType( "library" );
return projectDependency;
}
private ProjectDependency createProjectDependency( String groupId, String artifactId, String version, String type,
String classifier )
{
ProjectDependency projectDependency = new ProjectDependency();
projectDependency.setGroupId( groupId );
projectDependency.setArtifactId( artifactId );
projectDependency.setVersion( version );
projectDependency.setArtifactType( "library" );
projectDependency.setPublicKeyTokenId( classifier );
projectDependency.setArtifactType( type );
return projectDependency;
}
private void exportRepositoryToRdf( String fileName )
{
RDFHandler rdfxmlWriter;
try
{
File exportFile = new File( localRepository.getParentFile(), fileName );
rdfxmlWriter = new RDFXMLWriter( new FileOutputStream( exportFile ) );
}
catch ( IOException e )
{
// fail( e.getMessage() );
return;
}
try
{
RepositoryConnection repositoryConnection = rdfRepository.getConnection();
repositoryConnection.export( rdfxmlWriter );
}
catch ( RepositoryException e )
{
e.printStackTrace();
}
catch ( RDFHandlerException e )
{
e.printStackTrace();
}
}
}