Classify the '=' sign in variable/parameter/property as if it was an operator.

CyrusNajmabadi · CyrusNajmabadi · commit a6ea497ff1f1 · 2014-12-18T19:38:19.000-08:00
diff --git a/src/services/services.ts b/src/services/services.ts
@@ -4985,10 +4985,17 @@ module ts {
                 }
 
                 if (isPunctuation(tokenKind)) {
-                    // the '=' in a variable declaration is special cased here.
                     if (token) {
+                        if (tokenKind === SyntaxKind.EqualsToken) {
+                            // the '=' in a variable declaration is special cased here.
+                            if (token.parent.kind === SyntaxKind.VariableDeclaration ||
+                                token.parent.kind === SyntaxKind.PropertyDeclaration ||
+                                token.parent.kind === SyntaxKind.Parameter) {
+                                return ClassificationTypeNames.operator;
+                            }
+                        }
+
                         if (token.parent.kind === SyntaxKind.BinaryExpression ||
-                            token.parent.kind === SyntaxKind.VariableDeclaration ||
                             token.parent.kind === SyntaxKind.PrefixUnaryExpression ||
                             token.parent.kind === SyntaxKind.PostfixUnaryExpression ||
                             token.parent.kind === SyntaxKind.ConditionalExpression) {
diff --git a/tests/cases/fourslash/syntacticClassificationsConflictMarkers1.ts b/tests/cases/fourslash/syntacticClassificationsConflictMarkers1.ts
@@ -12,7 +12,7 @@ var c = classification;
 verify.syntacticClassificationsAre(
     c.keyword("class"), c.className("C"), c.punctuation("{"),
     c.comment("<<<<<<< HEAD"),
-    c.text("v"), c.punctuation("="), c.numericLiteral("1"), c.punctuation(";"),
+    c.text("v"), c.operator("="), c.numericLiteral("1"), c.punctuation(";"),
     c.comment("======="),
     c.text("v"), c.punctuation("="), c.numericLiteral("2"), c.punctuation(";"),
     c.comment(">>>>>>> Branch - a"),
