From c264cc74dc75d2bddda18120f6df9e278b025531 Mon Sep 17 00:00:00 2001
From: TomasVotruba <tomas.vot@gmail.com>
Date: Sun, 22 Oct 2017 12:07:17 +0200
Subject: [PATCH] add test for interface as well

---
 .../PhpParser/RemoveNodeRector/Correct/correct2.php.inc  | 9 +++++++++
 tests/Rector/Contrib/PhpParser/RemoveNodeRector/Test.php | 5 +++++
 .../PhpParser/RemoveNodeRector/Wrong/wrong2.php.inc      | 9 +++++++++
 3 files changed, 23 insertions(+)
 create mode 100644 tests/Rector/Contrib/PhpParser/RemoveNodeRector/Correct/correct2.php.inc
 create mode 100644 tests/Rector/Contrib/PhpParser/RemoveNodeRector/Wrong/wrong2.php.inc

diff --git a/tests/Rector/Contrib/PhpParser/RemoveNodeRector/Correct/correct2.php.inc b/tests/Rector/Contrib/PhpParser/RemoveNodeRector/Correct/correct2.php.inc
new file mode 100644
index 00000000000..33e442ce80d
--- /dev/null
+++ b/tests/Rector/Contrib/PhpParser/RemoveNodeRector/Correct/correct2.php.inc
@@ -0,0 +1,9 @@
+<?php declare(strict_types=1);
+
+class SomeVisitor implements \PhpParser\NodeVisitor
+{
+    public function leaveNode(\PhpParser\Node $node)
+    {
+        return \PhpParser\NodeTraverser::REMOVE_NODE;
+    }
+}
diff --git a/tests/Rector/Contrib/PhpParser/RemoveNodeRector/Test.php b/tests/Rector/Contrib/PhpParser/RemoveNodeRector/Test.php
index cd1562044c7..dd2fcd7e7ef 100644
--- a/tests/Rector/Contrib/PhpParser/RemoveNodeRector/Test.php
+++ b/tests/Rector/Contrib/PhpParser/RemoveNodeRector/Test.php
@@ -13,6 +13,11 @@ final class Test extends AbstractRectorTestCase
             __DIR__ . '/Wrong/wrong.php.inc',
             __DIR__ . '/Correct/correct.php.inc'
         );
+
+        $this->doTestFileMatchesExpectedContent(
+            __DIR__ . '/Wrong/wrong2.php.inc',
+            __DIR__ . '/Correct/correct2.php.inc'
+        );
     }
 
     /**
diff --git a/tests/Rector/Contrib/PhpParser/RemoveNodeRector/Wrong/wrong2.php.inc b/tests/Rector/Contrib/PhpParser/RemoveNodeRector/Wrong/wrong2.php.inc
new file mode 100644
index 00000000000..3ea1f54684e
--- /dev/null
+++ b/tests/Rector/Contrib/PhpParser/RemoveNodeRector/Wrong/wrong2.php.inc
@@ -0,0 +1,9 @@
+<?php declare(strict_types=1);
+
+class SomeVisitor implements \PhpParser\NodeVisitor
+{
+    public function leaveNode(\PhpParser\Node $node)
+    {
+        return false;
+    }
+}