Added capability of change Credentials after ProxmoxVE object creation

zzantares · Jun 27, 2014 · 2a0925f · 2a0925f
1 parent 1b71ebe
commit 2a0925f
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 7 deletions.
diff --git a/src/ProxmoxVE.php b/src/ProxmoxVE.php
@@ -201,6 +201,17 @@ public function getCredentials()
     }
 
 
+    /**
+     * Assign the passed Credentials object to the ProxmoxVE.
+     *
+     * @param ZzAntares\ProxmoxVE\Credentials $credentials to assign.
+     */
+    public function setCredentials($credentials)
+    {
+        $this->credentials = $credentials;
+    }
+
+
     /**
      * Performs a GET request to the Proxmox server.
      *

diff --git a/tests/ProxmoxVETest.php b/tests/ProxmoxVETest.php
@@ -15,15 +15,18 @@ class ProxmoxVETest extends \PHPUnit_Framework_TestCase
 {
     public function setUp()
     {
-        $this->credentials = new Credentials('myproxmox.tld', 'root', 'abc123');
+        $fakeToken = new AuthToken('csrf', 'ticket', 'owner');
 
-        $this->proxmox = $this->getMockBuilder('ZzAntares\ProxmoxVE\ProxmoxVE')
-                        ->disableOriginalConstructor()
-                        ->getMock();
+        $this->credentials = $this->getMockBuilder('ZzAntares\ProxmoxVE\Credentials')
+                                  ->setMethods(array('login'))
+                                  ->setConstructorArgs(array('myproxmox.tld', 'root', 'abc123'))
+                                  ->getMock();
 
-        $this->proxmox->expects($this->any())
-            ->method('getCredentials')
-            ->will($this->returnValue($this->credentials));
+        $this->credentials->expects($this->any())
+            ->method('login')
+            ->will($this->returnValue($fakeToken));
+
+        $this->proxmox = new ProxmoxVE($this->credentials);
     }
 
 
@@ -43,6 +46,15 @@ public function testConstructorThrowsExceptionWhenBadParamsArePassed()
     }
 
 
+    public function testChangesCredentialsCorrectly()
+    {
+        $newCredentials = new Credentials('host', 'user', 'pass');
+        $this->proxmox->setCredentials($newCredentials);
+
+        $this->assertEquals($newCredentials, $this->proxmox->getCredentials());
+    }
+
+
     /*
      * Add test for get, post, put and delete functions. Need to create mocks. 
      */