Merge 7a0b58a into e03e34e

Author: tylercd100

src/Api/DigitalOceanRebooter.php

@@ -0,0 +1,88 @@
+<?php
+
+namespace Tylercd100\Rebooter\Api;
+
+use GuzzleHttp\Client;
+use Tylercd100\Rebooter\ApiRebooter;
+
+class DigitalOceanRebooter extends ApiRebooter {
+
+    protected $token;
+    protected $server_id;
+    protected $host;
+    protected $client;
+    
+    /**
+     * @param string $token     API Token from DigitalOcean.com
+     * @param number $server_id The ID of the droplet you want to control
+     * @param string $host      The api host
+     * @param Client $client    The guzzle client to use
+     */
+    public function __construct($token, $server_id, $host = "api.digitalocean.com", Client $client = null){
+
+        if(!$client instanceof Client){
+            $client = new Client();
+        }
+
+        $this->client = $client;
+        $this->token = $token;
+        $this->server_id = $server_id;
+        $this->host = $host;
+    }
+
+    /**
+     * Executes a Boot command
+     * @return \Psr\Http\Message\ResponseInterface
+     */
+    public function boot(){
+        return $this->exec("power_on");
+    }
+
+    /**
+     * Executes a Reboot command
+     * @return \Psr\Http\Message\ResponseInterface
+     */
+    public function reboot(){
+        return $this->exec("reboot");
+    }
+
+    /**
+     * Executes a Shutdown command
+     * @return \Psr\Http\Message\ResponseInterface
+     */
+    public function shutdown(){
+        return $this->exec("power_off");
+    }
+
+    /**
+     * Builds the request URL for the API call
+     * @param  string $action The DigitalOcean API action
+     * @return string
+     */
+    protected function buildRequestUrl($action){
+        return "https://{$this->host}/v2/droplets/{$this->server_id}/actions";
+    }
+
+    /**
+     * Builds the request data for the API call
+     * @param  string $action The DigitalOcean API action
+     * @return array
+     */
+    protected function buildRequestData($action){
+        return ["type"=>$action];
+    }
+
+    /**
+     * Executes a command on the API
+     * @param string $action
+     * @return \Psr\Http\Message\ResponseInterface
+     */
+    protected function exec($action){
+        $url  = $this->buildRequestUrl($action);
+        $data = $this->buildRequestData($action);
+        return $this->client->request('POST', $url, [
+            'auth' => [$this->token,""],
+            'form_params' => $data,
+        ]);
+    }
+}

src/Api/LinodeRebooter.php

@@ -8,25 +8,25 @@
 class LinodeRebooter extends ApiRebooter {
 
     protected $token;
-    protected $linodeId;
+    protected $server_id;
     protected $host;
     protected $client;
 
     /**
-     * @param string $token    API Token from Linode.com
-     * @param number $linodeId The ID of the linode you want to control
-     * @param string $host     The api host
-     * @param Client $client   The guzzle client to use
+     * @param string $token     API Token from Linode.com
+     * @param number $server_id The ID of the linode you want to control
+     * @param string $host      The api host
+     * @param Client $client    The guzzle client to use
      */
-    public function __construct($token, $linodeId, $host = "api.linode.com", Client $client = null){
+    public function __construct($token, $server_id, $host = "api.linode.com", Client $client = null){
 
         if(!$client instanceof Client){
             $client = new Client();
         }
 
         $this->client = $client;
         $this->token = $token;
-        $this->linodeId = $linodeId;
+        $this->server_id = $server_id;
         $this->host = $host;
     }
 
@@ -60,7 +60,7 @@ public function shutdown(){
      * @return string
      */
     protected function buildRequestUrl($action){
-        return "https://{$this->host}/?api_key={$this->token}&api_action={$action}&LinodeID={$this->linodeId}";
+        return "https://{$this->host}/?api_key={$this->token}&api_action={$action}&LinodeID={$this->server_id}";
     }
 
     /**

tests/DigitalOceanTest.php

@@ -0,0 +1,46 @@
+<?php
+
+namespace Tylercd100\Rebooter\Tests;
+
+use Tylercd100\Rebooter\Api\DigitalOceanRebooter;
+use GuzzleHttp\Client;
+
+class DigitalOceanTest extends TestCase
+{
+    public function testItCreatesInstanceSuccessfully(){
+        $obj = new DigitalOceanRebooter("token",123456789);
+        $this->assertInstanceOf(DigitalOceanRebooter::class,$obj);
+    }
+
+    public function testItBuildsCorrectUrl(){
+        $obj = new DigitalOceanRebooter("token",123456789);
+        $actions = ['reboot','boot','shutdown'];
+        foreach($actions as $action){
+            $result = $this->invokeMethod($obj, 'buildRequestUrl', [$action]);
+            $expected = "https://api.linode.com/?api_key=token&api_action={$action}&LinodeID=123456789";
+            $this->assertEquals($expected,$result);
+        }
+    }
+
+    public function testItCallServerControllerMethods(){
+        $methods = ['reboot','boot','shutdown'];
+
+        foreach ($methods as $m) {
+            $mock = $this->getMock(DigitalOceanRebooter::class, array('exec'), ['token',1234]);
+            $mock->expects($this->once())
+                 ->method('exec');
+
+            $mock->{$m}();
+        }
+    }
+
+    public function testItCallsClientRequest(){
+        $mock = $this->getMock(Client::class, array('request'));
+
+        $mock->expects($this->once())
+             ->method('request');
+
+        $obj = new DigitalOceanRebooter('token',1234,'asdf.com',$mock);
+        $obj->reboot();
+    }
+}

tests/SshPasswordTest.php

@@ -27,10 +27,10 @@ public function testItCallServerControllerMethods(){
     }
 
     public function testSessionGetterAndSetter(){
-        $config = new Configuration("host",22);
-        $auth   = new Password("username", "password");
-        $session= new Session($config,$auth);
-        $obj    = new PasswordRebooter("host", "username", "password", 22);
+        $config  = new Configuration("host",22);
+        $auth    = new Password("username", "password");
+        $session = new Session($config,$auth);
+        $obj     = new PasswordRebooter("host", "username", "password", 22);
         $obj->setSession($session);
         $sessionFromGetter = $obj->getSession();
         $this->assertEquals($session,$sessionFromGetter);