Implement EnvDTE.CodeFunction.Attributes

The T4MVC template checks method attributes.

src/AddIns/Misc/PackageManagement/Project/Src/EnvDTE/CodeFunction.cs

@@ -60,7 +60,7 @@ namespace ICSharpCode.PackageManagement.EnvDTE
 		}
 		
 		public virtual CodeElements Attributes {
-			get { throw new NotImplementedException(); }
+			get { return new CodeAttributes(method); }
 		}
 		
 		public virtual bool CanOverride {

src/AddIns/Misc/PackageManagement/Test/Src/EnvDTE/CodeFunctionTests.cs

@@ -85,6 +85,11 @@ namespace PackageManagement.Tests.EnvDTE
 			helper.MakeMethodVirtual();
 		}
 		
+		void AddMethodAttribute(string attributeTypeName)
+		{
+			helper.AddAttributeToMethod(attributeTypeName);
+		}
+		
 		[Test]
 		public void Access_PublicFunction_ReturnsPublic()
 		{
@@ -361,5 +366,18 @@ namespace PackageManagement.Tests.EnvDTE
 			
 			Assert.IsTrue(canOverride);
 		}
+		
+		[Test]
+		public void Attributes_MethodHasOneAttribute_ReturnsOneAttribute()
+		{
+			CreatePublicFunction("MyClass.MyFunction");
+			AddMethodAttribute("System.ObsoleteAttribute");
+			
+			CodeElements attributes = codeFunction.Attributes;
+			
+			CodeAttribute2 attribute = attributes.FirstCodeAttribute2OrDefault();
+			Assert.AreEqual(1, attributes.Count);
+			Assert.AreEqual("System.ObsoleteAttribute", attribute.FullName);
+		}
 	}
 }

src/AddIns/Misc/PackageManagement/Test/Src/Helpers/AttributeHelper.cs

@@ -43,11 +43,18 @@ namespace PackageManagement.Tests.Helpers
 			NamedArguments.Add(name, value);
 		}
 		
-		public void AddAttributeToClass(IClass @class)
+		public void AddAttributeToClass(IClass c)
 		{
 			var attributes = new List<IAttribute>();
 			attributes.Add(Attribute);
-			@class.Stub(c => c.Attributes).Return(attributes);
+			c.Stub(item => item.Attributes).Return(attributes);
+		}
+		
+		public void AddAttributeToMethod(IMethod method)
+		{
+			var attributes = new List<IAttribute>();
+			attributes.Add(Attribute);
+			method.Stub(m => m.Attributes).Return(attributes);
 		}
 	}
 }

src/AddIns/Misc/PackageManagement/Test/Src/Helpers/ClassHelper.cs

@@ -33,10 +33,10 @@ namespace PackageManagement.Tests.Helpers
 			Class.Stub(c => c.CompilationUnit).Return(CompilationUnitHelper.CompilationUnit);
 		}
 		
-		public void AddAttributeToClass(string name)
+		public void AddAttributeToClass(string attributeTypeName)
 		{
 			var attributeHelper = new AttributeHelper();
-			attributeHelper.CreateAttribute(name);
+			attributeHelper.CreateAttribute(attributeTypeName);
 			attributeHelper.AddAttributeToClass(Class);
 		}
 		

src/AddIns/Misc/PackageManagement/Test/Src/Helpers/MethodHelper.cs

@@ -133,5 +133,12 @@ namespace PackageManagement.Tests.Helpers
 		{
 			Method.Stub(m => m.IsVirtual).Return(true);
 		}
+		
+		public void AddAttributeToMethod(string attributeTypeName)
+		{
+			var attributeHelper = new AttributeHelper();
+			attributeHelper.CreateAttribute(attributeTypeName);
+			attributeHelper.AddAttributeToMethod(Method);
+		}
 	}
 }