Code rebuild command fix. (#2694)

* Fix command.

* Add test for code-rebuild command.

Author: kporras07

src/Commands/Env/CodeRebuildCommand.php

@@ -37,7 +37,7 @@ public function rebuild(
         $site_env
     ) {
         $this->requireSiteIsNotFrozen($site_env);
-        $site = $this->getSite($site_env);
+        $site = $this->getSiteById($site_env);
         $env = $this->getEnv($site_env);
 
         if ($env->getName() === 'test' || $env->getName() === 'live') {

tests/Functional/EnvCommandsTest.php

@@ -471,4 +471,18 @@ public function testViewCommand()
         );
         $this->assertEquals($expectedUrl, $url);
     }
+
+    /**
+     * @test
+     * @covers \Pantheon\Terminus\Commands\Env\CodeRebuildCommand
+     *
+     * @group env
+     * @group short
+     */
+    public function testCodeRebuild()
+    {
+        $this->terminus(
+            sprintf('env:code-rebuild %s', $this->getSiteEnv())
+        );
+    }
 }