using System.ComponentModel;
using Microsoft.VisualStudio.TestTools.UnitTesting;
using Pinwheel.Extensions;

namespace Pinwheel
{
    [TestClass]
    public class AwardExtensionsTests
    {
        [TestMethod]
        public void GetAttribute_gets_attribute_of_proper_type()
        {
            var attribute = typeof(TestClass).GetAttribute<DisplayNameAttribute>();
            Assert.AreEqual("Name", attribute.DisplayName);
        }


        [DisplayName("Name")]
        [System.ComponentModel.Description("Description")]
        class TestClass { }
    }
}