From a6d3ca9ad67e7d195a3870c7c9ece258e79813df Mon Sep 17 00:00:00 2001
From: "Vitaliy.Bibaev" <vitaliy.bibaev@jetbrains.com>
Date: Tue, 5 Sep 2017 13:12:10 +0300
Subject: [PATCH] Add tests for assignment statements

---
 .../debugger/streams/trace/dsl/DslTestCase.kt  | 18 ++++++++++++++++++
 .../testData/dsl/java/assignment.out           |  2 ++
 .../testData/dsl/java/nestedAssigment.out      |  6 ++++++
 3 files changed, 26 insertions(+)
 create mode 100644 plugins/stream-debugger/testData/dsl/java/assignment.out
 create mode 100644 plugins/stream-debugger/testData/dsl/java/nestedAssigment.out

diff --git a/plugins/stream-debugger/src/test/java/com/intellij/debugger/streams/trace/dsl/DslTestCase.kt b/plugins/stream-debugger/src/test/java/com/intellij/debugger/streams/trace/dsl/DslTestCase.kt
index 0c19462d89df..7fe04d6c7a13 100644
--- a/plugins/stream-debugger/src/test/java/com/intellij/debugger/streams/trace/dsl/DslTestCase.kt
+++ b/plugins/stream-debugger/src/test/java/com/intellij/debugger/streams/trace/dsl/DslTestCase.kt
@@ -141,6 +141,24 @@ abstract class DslTestCase(private val directoryName: String, private val dsl: D
     }
   }
 
+  fun testAssignment() {
+    doTest {
+      val a = declare(variable("int", "a"), true)
+      a.assign(+"100")
+    }
+  }
+
+  fun testNestedAssigment() {
+    doTest {
+      val a = declare(variable("int", "a"), true)
+      ifBranch(+"true") {
+        a.assign(+"100")
+      }.elseBranch {
+        a.assign(+"200")
+      }
+    }
+  }
+
   private fun doTest(block: Dsl.() -> Unit) {
     dsl.block()
     check()
diff --git a/plugins/stream-debugger/testData/dsl/java/assignment.out b/plugins/stream-debugger/testData/dsl/java/assignment.out
new file mode 100644
index 000000000000..d0f6693bd94a
--- /dev/null
+++ b/plugins/stream-debugger/testData/dsl/java/assignment.out
@@ -0,0 +1,2 @@
+int a;
+a = 100;
diff --git a/plugins/stream-debugger/testData/dsl/java/nestedAssigment.out b/plugins/stream-debugger/testData/dsl/java/nestedAssigment.out
new file mode 100644
index 000000000000..1c573b01689f
--- /dev/null
+++ b/plugins/stream-debugger/testData/dsl/java/nestedAssigment.out
@@ -0,0 +1,6 @@
+int a;
+if(true) {
+  a = 100;
+} else { 
+  a = 200;
+};