Add traits test extension for TokenKind

atifaziz · Mar 26, 2019 · 563e7e6 · 563e7e6
1 parent cf176e3
commit 563e7e6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/CSharpMinifier/Token.cs b/src/CSharpMinifier/Token.cs
@@ -52,6 +52,9 @@ public static TokenKindTraits GetTraits(this TokenKind kind)
                  ? TraitsByKind[i]
                  : throw new ArgumentOutOfRangeException(nameof(kind));
         }
+
+        public static bool HasTraits(this TokenKind kind, TokenKindTraits traits) =>
+            (kind.GetTraits() & traits) == traits;
     }
 
     public readonly struct Token : IEquatable<Token>

diff --git a/tests/TokenKindTests.cs b/tests/TokenKindTests.cs
@@ -44,7 +44,9 @@ public class TokenKindTests
 
         public void Traits(TokenKind kind, TokenKindTraits traits)
         {
-            Assert.That(kind.GetTraits(), Is.EqualTo(traits));
+            Assert.That(kind.GetTraits()       , Is.EqualTo(traits));
+            Assert.That(kind.HasTraits(traits) , Is.True);
+            Assert.That(kind.HasTraits(~traits), Is.False);
         }
 
         [Test]