svn commit: r1082056 - /incubator/libcloud/trunk/test/compute/test_deployment.py

[to...@apache.org]

Author: tomaz
Date: Wed Mar 16 05:48:14 2011
New Revision: 1082056

URL: http://svn.apache.org/viewvc?rev=1082056&view=rev
Log:
Add another test for deployment.

Modified:
    incubator/libcloud/trunk/test/compute/test_deployment.py

Modified: incubator/libcloud/trunk/test/compute/test_deployment.py
URL: http://svn.apache.org/viewvc/incubator/libcloud/trunk/test/compute/test_deployment.py?rev=1082056&r1=1082055&r2=1082056&view=diff
==============================================================================
--- incubator/libcloud/trunk/test/compute/test_deployment.py (original)
+++ incubator/libcloud/trunk/test/compute/test_deployment.py Wed Mar 16 05:48:14 2011
@@ -18,16 +18,27 @@ import sys
 import unittest
 
 from libcloud.compute.deployment import MultiStepDeployment, Deployment
+from libcloud.compute.deployment import SSHKeyDeployment
 from libcloud.compute.base import Node
 from libcloud.compute.types import NodeState
+from libcloud.compute.ssh import BaseSSHClient
 from libcloud.compute.drivers.ec2 import EC2NodeDriver
 
 class MockDeployment(Deployment):
     def run(self, node, client):
         return node
 
+class MockClient(BaseSSHClient):
+    def put(self, file, contents):
+        return contents
+
 class DeploymentTests(unittest.TestCase):
 
+    def setUp(self):
+        self.node = Node(id=1, name='test', state=NodeState.RUNNING,
+                   public_ip=['1.2.3.4'], private_ip='1.2.3.5',
+                   driver=EC2NodeDriver)
+
     def test_multi_step_deployment(self):
         msd = MultiStepDeployment()
         self.assertEqual(len(msd.steps), 0)
@@ -35,10 +46,13 @@ class DeploymentTests(unittest.TestCase)
         msd.add(MockDeployment())
         self.assertEqual(len(msd.steps), 1)
 
-        node = Node(id=1, name='test', state=NodeState.RUNNING,
-                   public_ip=['1.2.3.4'], private_ip='1.2.3.5',
-                   driver=EC2NodeDriver)
-        self.assertEqual(node, msd.run(node=node, client=None))
+        self.assertEqual(self.node, msd.run(node=self.node, client=None))
+
+    def test_ssh_key_deployment(self):
+        sshd = SSHKeyDeployment(key='1234')
+
+        self.assertEqual(self.node, sshd.run(node=self.node,
+                        client=MockClient(hostname='localhost')))
 
 if __name__ == '__main__':
     sys.exit(unittest.main())