merge with kirill

1 files changed, 32 insertions, 7 deletions

diff --git a/nova/tests/api/openstack/contrib/test_floating_ips.py b/nova/tests/api/openstack/contrib/test_floating_ips.py
index 9e079c9b0..0faeaeb39 100644
--- a/nova/tests/api/openstack/contrib/test_floating_ips.py
+++ b/nova/tests/api/openstack/contrib/test_floating_ips.py
@@ -15,24 +15,33 @@
 from nova import context
 from nova import db
 from nova import test
+
+import stubout
+import webob
+
 from nova.api.openstack.contrib.floating_ips import FloatingIPController
 from nova.api.openstack.contrib.floating_ips import \
     _translate_floating_ip_view
 
 
 class FloatingIpTest(test.TestCase):
-    address = "10.10.10.10"
+    floating_ip_address = "10.10.10.10"
+    fixed_ip_address = '100.100.100.100'
+
+    def _create_fixed_ip(self):
+        """Create a fixed ip object. Returns address as string"""
+        return db.fixed_ip_create(self.context,
+                                  {'address': self.fixed_ip_address})
 
     def _create_floating_ip(self):
-        """Create a volume object."""
-        host = "fake_host"
+        """Create a floating ip object. Returns address as string"""
         return db.floating_ip_create(self.context,
-                                     {'address': self.address,
-                                      'host': host})
+                                     {'address': self.floating_ip_address})
 
     def setUp(self):
         super(FloatingIpTest, self).setUp()
-        self.floating_ips = FloatingIPController()
+        self.controller = FloatingIPController()
+        self.stubs = stubout.StubOutForTesting()
         self.context = context.get_admin_context()
 
     def test_translate_floating_ip_view(self):
@@ -42,6 +51,22 @@ class FloatingIpTest(test.TestCase):
         view = _translate_floating_ip_view(floating_ip)
         self.assertTrue('floating_ip' in view)
         self.assertTrue(view['floating_ip']['id'])
-        self.assertEqual(view['floating_ip']['ip'], self.address)
+        self.assertEqual(view['floating_ip']['ip'], self.floating_ip_address)
         self.assertEqual(view['floating_ip']['fixed_ip'], None)
         self.assertEqual(view['floating_ip']['instance_id'], None)
+
+
+    def test_associate_by_address(self):
+        fixed_ip_address = self._create_fixed_ip()
+        floating_ip_address = self._create_floating_ip()
+        floating_ip = db.floating_ip_get_by_address(self.context, floating_ip_address)
+
+        self.assertEqual(floating_ip['address'], self.floating_ip_address)
+        self.assertEqual(floating_ip['fixed_ip_id'], None)
+        body = {'associate_address': {'fixed_ip': self.fixed_ip_address}}
+
+        req = webob.Request.blank('/v1.1/floating_ips//associate')
+        raise Exception(req.__dict__)
+        #self.controller.associate(req, self.floating_ip_address, body)
+        #self.assertEqual(floating_ip['fixed_ip'], self.fixed_ip_address)
+