AutoSkipTypeTests.cs

using Mono.Cecil;
using NUnit.Framework;
using Obfuscar;
using System;
using System.Collections.Generic;
using System.IO;

namespace ObfuscarTests
{
	[TestFixture]
	public class AutoSkipTypeTests
	{
		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 CheckHidePrivateApiFalse ()
		{
			string xml = String.Format (
				             @"<?xml version='1.0'?>" +
				             @"<Obfuscator>" +
				             @"<Var name='InPath' value='{0}' />" +
				             @"<Var name='OutPath' value='{1}' />" +
				             @"<Var name='HidePrivateApi' value='false' />" +
				             @"<Module file='$(InPath)\AssemblyWithTypes.dll'>" +
				             @"</Module>" +
				             @"</Obfuscator>", TestHelper.InputPath, TestHelper.OutputPath);

			var obfuscator = TestHelper.BuildAndObfuscate ("AssemblyWithTypes", string.Empty, xml);
			var map = obfuscator.Mapping;

			string assmName = "AssemblyWithTypes.dll";

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

			var classBType = inAssmDef.MainModule.GetType ("TestClasses.InternalClass");
			var classB = map.GetClass (new TypeKey (classBType));

			Assert.IsTrue (classB.Status == ObfuscationStatus.Skipped, "Internal class is obfuscated");

			var classAType = inAssmDef.MainModule.GetType ("TestClasses.PublicClass");
			var classA = map.GetClass (new TypeKey (classAType));
			var classAmethod1 = FindByName (classAType, "PrivateMethod");
			var classAmethod2 = FindByName (classAType, "PublicMethod");

			var classAMethod1 = map.GetMethod (new MethodKey (classAmethod1));
			var classAMethod2 = map.GetMethod (new MethodKey (classAmethod2));

			Assert.IsTrue (classA.Status == ObfuscationStatus.Renamed, "Public class is not obfuscated");
			Assert.IsTrue (classAMethod1.Status == ObfuscationStatus.Skipped, "private method is obfuscated.");
			Assert.IsTrue (classAMethod2.Status == ObfuscationStatus.Renamed, "pubilc method is not obfuscated.");
		}

		[Test]
		public void CheckHidePrivateApiTrue ()
		{
			string xml = String.Format (
				             @"<?xml version='1.0'?>" +
				             @"<Obfuscator>" +
				             @"<Var name='InPath' value='{0}' />" +
				             @"<Var name='OutPath' value='{1}' />" +
				             @"<Var name='HidePrivateApi' value='true' />" +
				             @"<Module file='$(InPath)\AssemblyWithTypes.dll'>" +
				             @"</Module>" +
				             @"</Obfuscator>", TestHelper.InputPath, TestHelper.OutputPath);

			var obfuscator = TestHelper.BuildAndObfuscate ("AssemblyWithTypes", string.Empty, xml);
			var map = obfuscator.Mapping;

			string assmName = "AssemblyWithTypes.dll";
			AssemblyDefinition inAssmDef = AssemblyDefinition.ReadAssembly (
				                                        Path.Combine (TestHelper.InputPath, assmName));

			var classBType = inAssmDef.MainModule.GetType ("TestClasses.InternalClass");
			var classB = map.GetClass (new TypeKey (classBType));

			Assert.IsTrue (classB.Status == ObfuscationStatus.Renamed, "Internal class should have been obfuscated");
		}

		[Test]
		public void CheckKeepPublicApiFalse ()
		{
			string xml = String.Format (
				             @"<?xml version='1.0'?>" +
				             @"<Obfuscator>" +
				             @"<Var name='InPath' value='{0}' />" +
				             @"<Var name='OutPath' value='{1}' />" +
				             @"<Var name='HidePrivateApi' value='true' />" +
				             @"<Module file='$(InPath)\AssemblyWithTypes.dll'>" +
				             @"</Module>" +
				             @"</Obfuscator>", TestHelper.InputPath, TestHelper.OutputPath);

			var obfuscator = TestHelper.BuildAndObfuscate ("AssemblyWithTypes", string.Empty, xml);
			var map = obfuscator.Mapping;

			string assmName = "AssemblyWithTypes.dll";

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

			TypeDefinition classAType = inAssmDef.MainModule.GetType ("TestClasses.PublicClass");
			MethodDefinition classAmethod1 = FindByName (classAType, "PrivateMethod");
			MethodDefinition classAmethod2 = FindByName (classAType, "PublicMethod");

			ObfuscatedThing classAMethod1 = map.GetMethod (new MethodKey (classAmethod1));
			ObfuscatedThing classAMethod2 = map.GetMethod (new MethodKey (classAmethod2));

			var classA = map.GetClass (new TypeKey (classAType));
			Assert.IsTrue (classA.Status == ObfuscationStatus.Renamed, "Public class should have been obfuscated");
			Assert.IsTrue (classAMethod1.Status == ObfuscationStatus.Renamed, "private method is not obfuscated.");
			Assert.IsTrue (classAMethod2.Status == ObfuscationStatus.Renamed, "pubilc method is not obfuscated.");

			var protectedMethod = FindByName (classAType, "ProtectedMethod");
			var protectedAfter = map.GetMethod (new MethodKey (protectedMethod));
			Assert.IsTrue (protectedAfter.Status == ObfuscationStatus.Renamed, "protected method is not obfuscated.");
		}

		[Test]
		public void CheckKeepPublicApiTrue ()
		{
			string xml = String.Format (
				             @"<?xml version='1.0'?>" +
				             @"<Obfuscator>" +
				             @"<Var name='InPath' value='{0}' />" +
				             @"<Var name='OutPath' value='{1}' />" +
				             @"<Var name='KeepPublicApi' value='true' />" +
				             @"<Var name='HidePrivateApi' value='true' />" +
				             @"<Module file='$(InPath)\AssemblyWithTypes.dll'>" +
				             @"</Module>" +
				             @"</Obfuscator>", TestHelper.InputPath, TestHelper.OutputPath);

			Obfuscar.Obfuscator obfuscator = TestHelper.BuildAndObfuscate ("AssemblyWithTypes", string.Empty, xml);
			var map = obfuscator.Mapping;

			string assmName = "AssemblyWithTypes.dll";

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

			TypeDefinition classAType = inAssmDef.MainModule.GetType ("TestClasses.PublicClass");
			MethodDefinition classAmethod1 = FindByName (classAType, "PrivateMethod");
			MethodDefinition classAmethod2 = FindByName (classAType, "PublicMethod");

			ObfuscatedThing classAMethod1 = map.GetMethod (new MethodKey (classAmethod1));
			ObfuscatedThing classAMethod2 = map.GetMethod (new MethodKey (classAmethod2));
			var classA = map.GetClass (new TypeKey (classAType));
			Assert.IsTrue (classA.Status == ObfuscationStatus.Skipped, "Public class shouldn't have been obfuscated");
			Assert.IsTrue (classAMethod1.Status == ObfuscationStatus.Renamed, "private method is not obfuscated.");
			Assert.IsTrue (classAMethod2.Status == ObfuscationStatus.Skipped, "pubilc method is obfuscated.");

			var protectedMethod = FindByName (classAType, "ProtectedMethod");
			var protectedAfter = map.GetMethod (new MethodKey (protectedMethod));
			Assert.IsTrue (protectedAfter.Status == ObfuscationStatus.Skipped, "protected method is obfuscated.");
		}

		[Test]
		public void CheckSkipNamespace ()
		{
			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)\AssemblyWithTypes.dll'>" +
				                      @"<SkipNamespace name='TestClasses1' />" +
				                      @"</Module>" +
				                      @"</Obfuscator>", TestHelper.InputPath, TestHelper.OutputPath);

			Obfuscar.Obfuscator obfuscator = TestHelper.BuildAndObfuscate ("AssemblyWithTypes", string.Empty, xml);
			var map = obfuscator.Mapping;

			string assmName = "AssemblyWithTypes.dll";

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

			TypeDefinition classAType = inAssmDef.MainModule.GetType ("TestClasses1.PublicClass");
			var classA = map.GetClass (new TypeKey (classAType));
			Assert.IsTrue (classA.Status == ObfuscationStatus.Skipped, "Public class shouldn't have been obfuscated");
		}
	}
}