test: call PhpStreamWrapper::restore() as soon as possible

If it is not called, assertions may not work correctly.

Author: kenjis

tests/system/CLI/CLITest.php

@@ -76,9 +76,9 @@ public function testWaitZero(): void
         $time = time();
         CLI::wait(0);
 
-        $this->assertCloseEnough(0, time() - $time);
-
         PhpStreamWrapper::restore();
+
+        $this->assertCloseEnough(0, time() - $time);
     }
 
     public function testPrompt(): void
@@ -90,9 +90,9 @@ public function testPrompt(): void
 
         $output = CLI::prompt('What is your favorite color?');
 
-        $this->assertSame($expected, $output);
-
         PhpStreamWrapper::restore();
+
+        $this->assertSame($expected, $output);
     }
 
     public function testPromptByMultipleKeys(): void
@@ -105,14 +105,13 @@ public function testPromptByMultipleKeys(): void
         $options = ['Playing game', 'Sleep', 'Badminton'];
         $output  = CLI::promptByMultipleKeys('Select your hobbies:', $options);
 
+        PhpStreamWrapper::restore();
+
         $expected = [
             0 => 'Playing game',
             1 => 'Sleep',
         ];
-
         $this->assertSame($expected, $output);
-
-        PhpStreamWrapper::restore();
     }
 
     public function testNewLine(): void