FunctionOverridingTests.cs

#region Copyright (c) 2007 Ryan Williams <drcforbin@gmail.com>
/// <copyright>
/// Copyright (c) 2007 Ryan Williams <drcforbin@gmail.com>
/// 
/// Permission is hereby granted, free of charge, to any person obtaining a copy
/// of this software and associated documentation files (the "Software"), to deal
/// in the Software without restriction, including without limitation the rights
/// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
/// copies of the Software, and to permit persons to whom the Software is
/// furnished to do so, subject to the following conditions:
/// 
/// The above copyright notice and this permission notice shall be included in
/// all copies or substantial portions of the Software.
/// 
/// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
/// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
/// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
/// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
/// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
/// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
/// THE SOFTWARE.
/// </copyright>
#endregion
using System;
using System.IO;
using System.Collections.Generic;
using System.Text;
using System.CodeDom.Compiler;
using NUnit.Framework;
using Mono.Cecil;
using Obfuscar;

namespace ObfuscarTests
{
	[TestFixture]
	public class FunctionOverridingTests
	{
		Obfuscar.ObfuscationMap BuildAndObfuscateAssemblies ()
		{
			string xml = String.Format (
				             @"<?xml version='1.0'?>" +
				             @"<Obfuscator>" +
				             @"<Var name='InPath' value='{0}' />" +
				             @"<Var name='OutPath' value='{1}' />" +
				             @"<Var name='KeepPublicApi' value='false' />" +
				             @"<Var name='HidePrivateApi' value='true' />" +
				             @"<Module file='$(InPath)\AssemblyWithOverrides.dll' />" +
				             @"</Obfuscator>", TestHelper.InputPath, TestHelper.OutputPath);

			Obfuscar.Obfuscator obfuscator = TestHelper.BuildAndObfuscate ("AssemblyWithOverrides", String.Empty, xml);

			return obfuscator.Mapping;
		}

		MethodDefinition FindByName (TypeDefinition typeDef, string name)
		{
			foreach (MethodDefinition method in typeDef.Methods)
				if (method.Name == name)
					return method;

			Assert.Fail (String.Format ("Expected to find method: {0}", name));
			return null; // never here
		}

		[Test]
		public void CheckClassHasAttribute ()
		{
			Obfuscar.ObfuscationMap map = BuildAndObfuscateAssemblies ();

			string assmName = "AssemblyWithOverrides.dll";

			AssemblyDefinition inAssmDef = AssemblyDefinition.ReadAssembly (
				                               Path.Combine (TestHelper.InputPath, assmName));

			AssemblyDefinition outAssmDef = AssemblyDefinition.ReadAssembly (
				                                Path.Combine (TestHelper.OutputPath, assmName));

			TypeDefinition classAType = inAssmDef.MainModule.GetType ("TestClasses.ClassA");
			MethodDefinition classAmethod2 = FindByName (classAType, "Method2");
			MethodDefinition classAcompare = FindByName (classAType, "CompareTo");

			TypeDefinition classBType = inAssmDef.MainModule.GetType ("TestClasses.ClassB");
			MethodDefinition classBmethod2 = FindByName (classBType, "Method2");
			MethodDefinition classBcompare = FindByName (classBType, "CompareTo");

			TypeDefinition classCType = inAssmDef.MainModule.GetType ("TestClasses.ClassC");
			MethodDefinition classCmethod1 = FindByName (classCType, "Method1");

			TypeDefinition classDType = inAssmDef.MainModule.GetType ("TestClasses.ClassD");
			MethodDefinition classDmethod1 = FindByName (classDType, "Method1");

			Obfuscar.ObfuscatedThing classAEntry = map.GetMethod (new Obfuscar.MethodKey (classAmethod2));
			ObfuscatedThing classACompareEntry = map.GetMethod (new MethodKey (classAcompare));
			Obfuscar.ObfuscatedThing classBEntry = map.GetMethod (new Obfuscar.MethodKey (classBmethod2));
			Obfuscar.ObfuscatedThing classBCompareEntry = map.GetMethod (new Obfuscar.MethodKey (classBcompare));
			ObfuscatedThing classCEntry = map.GetMethod (new MethodKey (classCmethod1));
			ObfuscatedThing classDEntry = map.GetMethod (new MethodKey (classDmethod1));

			var classFType = inAssmDef.MainModule.GetType ("TestClasses.ClassF");
			var classFmethod = FindByName (classFType, "Test");

			var classGType = inAssmDef.MainModule.GetType ("TestClasses.ClassG");
			var classGmethod = FindByName (classGType, "Test");

			var classFEntry = map.GetMethod (new MethodKey (classFmethod));
			var classGEntry = map.GetMethod (new MethodKey (classGmethod));

			Assert.IsTrue (
				classAEntry.Status == Obfuscar.ObfuscationStatus.Renamed &&
				classBEntry.Status == Obfuscar.ObfuscationStatus.Renamed,
				"Both methods should have been renamed.");
			
			Assert.IsTrue (
				classAEntry.StatusText == classBEntry.StatusText,
				"Both methods should have been renamed to the same thing.");

			Assert.IsTrue (classACompareEntry.Status == ObfuscationStatus.Skipped);

			Assert.IsTrue (classBCompareEntry.Status == ObfuscationStatus.Skipped);

			Assert.IsTrue (classCEntry.Status == ObfuscationStatus.Renamed);

			Assert.IsTrue (classDEntry.Status == ObfuscationStatus.Renamed);

			Assert.IsTrue (classFEntry.Status == ObfuscationStatus.Renamed && classGEntry.Status == ObfuscationStatus.Renamed, "Both methods should have been renamed.");

			Assert.IsTrue (classFEntry.StatusText == classGEntry.StatusText, "Both methods should have been renamed to the same thing.");
		}
	}
}