Add UnitTest Trait for #4

src/Xunit.Categories/UnitTestAttribute.cs

@@ -0,0 +1,28 @@
+using System;
+using Xunit.Sdk;
+
+namespace Xunit.Categories
+{
+    [TraitDiscoverer(UnitTestDiscoverer.DiscovererTypeName, DiscovererUtil.AssemblyName)]
+    [AttributeUsage(AttributeTargets.Class | AttributeTargets.Method, AllowMultiple = true)]
+    public class UnitTestAttribute:Attribute, ITraitAttribute
+    {
+        public UnitTestAttribute()
+        {
+
+        }
+
+        public UnitTestAttribute(string name)
+        {
+            this.Identifier = name;
+        }
+
+        public UnitTestAttribute(long id)
+        {
+            this.Identifier = id.ToString();
+        }
+
+        public string Identifier { get; private set; }
+
+    }
+}

src/Xunit.Categories/UnitTestDiscoverer.cs

@@ -0,0 +1,21 @@
+using System.Collections.Generic;
+using Xunit.Abstractions;
+using Xunit.Sdk;
+
+namespace Xunit.Categories
+{
+    public class UnitTestDiscoverer:ITraitDiscoverer
+    {
+        internal const string DiscovererTypeName = DiscovererUtil.AssemblyName + "." + nameof(UnitTestDiscoverer);
+
+        public IEnumerable<KeyValuePair<string, string>> GetTraits(IAttributeInfo traitAttribute)
+        {
+            var name = traitAttribute.GetNamedArgument<string>("Identifier");
+
+            yield return new KeyValuePair<string, string>("Category", "UnitTest");
+
+            if (!string.IsNullOrWhiteSpace(name))
+                yield return new KeyValuePair<string, string>("UnitTest", name);
+        }
+    }
+}

src/Xunit.Categories/Xunit.Categories.csproj

@@ -4,8 +4,8 @@
     <Authors>Brendan Connolly</Authors>
     <Company>Brendan Connolly</Company>
     <Description>Friendly Traits for Xunit</Description>
-    <Version>2.0.0</Version>
-    <FileVersion>2.0.0</FileVersion>
+    <Version>2.0.1</Version>
+    <FileVersion>2.0.1</FileVersion>
 
     <TargetFramework>netstandard1.1</TargetFramework>
   </PropertyGroup>