[TEST] Add support for teardowns in yaml tests

Author: polyfractal

tests/Elasticsearch/Tests/YamlRunnerTest.php

@@ -136,7 +136,7 @@ public function setUp()
      * @dataProvider yamlProvider
      * @group sync
      */
-    public function testIntegration($testProcedure, bool $skip, $setupProcedure, string $fileName)
+    public function testIntegration($testProcedure, bool $skip, $setupProcedure, $teardownProcedure, string $fileName)
     {
         if ($skip) {
             static::markTestIncomplete($testProcedure);
@@ -151,14 +151,21 @@ public function testIntegration($testProcedure, bool $skip, $setupProcedure, str
             $this->waitForYellow();
         }
 
-        $this->processProcedure(current($testProcedure), key($testProcedure), $fileName);
+        try {
+            $this->processProcedure(current($testProcedure), key($testProcedure), $fileName);
+        } finally {
+            if (null !== $teardownProcedure) {
+                $this->processProcedure(current($teardownProcedure), 'teardown', $fileName);
+                $this->waitForYellow();
+            }
+        }
     }
 
     /**
      * @dataProvider yamlProvider
      * @group async
      */
-    public function testAsyncIntegration($testProcedure, bool $skip, $setupProcedure, string $fileName)
+    public function testAsyncIntegration($testProcedure, bool $skip, $setupProcedure, $teardownProcedure, string $fileName)
     {
         if ($skip) {
             static::markTestIncomplete($testProcedure);
@@ -173,7 +180,14 @@ public function testAsyncIntegration($testProcedure, bool $skip, $setupProcedure
             $this->waitForYellow();
         }
 
-        $this->processProcedure(current($testProcedure), key($testProcedure), $fileName, true);
+        try {
+            $this->processProcedure(current($testProcedure), key($testProcedure), $fileName, true);
+        } finally {
+            if (null !== $teardownProcedure) {
+                $this->processProcedure(current($teardownProcedure), 'teardown', $fileName);
+                $this->waitForYellow();
+            }
+        }
     }
 
     /**
@@ -917,6 +931,7 @@ private function splitDocument(SplFileInfo $file, string $path, string $filter =
         $documentsParsed = [];
         $setup = null;
         $setupSkip = false;
+        $teardown = null;
         $fileName = str_replace($path . '/', '', $file);
 
         if (array_key_exists($fileName, static::$fatalFiles)) {
@@ -964,8 +979,10 @@ private function splitDocument(SplFileInfo $file, string $path, string $filter =
             if (!$skip && key($documentParsed) === 'setup') {
                 $setup = $documentParsed;
                 $setupSkip = $skip;
-            } else {
-                $documentsParsed[] = [$documentParsed, $skip || $setupSkip, $setup, $fileName];
+            } else if (!$skip && key($documentParsed) === 'teardown') {
+                $teardown = $documentParsed;
+            }  else {
+                $documentsParsed[] = [$documentParsed, $skip || $setupSkip, $setup, $teardown, $fileName];
             }
         }