code & test

laravel · Jan 4, 2024 · e972ac0 · e972ac0
1 parent e64f6fb
commit e972ac0
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 5 deletions.
diff --git a/src/Compiler.php b/src/Compiler.php
@@ -272,7 +272,7 @@ public function compileInclude($value)
     protected function compileServers($value)
     {
         $value = preg_replace_callback('/@servers\(\[(.*?)\]\)/s', function ($matches) {
-            return '@servers(['.preg_replace('/\s+/', '', $matches[1]).'])';
+            return '@servers(['.str(preg_replace('/\s+/', ' ', $matches[1]))->squish()->trim(' ')->value().'])';
         }, $value);
 
         $pattern = $this->createMatcher('servers');

diff --git a/tests/CompilerTest.php b/tests/CompilerTest.php
@@ -43,7 +43,7 @@ public function test_it_compiles_server_statement()
         $compiler = new Compiler();
         $result = $compiler->compile($str);
 
-        $this->assertSame($result, "<?php \$__container->servers(['foo'=>'bar']); ?>");
+        $this->assertSame($result, "<?php \$__container->servers(['foo' => 'bar']); ?>");
 
         $str = <<<'EOL'
 @servers([
@@ -57,7 +57,7 @@ public function test_it_compiles_server_statement()
         $compiler = new Compiler();
         $result = $compiler->compile($str);
 
-        $this->assertSame($result, "<?php \$__container->servers(['foo'=>['bar','baz','bah']]); ?>");
+        $this->assertSame($result, "<?php \$__container->servers(['foo' => [ 'bar', 'baz', 'bah' ]]); ?>");
 
         $str = <<<'EOL'
 @servers([
@@ -68,14 +68,47 @@ public function test_it_compiles_server_statement()
         $compiler = new Compiler();
         $result = $compiler->compile($str);
 
-        $this->assertSame($result, "<?php \$__container->servers(['foo'=>['bar'],'bar'=>['baz']]); ?>");
+        $this->assertSame($result, "<?php \$__container->servers(['foo' => ['bar'], 'bar' => ['baz']]); ?>");
 
         $str = <<<'EOL'
 @servers(['foo' => 'bar'])
 EOL;
         $compiler = new Compiler();
         $result = $compiler->compile($str);
 
-        $this->assertSame($result, "<?php \$__container->servers(['foo'=>'bar']); ?>");
+        $this->assertSame($result, "<?php \$__container->servers(['foo' => 'bar']); ?>");
+    }
+
+    public function test_it_compiles_server_statement_with_setup_raw_format()
+    {
+        $str = <<<'EOL'
+@setup
+    $user = 'foo';
+    $server = 'bar';
+    $port = 22;
+@endsetup
+
+@servers([
+    'foo' => "{$user}@{$server} -p {$port}"
+])
+EOL;
+        $compiler = new Compiler();
+        $result = $compiler->compile($str);
+
+        $expected = <<<EOL
+<?php \$port = isset(\$port) ? \$port : null; ?>
+<?php \$server = isset(\$server) ? \$server : null; ?>
+<?php \$user = isset(\$user) ? \$user : null; ?>
+<?php
+    \$user = 'foo';
+    \$server = 'bar';
+    \$port = 22;
+?>
+
+<?php \$__container->servers(['foo' => "{\$user}@{\$server} -p {\$port}"]); ?>
+EOL;
+
+
+        $this->assertSame($result, $expected);
     }
 }