From c687add579382a0910e985a36f88434312097dcc Mon Sep 17 00:00:00 2001
From: Vladimir Matveev <vladima@microsoft.com>
Date: Tue, 14 Mar 2017 22:36:38 -0700
Subject: [PATCH] add tests to add and remove members in class

---
 src/harness/unittests/textChanges.ts          | 42 +++++++++++++++++++
 .../textChanges/deleteNodeAfterInClass1.js    | 12 ++++++
 .../textChanges/deleteNodeAfterInClass2.js    | 12 ++++++
 .../textChanges/insertNodeAfterInClass1.js    | 12 ++++++
 .../textChanges/insertNodeAfterInClass2.js    | 12 ++++++
 5 files changed, 90 insertions(+)
 create mode 100644 tests/baselines/reference/textChanges/deleteNodeAfterInClass1.js
 create mode 100644 tests/baselines/reference/textChanges/deleteNodeAfterInClass2.js
 create mode 100644 tests/baselines/reference/textChanges/insertNodeAfterInClass1.js
 create mode 100644 tests/baselines/reference/textChanges/insertNodeAfterInClass2.js

diff --git a/src/harness/unittests/textChanges.ts b/src/harness/unittests/textChanges.ts
index 8724f7aa39..b2fc1ae577 100644
--- a/src/harness/unittests/textChanges.ts
+++ b/src/harness/unittests/textChanges.ts
@@ -667,5 +667,47 @@ class A {
                 }
             });
         }
+        {
+            const text = `
+class A {
+    x
+}
+`;
+            runSingleFileTest("insertNodeAfterInClass1", noop, text, /*validateNodes*/ false, (sourceFile, changeTracker) => {
+                changeTracker.insertNodeAfter(sourceFile, findChild("x", sourceFile), createProperty(undefined, undefined, "a", undefined, createKeywordTypeNode(SyntaxKind.BooleanKeyword), undefined), { suffix: newLineCharacter });
+            });
+        }
+        {
+            const text = `
+class A {
+    x;
+}
+`;
+            runSingleFileTest("insertNodeAfterInClass2", noop, text, /*validateNodes*/ false, (sourceFile, changeTracker) => {
+                changeTracker.insertNodeAfter(sourceFile, findChild("x", sourceFile), createProperty(undefined, undefined, "a", undefined, createKeywordTypeNode(SyntaxKind.BooleanKeyword), undefined), { suffix: newLineCharacter });
+            });
+        }
+        {
+            const text = `
+class A {
+    x;
+    y = 1;
+}
+`;
+            runSingleFileTest("deleteNodeAfterInClass1", noop, text, /*validateNodes*/ false, (sourceFile, changeTracker) => {
+                changeTracker.deleteNode(sourceFile, findChild("x", sourceFile));
+            });
+        }
+        {
+            const text = `
+class A {
+    x
+    y = 1;
+}
+`;
+            runSingleFileTest("deleteNodeAfterInClass2", noop, text, /*validateNodes*/ false, (sourceFile, changeTracker) => {
+                changeTracker.deleteNode(sourceFile, findChild("x", sourceFile));
+            });
+        }
     });
 }
\ No newline at end of file
diff --git a/tests/baselines/reference/textChanges/deleteNodeAfterInClass1.js b/tests/baselines/reference/textChanges/deleteNodeAfterInClass1.js
new file mode 100644
index 0000000000..d8d54153ea
--- /dev/null
+++ b/tests/baselines/reference/textChanges/deleteNodeAfterInClass1.js
@@ -0,0 +1,12 @@
+===ORIGINAL===
+
+class A {
+    x;
+    y = 1;
+}
+
+===MODIFIED===
+
+class A {
+    y = 1;
+}
diff --git a/tests/baselines/reference/textChanges/deleteNodeAfterInClass2.js b/tests/baselines/reference/textChanges/deleteNodeAfterInClass2.js
new file mode 100644
index 0000000000..8d18d82021
--- /dev/null
+++ b/tests/baselines/reference/textChanges/deleteNodeAfterInClass2.js
@@ -0,0 +1,12 @@
+===ORIGINAL===
+
+class A {
+    x
+    y = 1;
+}
+
+===MODIFIED===
+
+class A {
+    y = 1;
+}
diff --git a/tests/baselines/reference/textChanges/insertNodeAfterInClass1.js b/tests/baselines/reference/textChanges/insertNodeAfterInClass1.js
new file mode 100644
index 0000000000..87206e588f
--- /dev/null
+++ b/tests/baselines/reference/textChanges/insertNodeAfterInClass1.js
@@ -0,0 +1,12 @@
+===ORIGINAL===
+
+class A {
+    x
+}
+
+===MODIFIED===
+
+class A {
+    x
+    a: boolean;
+}
diff --git a/tests/baselines/reference/textChanges/insertNodeAfterInClass2.js b/tests/baselines/reference/textChanges/insertNodeAfterInClass2.js
new file mode 100644
index 0000000000..1d583d8ac9
--- /dev/null
+++ b/tests/baselines/reference/textChanges/insertNodeAfterInClass2.js
@@ -0,0 +1,12 @@
+===ORIGINAL===
+
+class A {
+    x;
+}
+
+===MODIFIED===
+
+class A {
+    x;
+    a: boolean;
+}