123 lines
3.3 KiB
C#
123 lines
3.3 KiB
C#
// ****************************************************************
|
|
// This is free software licensed under the NUnit license. You
|
|
// may obtain a copy of the license as well as information regarding
|
|
// copyright ownership at http://nunit.org/?p=license&r=2.4.
|
|
// ****************************************************************
|
|
|
|
namespace NUnit.Core
|
|
{
|
|
using System;
|
|
using System.IO;
|
|
using System.Reflection;
|
|
using System.Collections;
|
|
|
|
/// <summary>
|
|
/// Class adapted from NUnitAddin for use in handling assemblies that are not
|
|
/// found in the test AppDomain.
|
|
/// </summary>
|
|
public class AssemblyResolver : MarshalByRefObject, IDisposable
|
|
{
|
|
private class AssemblyCache
|
|
{
|
|
private Hashtable _resolved = new Hashtable();
|
|
|
|
public bool Contains( string name )
|
|
{
|
|
return _resolved.ContainsKey( name );
|
|
}
|
|
|
|
public Assembly Resolve( string name )
|
|
{
|
|
if ( _resolved.ContainsKey( name ) )
|
|
return (Assembly)_resolved[name];
|
|
|
|
return null;
|
|
}
|
|
|
|
public void Add( string name, Assembly assembly )
|
|
{
|
|
_resolved[name] = assembly;
|
|
}
|
|
}
|
|
|
|
private AssemblyCache _cache = new AssemblyCache();
|
|
|
|
private ArrayList _dirs = new ArrayList();
|
|
|
|
public AssemblyResolver()
|
|
{
|
|
AppDomain.CurrentDomain.AssemblyResolve += new ResolveEventHandler(CurrentDomain_AssemblyResolve);
|
|
}
|
|
|
|
public void Dispose()
|
|
{
|
|
AppDomain.CurrentDomain.AssemblyResolve -= new ResolveEventHandler(CurrentDomain_AssemblyResolve);
|
|
}
|
|
|
|
public void AddFile( string file )
|
|
{
|
|
Assembly assembly = Assembly.LoadFrom( file );
|
|
_cache.Add(assembly.GetName().FullName, assembly);
|
|
}
|
|
|
|
public void AddFiles( string directory, string pattern )
|
|
{
|
|
if ( Directory.Exists( directory ) )
|
|
foreach( string file in Directory.GetFiles( directory, pattern ) )
|
|
AddFile( file );
|
|
}
|
|
|
|
public void AddDirectory( string directory )
|
|
{
|
|
if ( Directory.Exists( directory ) )
|
|
_dirs.Add( directory );
|
|
}
|
|
|
|
private Assembly CurrentDomain_AssemblyResolve(object sender, ResolveEventArgs args)
|
|
{
|
|
string fullName = args.Name;
|
|
int index = fullName.IndexOf(',');
|
|
if(index == -1) // Only resolve using full name.
|
|
{
|
|
NTrace.Debug( string.Format("Not a strong name: {0}", fullName ),
|
|
"'AssemblyResolver'" );
|
|
return null;
|
|
}
|
|
|
|
if ( _cache.Contains( fullName ) )
|
|
{
|
|
NTrace.Info( string.Format( "Resolved from Cache: {0}", fullName ),
|
|
"'AssemblyResolver'" );
|
|
return _cache.Resolve(fullName);
|
|
}
|
|
|
|
foreach( string dir in _dirs )
|
|
{
|
|
foreach( string file in Directory.GetFiles( dir, "*.dll" ) )
|
|
{
|
|
string fullFile = Path.Combine( dir, file );
|
|
try
|
|
{
|
|
if ( AssemblyName.GetAssemblyName( fullFile ).FullName == fullName )
|
|
{
|
|
NTrace.Info( string.Format( "Added to Cache: {0}", fullFile ),
|
|
"'AssemblyResolver'" );
|
|
AddFile( fullFile );
|
|
return _cache.Resolve( fullName );
|
|
}
|
|
}
|
|
catch
|
|
{
|
|
// Keep going if there's a bad assembly
|
|
NTrace.Debug( string.Format( "Bad assembly: {0}", fullFile ), "AssemblyResolver");
|
|
}
|
|
}
|
|
}
|
|
|
|
NTrace.Debug( string.Format( "Not in Cache: {0}", fullName),
|
|
"'AssemblyResolver'");
|
|
return null;
|
|
}
|
|
}
|
|
}
|