tests: refactoring

nette · Oct 4, 2022 · 01bb9ed · 01bb9ed
1 parent e9387f9
commit 01bb9ed
Show file tree

Hide file tree

Showing 3 changed files with 45 additions and 53 deletions.
diff --git a/tests/PhpGenerator/Extractor.extractAll.phpt b/tests/PhpGenerator/Extractor.extractAll.phpt
@@ -30,59 +30,8 @@ sameFile(__DIR__ . '/expected/Extractor.traits.expect', (string) $file);
 $file = (new Extractor(file_get_contents(__DIR__ . '/fixtures/bodies.php')))->extractAll();
 sameFile(__DIR__ . '/expected/Extractor.bodies.expect', (string) $file);
 
-$file = (new Extractor(
-	<<<'XX'
-<?php
-class Class1
-{
-	public function foo()
-	{
-		new class {
-			function bar() {
-			}
-		};
-	}
-}
-
-function () {};
-
-/** doc */
-function foo(A $a): B|C
-{
-	function bar()
-	{
-	}
-}
-
-XX
-))->extractAll();
-Assert::type(Nette\PhpGenerator\PhpFile::class, $file);
-Assert::match(<<<'XX'
-<?php
-
-class Class1
-{
-	public function foo()
-	{
-		new class {
-			function bar() {
-			}
-		};
-	}
-}
-
-/**
- * doc
- */
-function foo(A $a): B|C
-{
-	function bar()
-	{
-	}
-}
-XX
-, (string) $file);
-
+$file = (new Extractor(file_get_contents(__DIR__ . '/fixtures/extractor.php')))->extractAll();
+sameFile(__DIR__ . '/expected/Extractor.expect', (string) $file);
 
 Assert::exception(function () {
 	(new Extractor(''));

diff --git a/tests/PhpGenerator/expected/Extractor.expect b/tests/PhpGenerator/expected/Extractor.expect
@@ -0,0 +1,22 @@
+<?php
+
+class Class1
+{
+	public function foo()
+	{
+		new class {
+			function bar() {
+			}
+		};
+	}
+}
+
+/**
+ * doc
+ */
+function foo(A $a): B|C
+{
+	function bar()
+	{
+	}
+}
diff --git a/tests/PhpGenerator/fixtures/extractor.php b/tests/PhpGenerator/fixtures/extractor.php
@@ -0,0 +1,21 @@
+<?php
+class Class1
+{
+	public function foo()
+	{
+		new class {
+			function bar() {
+			}
+		};
+	}
+}
+
+function () {};
+
+/** doc */
+function foo(A $a): B|C
+{
+	function bar()
+	{
+	}
+}